Spaces:

anaghanagesh
/

LLM

Sleeping

App Files Files Community

anaghanagesh commited on 12 days ago

Commit

dd2a5d2

verified ·

1 Parent(s): dd16a13

Update app.py

Browse files

Files changed (1) hide show

app.py +232 -45

app.py CHANGED Viewed

@@ -1,69 +1,256 @@
-def create_molecule_viewer(smiles):
-    return f"""
-    <iframe
-        width="100%"
-        height="500"
-        frameborder="0"
-        srcdoc='
-        <html>
-        <head>
-        <script src="https://3Dmol.org/build/3Dmol-min.js"></script>
-        </head>
-        <body style="margin:0">
-        <div
-            id="viewer"
-            style="
-                width:100vw;
-                height:100vh;
-                position:relative;
-                background:white;
-            ">
-        </div>
-        <script>
-        let viewer = $3Dmol.createViewer(
-            document.getElementById("viewer"),
             {{
                 backgroundColor: "white"
             }}
         );
-        viewer.addModel(
-            "{smiles}",
-            "smi"
-        );
-        viewer.setStyle(
-            {{}},
-            {{
-                stick: {{}},
-                sphere: {{
-                    scale:0.3
-                }}
             }}
-        );
-        viewer.zoomTo();
-        viewer.spin(true);
-        viewer.render();
-        </script>
-        </body>
-        </html>
-        '>
-    </iframe>
-    """

+import gradio as gr
+from agents import run_pipeline
+# ====================================
+# 3D MOLECULE VIEWER
+# ====================================
+def create_molecule_viewer(smiles):
+    html = f"""
+    <html>
+    <head>
+    <script src="https://3Dmol.org/build/3Dmol-min.js"></script>
+    </head>
+    <body style="margin:0; background:white;">
+    <div
+        id="container"
+        style="
+            width:100%;
+            height:500px;
+            position:relative;
+        ">
+    </div>
+    <script>
+    let element =
+        document.getElementById("container");
+    let viewer =
+        $3Dmol.createViewer(
+            element,
             {{
                 backgroundColor: "white"
             }}
         );
+    viewer.addModel(
+        `{smiles}`,
+        "smi"
+    );
+    viewer.setStyle(
+        {{}},
+        {{
+            stick: {{
+                radius:0.18
+            }},
+            sphere: {{
+                scale:0.30
             }}
+        }}
+    );
+    viewer.zoomTo();
+    viewer.spin(true);
+    viewer.render();
+    </script>
+    </body>
+    </html>
+    """
+    return html
+# ====================================
+# MAIN ANALYSIS FUNCTION
+# ====================================
+def analyze(symptoms, followup, audio):
+    try:
+        combined_input = ""
+        # TEXT INPUT
+        if symptoms and symptoms.strip():
+            combined_input += symptoms
+        # FOLLOW-UP INPUT
+        if followup and followup.strip():
+            combined_input += (
+                " " + followup
+            )
+        # AUDIO SUPPORT
+        result = run_pipeline(
+            combined_input,
+            audio
+        )
+        response = result["response"]
+        pubmed = "\n".join(
+            result["pubmed"]
+        )
+        molecule_info = ""
+        molecule_html = ""
+        # MOLECULE DISPLAY
+        if result["molecules"]:
+            molecule = result["molecules"][0]
+            molecule_info = f"""
+Drug: {molecule['name']}
+SMILES:
+{molecule['smiles']}
+Molecular Weight:
+{molecule['molecular_weight']}
+"""
+            molecule_html = create_molecule_viewer(
+                molecule["smiles"]
+            )
+        return (
+            response,
+            pubmed,
+            molecule_info,
+            molecule_html
+        )
+    except Exception as e:
+        print("APP ERROR:", e)
+        return (
+            f"Error: {str(e)}",
+            "",
+            "",
+            ""
+        )
+# ====================================
+# CUSTOM CSS
+# ====================================
+custom_css = """
+body {
+    background:
+    linear-gradient(
+        to right,
+        #141e30,
+        #243b55
+    );
+    color:white;
+}
+.gradio-container {
+    animation:
+    fadeIn 1.5s ease-in-out;
+}
+textarea {
+    font-size:16px !important;
+}
+"""
+# ====================================
+# UI
+# ====================================
+with gr.Blocks(
+    theme=gr.themes.Soft(),
+    css=custom_css
+) as demo:
+    gr.Markdown(
+        """
+# 🧬 AI Biomedical Assistant
+Real-Time Retrieval-Augmented Biomedical Intelligence System
+"""
+    )
+    symptoms_input = gr.Textbox(
+        label="🩺 Symptoms",
+        placeholder="Example: chest pain, nausea, fever..."
+    )
+    followup_input = gr.Textbox(
+        label="💬 Follow-Up Information",
+        placeholder="Example: duration, severity..."
+    )
+    audio_input = gr.Audio(
+        sources=["microphone"],
+        type="filepath",
+        label="🎙️ Record Symptoms"
+    )
+    analyze_btn = gr.Button(
+        "🧠 Analyze"
+    )
+    diagnosis_output = gr.Markdown(
+        label="🧠 Biomedical Report"
+    )
+    pubmed_output = gr.Textbox(
+        label="📄 PubMed IDs",
+        lines=5
+    )
+    molecule_output = gr.Textbox(
+        label="🧪 Drug Information",
+        lines=8
+    )
+    molecule_viewer = gr.HTML(
+        label="🧬 Animated 3D Molecule"
+    )
+    analyze_btn.click(
+        fn=analyze,
+        inputs=[
+            symptoms_input,
+            followup_input,
+            audio_input
+        ],
+        outputs=[
+            diagnosis_output,
+            pubmed_output,
+            molecule_output,
+            molecule_viewer
+        ]
+    )
+demo.queue()
+demo.launch()