Update app.py

app.py

@@ -165,6 +165,20 @@ def function(model_name: str, num_molecules: int, seed_num: int):
 
 
 with gr.Blocks(theme=gr.themes.Ocean()) as demo:
+    # Add custom CSS for styling
+    gr.HTML("""
+    <style>
+    #metrics-container {
+        border: 1px solid rgba(128, 128, 128, 0.3);
+        border-radius: 8px;
+        padding: 15px;
+        margin-top: 15px;
+        margin-bottom: 15px;
+        background-color: rgba(255, 255, 255, 0.05);
+    }
+    </style>
+    """)
+
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("# DrugGEN: Target Centric De Novo Design of Drug Candidate Molecules with Graph Generative Deep Adversarial Networks")
@@ -186,8 +200,6 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
             
             with gr.Accordion("About DrugGEN Models", open=False):
                 gr.Markdown("""
-## Model Variations
-
 ### DrugGEN-AKT1
 This model is designed to generate molecules targeting the human AKT1 protein (UniProt ID: P31749).
 
@@ -269,7 +281,7 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
                 label="Download All Generated Molecules (SMILES format)",
             )
             
-            with gr.Box():
+            with gr.Group(elem_id="metrics-container"):
                 gr.Markdown("### Performance Metrics")
                 
                 with gr.Row():
@@ -320,7 +332,5 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
         ], 
         api_name="inference"
     )
-
 demo.queue()
-demo.launch()
-
+demo.launch()