Update app.py

app.py

@@ -33,7 +33,11 @@ def drug_discovery(disease, symptoms):
 
     # Convert binary to base64
     img_base64 = base64.b64encode(img_data).decode("utf-8")
-    img_html = f'<img src="data:image/png;base64,{img_base64}" alt="2D Molecule">'
+    img_html = f'''<div style="text-align:center; margin-top: 10px; animation: fadeIn 2s ease-in-out;">
+        <img src="data:image/png;base64,{img_base64}" alt="2D Molecule" 
+             style="border-radius: 16px; box-shadow: 0 6px 20px rgba(0,128,128,0.2); border: 1px solid #ccc;">
+        <div style='font-family: Arial, sans-serif; color: #2c3e50; margin-top: 8px; animation: slideUp 1.5s ease-in-out;'>💊 Visualized Drug Molecule (2D)</div>
+    </div>'''
 
     # 3D molecule
     mol3d = Chem.AddHs(mol)
@@ -41,36 +45,69 @@ def drug_discovery(disease, symptoms):
     AllChem.UFFOptimizeMolecule(mol3d)
     mb = Chem.MolToMolBlock(mol3d)
 
-    viewer = py3Dmol.view(width=400, height=400)
+    viewer = py3Dmol.view(width=420, height=420)
     viewer.addModel(mb, "mol")
-    viewer.setStyle({"stick": {}})
+    viewer.setStyle({"stick": {"colorscheme": "tealCarbon"}})
     viewer.setBackgroundColor("white")
     viewer.zoomTo()
+    viewer.rotate(1)
+    viewer.spin(True)
     viewer_html_raw = viewer._make_html()
 
-    # Use iframe rendering to ensure visibility in Gradio
-    viewer_html = f"""
-    <iframe srcdoc="{viewer_html_raw.replace('"', '&quot;')}" 
-            width="420" height="420" frameborder="0"></iframe>
-    """
+    viewer_html = f'''
+    <div style="text-align:center; margin-top: 20px; animation: zoomIn 2s ease-in-out;">
+        <iframe srcdoc="{viewer_html_raw.replace('"', '&quot;')}" 
+                width="440" height="440" frameborder="0" 
+                style="border-radius: 16px; box-shadow: 0 8px 30px rgba(0,128,128,0.25);"></iframe>
+        <div style='font-family: Arial, sans-serif; color: #2c3e50; margin-top: 8px; animation: slideUp 1.5s ease-in-out;'>🧬 Animated 3D Molecule (Stick View)</div>
+    </div>'''
 
     return literature, smiles, img_html, viewer_html
 
 # Gradio UI
-disease_input = gr.Textbox(label="Enter Disease", value="lung cancer")
-symptom_input = gr.Textbox(label="Enter Symptoms", value="shortness of breath, weight loss")
-lit_output = gr.Textbox(label="Literature Insights")
-smiles_output = gr.Textbox(label="SMILES String")
-img_output = gr.HTML(label="2D Molecule")
-viewer_output = gr.HTML(label="3D Molecule Viewer")
+disease_input = gr.Textbox(label="🏥 Enter Disease (e.g., lung cancer)", value="lung cancer")
+symptom_input = gr.Textbox(label="💉 Enter Symptoms (e.g., cough, weight loss)", value="shortness of breath, weight loss")
+lit_output = gr.Textbox(label="📰 Literature Insights from BioGPT")
+smiles_output = gr.Textbox(label="🧪 SMILES Representation")
+img_output = gr.HTML(label="🖼️ Molecule 2D Visualization")
+viewer_output = gr.HTML(label="🔬 3D Drug Molecule Animation")
+
+custom_css = """
+@keyframes fadeIn {
+    from {opacity: 0;}
+    to {opacity: 1;}
+}
+
+@keyframes slideUp {
+    from {transform: translateY(40px); opacity: 0;}
+    to {transform: translateY(0); opacity: 1;}
+}
+
+@keyframes zoomIn {
+    from {transform: scale(0.5); opacity: 0;}
+    to {transform: scale(1); opacity: 1;}
+}
+
+body {
+    background: linear-gradient(to right, #e0f7fa, #ffffff);
+    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+}
+
+.gradio-container {
+    animation: fadeIn 1.5s ease-in-out;
+}
+"""
 
 iface = gr.Interface(
     fn=drug_discovery,
     inputs=[disease_input, symptom_input],
     outputs=[lit_output, smiles_output, img_output, viewer_output],
-    title="🧬 Drug Discovery using LLMs",
-    description="Enter a disease and symptoms to generate literature insights and view the molecule in 2D and interactive 3D."
+    title="🏥 AI-Powered Drug Discovery for Hospitals",
+    description="This hospital-themed platform takes a disease and symptoms as input, retrieves biomedical insights using BioGPT, and visualizes potential drug molecules in 2D and animated 3D. Ideal for clinical research and pharma innovation.",
+    theme="default",
+    css=custom_css
 )
 
 iface.launch(share=True)
 
+