Add Tkinter-based GUI for protein location prediction tool

GUI.py → gui.py

@@ -1,5 +1,5 @@
 """
-Protein Location Predictor CLI
+Protein Location Predictor GUI
 This module provides a Tkinter-based GUI for loading FASTA files
      and running protein location prediction tools.
 """
@@ -13,7 +13,8 @@ FASTA_FILE_PATH = None  # Global or instance variable
 
 def load_fasta_file():
     """
-    Opens a file dialog for the user to select a FASTA file and stores the selected file path in a global variable.
+    Opens a file dialog for the user to select a FASTA file and
+      stores the selected file path in a global variable.
 
     If a file is selected, displays an information message with the file path.
     If no file is selected, displays a warning message.
@@ -25,7 +26,7 @@ def load_fasta_file():
     Global Variables:
         FASTA_FILE_PATH (str): Path to the selected FASTA file.
     """
-    global FASTA_FILE_PATH # pylint: disable=global-statement 
+    global FASTA_FILE_PATH # pylint: disable=global-statement
     FASTA_FILE_PATH = filedialog.askopenfilename(
         filetypes=[("FASTA files", "*.fasta *.fa")],
         title="Select a FASTA file"
@@ -67,7 +68,7 @@ def run_esm300():
     predict_with_esm(fasta_path = FASTA_FILE_PATH,
                      model = 'esmc_300m')
     torch.cuda.empty_cache()
-    
+
 def run_esm600():
     """
     Runs the protein location prediction process.
@@ -114,7 +115,7 @@ def menu():
     file_menu.add_command(label='Load FASTA', command=load_fasta_file)
     file_menu.add_command(label='Close', command=root.quit)
     file_menu.add_separator()
-    menubar.add_cascade(label="File", menu=file_menu, underline=0) 
+    menubar.add_cascade(label="File", menu=file_menu, underline=0)
 
     # help menu
     help_menu = Menu(menubar, tearoff=0)
@@ -122,17 +123,16 @@ def menu():
     help_menu.add_command(label='About...')
     menubar.add_cascade(label="Help", menu=help_menu, underline=0)
 
-    # =========================
-    # Add Buttons Below Menu
-    # =========================
-
-    btn_prost = tk.Button(root, text="Predict with Prost", command=run_prost) #Predict with Prost
+    btn_prost = tk.Button(root, text="Predict with Prost",
+                           command=run_prost) #Predict with Prost
     btn_prost.pack(pady=5)
 
-    btn_esm300 = tk.Button(root, text="Predict with ESM C 300m", command = run_esm300) #Predict with ESM C 300m
+    btn_esm300 = tk.Button(root, text="Predict with ESM C 300m",
+                            command = run_esm300) #Predict with ESM C 300m
     btn_esm300.pack(pady=5)
 
-    btn_esm300 = tk.Button(root, text="Predict with ESM C 600m", command = run_esm600) #Predict with ESM C 600m
+    btn_esm300 = tk.Button(root, text="Predict with ESM C 600m",
+                            command = run_esm600) #Predict with ESM C 600m
     btn_esm300.pack(pady=5)
 
     btn_exit = tk.Button(root, text="Exit", command=root.quit)
@@ -141,4 +141,7 @@ def menu():
     root.mainloop()
 
 if __name__ == "__main__":
-    menu()
+    menu()
+
+# End-of-file (EOF)
+