Update app.py

app.py

@@ -764,9 +764,6 @@ def extract_aa_seq_cb(structure_file, protein_text):
     return prot_seq_out, status_html
 
 def extract_sa_seq_cb(structure_file, protein_text):
-    """
-    Extract structure-aware (SA) sequence using Foldseek.
-    """
 
     prot_seq_out = (protein_text or "").strip()
     msgs = []
@@ -1230,19 +1227,20 @@ with gr.Blocks() as demo:
                         elem_id="example-btn",
                         # variant="secondary"
                     )
-                protein_seq.render()
-                drug_seq.render()
+                
                 structure_file.render()
-                btn_extract_aa = gr.Button(
-                    "Extract amino acid sequence",
-                    elem_id="extract-aa-btn"
-                )
+                with gr.Row():
+                    btn_extract_aa = gr.Button(
+                        "Extract amino acid sequence",
+                        elem_id="extract-aa-btn"
+                    )
                 
-                btn_extract_sa = gr.Button(
-                    "Extract structure-aware sequence",
-                    elem_id="extract-sa-btn"
-                )
-
+                    btn_extract_sa = gr.Button(
+                        "Extract structure-aware sequence",
+                        elem_id="extract-sa-btn"
+                    )
+                protein_seq.render()
+                drug_seq.render()
             # ────────────────
             # RIGHT PANEL
             # ────────────────
@@ -1261,16 +1259,16 @@ with gr.Blocks() as demo:
                     value="1",
                     interactive=True,
                 )
-
-                btn_infer = gr.Button(
-                    "Inference",
-                    elem_id="inference-btn"
-                )
-
-                clear_btn = gr.Button(
-                    "Clear",
-                    elem_id="clear-btn"
-                )
+                with gr.Row():
+                    btn_infer = gr.Button(
+                        "Inference",
+                        elem_id="inference-btn"
+                    )
+                
+                    clear_btn = gr.Button(
+                        "Clear",
+                        elem_id="clear-btn"
+                    )
 
     # ───────────────────────────────
     # Outputs
@@ -1280,8 +1278,6 @@ with gr.Blocks() as demo:
         output_full  = gr.HTML(elem_id="result-full") 
 
 
-
-
     # ───────────────────────────────
     # Example Loader Callback
     # ───────────────────────────────