Update app.py

app.py

@@ -6,8 +6,20 @@ from foldseek_util import get_struc_seq
 
 # Assuming 'predict_stability' is your function that predicts protein stability
 def predict_stability(model_choice, organism_choice, pdb_file=None, sequence=None):
-    # Dummy return for illustration; replace with your actual prediction logic
-    return f"Predicted Stability using {model_choice} for {organism_choice}: Example Output"
+    # Check if pdb_file is provided
+    if pdb_file:
+        pdb_path = pdb_file.name  # Get the path of the uploaded PDB file
+        sequence = get_foldseek_seq(pdb_path)
+        if not sequence:
+            return "Failed to extract sequence from the PDB file."
+    
+    # If sequence is provided directly
+    if sequence:
+        # Add logic to predict stability using the sequence
+        return f"Predicted Stability using {model_choice} for {organism_choice}: Example Output with sequence {sequence[:10]}..."
+    else:
+        return "No valid input provided."
+    
 
 def get_foldseek_seq(pdb_path):
     try:
@@ -50,7 +62,6 @@ with gr.Blocks() as demo:
         with gr.TabItem("Upload PDB File"):
             gr.Markdown("### Upload your PDB file:")
             pdb_file = gr.File(label="Upload PDB File")
-            output = get_foldseek_seq(pdb_file)
             
             predict_button = gr.Button("Predict Stability")
             prediction_output = gr.Textbox(label=str(output), interactive=False)