Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 
 # Assuming 'predict_stability' is your function that predicts protein stability
-def predict_stability(pdb_file=None, sequence=None):
+def predict_stability(model_choice, organism_choice, pdb_file=None, sequence=None):
     # Dummy return for illustration; replace with your actual prediction logic
-    return "Predicted Stability: Example Output"
+    return f"Predicted Stability using {model_choice} for {organism_choice}: Example Output"
 
 # Gradio Interface
 with gr.Blocks() as demo:
@@ -13,11 +13,21 @@ with gr.Blocks() as demo:
         **Predict the protein half-life from its sequence or PDB file.**
         """
     )
-    gr.Markdown("![model image](https://github.com/sagawatatsuya/PLTNUM/blob/main/model-image.png?raw=true)")
-
-    model_choice = gr.Radio(choices=["SaProt", "ESM-2"], label="Select PLTNUM's base model.", value="SaProt")
-    organism_choice = gr.Radio(choices=["Mouse", "Human"], label="Select the target organism.", value="Mouse")
     
+    gr.Image("https://github.com/sagawatatsuya/PLTNUM/blob/main/model-image.png?raw=true", label="Model Image")
+
+    # Model and Organism selection in the same row to avoid layout issues
+    with gr.Row():
+        model_choice = gr.Radio(
+            choices=["SaProt", "ESM-2"], 
+            label="Select PLTNUM's base model.", 
+            value="SaProt"
+        )
+        organism_choice = gr.Radio(
+            choices=["Mouse", "Human"], 
+            label="Select the target organism.", 
+            value="Mouse"
+        )
     
     with gr.Tabs():
         with gr.TabItem("Upload PDB File"):
@@ -26,7 +36,7 @@ with gr.Blocks() as demo:
             predict_button = gr.Button("Predict Stability")
             prediction_output = gr.Textbox(label="Stability Prediction", interactive=False)
 
-            predict_button.click(fn=predict_stability, inputs=[model_choice, pdb_file], outputs=prediction_output)
+            predict_button.click(fn=predict_stability, inputs=[model_choice, organism_choice, pdb_file], outputs=prediction_output)
         
         with gr.TabItem("Enter Protein Sequence"):
             gr.Markdown("### Enter the protein sequence:")
@@ -38,12 +48,13 @@ with gr.Blocks() as demo:
             predict_button = gr.Button("Predict Stability")
             prediction_output = gr.Textbox(label="Stability Prediction", interactive=False)
 
-            predict_button.click(fn=predict_stability, inputs=[model_choice, sequence], outputs=prediction_output)
+            predict_button.click(fn=predict_stability, inputs=[model_choice, organism_choice, sequence], outputs=prediction_output)
     
     gr.Markdown(
         """
         ### How to Use:
-        - **Select Model**: Choose between 'Model A' or 'Model B' for your prediction.
+        - **Select Model**: Choose between 'SaProt' or 'ESM-2' for your prediction.
+        - **Select Organism**: Choose between 'Mouse' or 'Human'.
         - **Upload PDB File**: Choose the 'Upload PDB File' tab and upload your file.
         - **Enter Sequence**: Alternatively, switch to the 'Enter Protein Sequence' tab and input your sequence.
         - **Predict**: Click 'Predict Stability' to receive the prediction.
@@ -57,4 +68,4 @@ with gr.Blocks() as demo:
         """
     )
 
-demo.launch()
+demo.launch()