Update app.py

app.py

@@ -52,9 +52,9 @@ UPPER_3TO1  = {k.upper(): v for k, v in THREE_TO_ONE.items()}
 
 EXAMPLES = [
     {
-        "pdb": "6M0J", "label": "6M0J", "subtitle": "ACE2 ↔ SARS-CoV-2 RBD",
-        "desc": "Human ACE2 bound to SARS-CoV-2 spike RBD. Chain A = ACE2, Chain E = RBD.",
-        "mode": "Protein-Protein", "chain_a": "A", "chain_b": "E",
+        "pdb": "1YCR", "label": "1YCR", "subtitle": "MDM2 ↔ p53",
+        "desc": "MDM2 oncoprotein bound to the p53 transactivation peptide. Chain A = MDM2, Chain B = p53. Classic drug-target complex.",
+        "mode": "Protein-Protein", "chain_a": "A", "chain_b": "B",
     },
     {
         "pdb": "1BRS", "label": "1BRS", "subtitle": "Barnase ↔ Barstar",
@@ -1125,12 +1125,12 @@ def build_ui():
                                 ppi_a = gr.Textbox(
                                     label="Target Protein — Seq A",
                                     placeholder="Paste FASTA or raw sequence…  |  = chain separator",
-                                    lines=6, max_lines=12,
+                                    lines=4, max_lines=8,
                                 )
                                 ppi_b = gr.Textbox(
                                     label="Binder Protein — Seq B  ( | = chain separator)",
                                     placeholder="Paste FASTA or raw sequence…",
-                                    lines=6, max_lines=12,
+                                    lines=4, max_lines=8,
                                 )
 
                         # Ab-Ag input panel
@@ -1139,19 +1139,19 @@ def build_ui():
                                 ab_ag = gr.Textbox(
                                     label="Antigen / Target — Seq A",
                                     placeholder="Antigen sequence…",
-                                    lines=6, max_lines=12, scale=2,
+                                    lines=4, max_lines=8, scale=2,
                                 )
                                 with gr.Column(scale=3):
                                     with gr.Row():
                                         ab_h = gr.Textbox(
                                             label="Heavy Chain (H)",
                                             placeholder="VH sequence…",
-                                            lines=6, max_lines=12,
+                                            lines=4, max_lines=8,
                                         )
                                         ab_l = gr.Textbox(
                                             label="Light Chain (L)",
                                             placeholder="VL sequence…",
-                                            lines=6, max_lines=12,
+                                            lines=4, max_lines=8,
                                         )
 
                         with gr.Row():
@@ -1161,12 +1161,30 @@ def build_ui():
 
                         result_html = gr.HTML("")
 
-                        # ─── Lower section: examples + visualizations ─────
+                        # ─── Lower section: PDB controls + visualizations ─────
                         with gr.Row():
-                            with gr.Column(scale=2):
-                                gr.HTML('<div class="sec-hdr">Quick Examples</div>')
+                            # ── LEFT: Custom PDB Fetch ───────────────────
+                            with gr.Column(scale=1, min_width=240):
+                                gr.HTML('<div class="sec-hdr">Custom PDB Fetch</div>')
+                                pdb_in = gr.Textbox(label="PDB ID", placeholder="1YCR, 1BRS, 2VXQ…")
+
+                                # PPI chain selectors
+                                with gr.Group(visible=True) as ppi_chains_row:
+                                    ca_in = gr.Textbox(label="Side A chain(s)", placeholder="A or A,B")
+                                    cb_in = gr.Textbox(label="Side B chain(s)", placeholder="B or B,C")
+
+                                # Ab-Ag chain selectors
+                                with gr.Group(visible=False) as abag_chains_row:
+                                    ch_in  = gr.Textbox(label="Heavy",   placeholder="H")
+                                    cl_in  = gr.Textbox(label="Light",   placeholder="L")
+                                    cag_in = gr.Textbox(label="Antigen", placeholder="A or A,B")
 
-                                # Render example cards + buttons
+                                fetch_btn   = gr.Button("🔍 Fetch PDB", variant="secondary", size="sm")
+                                pdb_info_md = gr.Markdown("")
+
+                            # ── MIDDLE: Quick Examples ───────────────────
+                            with gr.Column(scale=1, min_width=240):
+                                gr.HTML('<div class="sec-hdr">Quick Examples</div>')
                                 example_buttons = []
                                 for ex in EXAMPLES:
                                     gr.HTML(
@@ -1178,23 +1196,7 @@ def build_ui():
                                     b = gr.Button(f"⬇ Load {ex['pdb']}", size="sm")
                                     example_buttons.append((b, ex))
 
-                                gr.HTML('<div class="sec-hdr" style="margin-top:18px">Custom PDB Fetch</div>')
-                                pdb_in = gr.Textbox(label="PDB ID", placeholder="6M0J, 1BRS, 2VXQ…")
-
-                                # PPI chain selectors
-                                with gr.Row(visible=True) as ppi_chains_row:
-                                    ca_in = gr.Textbox(label="Side A chain(s)", placeholder="A or A,B")
-                                    cb_in = gr.Textbox(label="Side B chain(s)", placeholder="E or B,C")
-
-                                # Ab-Ag chain selectors
-                                with gr.Row(visible=False) as abag_chains_row:
-                                    ch_in  = gr.Textbox(label="Heavy",   placeholder="H")
-                                    cl_in  = gr.Textbox(label="Light",   placeholder="L")
-                                    cag_in = gr.Textbox(label="Antigen", placeholder="A or A,B")
-
-                                fetch_btn   = gr.Button("🔍 Fetch PDB", variant="secondary")
-                                pdb_info_md = gr.Markdown("")
-
+                            # ── RIGHT: Visualizations ────────────────────
                             with gr.Column(scale=3):
                                 with gr.Tabs():
                                     with gr.Tab("🧬 Structure"):
@@ -1262,7 +1264,7 @@ with Integrated Gradients explainability.
 | Viewer | PDB filtered to **selected chains only** |
 | Hardware | HF ZeroGPU (dynamic A100); CPU fallback supported |
 
-**Examples** · **6M0J** — ACE2 ↔ SARS-CoV-2 RBD &nbsp;·&nbsp; **1BRS** — Barnase ↔ Barstar
+**Examples** · **1YCR** — MDM2 ↔ p53 &nbsp;·&nbsp; **1BRS** — Barnase ↔ Barstar
 
 🤗 [Harshit494/BALM-PPI](https://huggingface.co/Harshit494/BALM-PPI) &nbsp;·&nbsp;
 💻 [rgorantla04/BALM-PPI](https://github.com/rgorantla04/BALM-PPI)
@@ -1365,4 +1367,8 @@ if __name__ == "__main__":
         traceback.print_exc()
 
     demo = build_ui()
-    demo.queue(max_size=20).launch(css=CSS, theme=gr.themes.Soft())
+    demo.queue(max_size=20).launch(
+        css=CSS,
+        theme=gr.themes.Soft(),
+        ssr_mode=False,   # cleaner logs; SSR is experimental in Gradio 6
+    )