jpuglia commited on
Commit
fbdd746
·
1 Parent(s): 2941a1c

Refactor GUI: Reintroduce Tkinter menu for Protein Tools application with updated button functionality

Browse files
Files changed (2) hide show
  1. src/cli.py → cli.py +5 -2
  2. src/my_utils.py +1 -1
src/cli.py → cli.py RENAMED
@@ -50,8 +50,11 @@ def menu():
50
  # Add Buttons Below Menu
51
  # =========================
52
 
53
- btn_fasta = tk.Button(root, text="Load FASTA", command=predict_with_prost)
54
- btn_fasta.pack(pady=5)
 
 
 
55
 
56
  btn_exit = tk.Button(root, text="Exit", command=root.quit)
57
  btn_exit.pack(pady=5)
 
50
  # Add Buttons Below Menu
51
  # =========================
52
 
53
+ btn_prost = tk.Button(root, text="Predict with Prost", command=predict_with_prost)
54
+ btn_prost.pack(pady=5)
55
+
56
+ btn_ESM = tk.Button(root, text="Predict with ESMC", command=print(NotImplementedError("ESM functionality not implemented yet."))) #type: ignore
57
+ btn_ESM.pack(pady=5)
58
 
59
  btn_exit = tk.Button(root, text="Exit", command=root.quit)
60
  btn_exit.pack(pady=5)
src/my_utils.py CHANGED
@@ -785,7 +785,7 @@ def predict_with_prost():
785
 
786
  print("Loading pre-trained SVM model for prediction...")
787
  try:
788
- predictor = load('/home/juan/ProteinLocationPredictor/ProteinLocationPredictor/Models/rfProst.joblib')
789
  except FileNotFoundError:
790
  print("Error: Could not find the model file '../ProteinLocationPredictor/Models/svmProst.joblib'")
791
  print("Please check the path to your trained model.")
 
785
 
786
  print("Loading pre-trained SVM model for prediction...")
787
  try:
788
+ predictor = load('/home/juan/ProteinLocationPredictor/Models/rfProst.joblib')
789
  except FileNotFoundError:
790
  print("Error: Could not find the model file '../ProteinLocationPredictor/Models/svmProst.joblib'")
791
  print("Please check the path to your trained model.")