Add downloadable example file buttons

app.py

@@ -86,9 +86,9 @@ def score_wrapper(protein_file, ligand_file, cphmd, qmmm):
 
 with gr.Blocks(title="GAIA Drug Discovery", theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
-    # 💊 GAIA Drug Discovery Platform
+    # 💊 GAIA: Physics-Based Drug Discovery Platform
     
-    **Upload protein and ligand files to predict binding affinity and safety.**
+    **Upload protein and ligand files to predict binding affinity and safety in seconds.**
     
     ### Features:
     - 🔬 Physics-based binding prediction (0.313 RMSE on T4-Lysozyme)
@@ -107,14 +107,19 @@ with gr.Blocks(title="GAIA Drug Discovery", theme=gr.themes.Soft()) as demo:
                 qmmm_check = gr.Checkbox(label="⚛️ QM/MM (GFN2-xTB)", value=False)
             
             submit_btn = gr.Button("🚀 Predict Binding", variant="primary")
-            
-            gr.Markdown("""
-            ### 📖 Example Files
-            - **Protein**: HLA-B27 receptor (receptor.pdb)
-            - **Ligand**: methylsulfone_good.pdb (known binder)
-            - **Ligand**: bulky_maleimide.pdb (known fail)
-            """)
-        
+    
+    # Example Files Section with Download Buttons
+    gr.Markdown("""
+    ### 📖 Example Files
+    Download these files to test GAIA:
+    """)
+    
+    with gr.Row():
+        gr.File(label="📄 Protein (receptor.pdb)", value="receptor.pdb", interactive=False, scale=1)
+        gr.File(label="📄 Ligand - Binder (methylsulfone_good.pdb)", value="methylsulfone_good.pdb", interactive=False, scale=1)
+        gr.File(label="📄 Ligand - Fail (bulky_maleimide.pdb)", value="bulky_maleimide.pdb", interactive=False, scale=1)
+    
+    with gr.Row():
         with gr.Column():
             output = gr.Textbox(label="📊 Results", lines=15, interactive=False)