Fixed small bug

kiwijuice56 · Jan 27, 2024 · fe4890a · fe4890a
1 parent 17891cc
commit fe4890a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -51,11 +51,11 @@
 
 # Initialize the protein
 protein = Protein(file, chain_id=chain_id, prompt_for_chain=visible_window)
-
+print(f"{DEFAULT_COLOR}Processing complete.")
 
 # If running the program from a command line, quit and do not show the window
 if not visible_window:
-    exit()
+    sys.exit()
 
 # Initialize window
 window = pyglet.window.Window(resizable=True, vsync=0)

diff --git a/protein.py b/protein.py
@@ -120,7 +120,7 @@ def __init__(self, pdb_path, prompt_for_chain=True, chain_id=None, verbose=False
                 chain = other_chain
                 break
 
-        print(self.OUTPUT_COLOR_GOOD + f"Rendering chain with ID '{chain_id}'.")
+        print(self.OUTPUT_COLOR_GOOD + f"Loading chain with ID '{chain_id}'.")
 
         # Parse the .pdb file for the protein sequence
         physical_record = None