Refactor GUI: Reintroduce Tkinter menu for Protein Tools application with updated button functionality
Browse files- src/cli.py → cli.py +5 -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 |
-
|
| 54 |
-
|
|
|
|
|
|
|
|
|
|
| 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/
|
| 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.")
|