Update app.py

app.py

@@ -1,10 +1,25 @@
 import gradio as gr
+import sys
+import random
+sys.path.append("scripts/")
+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"
 
+def get_foldseek_seq(pdb_path):
+    parsed_seqs = get_struc_seq(
+        "bin/foldseek",
+        pdb_path,
+        ["A"],
+        process_id=random.randint(0, 10000000),
+    )["A"]
+    return parsed_seqs
+
+
+
 # Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown(
@@ -33,8 +48,10 @@ 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="Stability Prediction", interactive=False)
+            prediction_output = gr.Textbox(label=str(output), interactive=False)
 
             predict_button.click(fn=predict_stability, inputs=[model_choice, organism_choice, pdb_file], outputs=prediction_output)