Spaces:

anaghanagesh
/

LLM

Sleeping

App Files Files Community

anaghanagesh commited on 14 days ago

Commit

dd16a13

verified ·

1 Parent(s): bb16e93

Update app.py

Browse files

Files changed (1) hide show

app.py +13 -180

app.py CHANGED Viewed

@@ -1,21 +1,13 @@
-import gradio as gr
-from agents import run_pipeline
-# ====================================
-# 3D MOLECULE VIEWER
-# ====================================
 def create_molecule_viewer(smiles):
-    iframe = f"""
     <iframe
         width="100%"
-        height="520"
         frameborder="0"
         srcdoc='
         <html>
         <head>
@@ -24,7 +16,7 @@ def create_molecule_viewer(smiles):
         </head>
-        <body style="margin:0;">
         <div
             id="viewer"
@@ -38,16 +30,12 @@ def create_molecule_viewer(smiles):
         <script>
-        let element =
-            document.getElementById("viewer");
-        let viewer =
-            $3Dmol.createViewer(
-                element,
-                {{
-                    backgroundColor: "white"
-                }}
-            );
         viewer.addModel(
             "{smiles}",
@@ -59,7 +47,7 @@ def create_molecule_viewer(smiles):
             {{
                 stick: {{}},
                 sphere: {{
-                    scale: 0.30
                 }}
             }}
         );
@@ -75,162 +63,7 @@ def create_molecule_viewer(smiles):
         </body>
         </html>
         '>
     </iframe>
-    """
-    return iframe
-# ====================================
-# MAIN ANALYSIS
-# ====================================
-def analyze(symptoms, followup, audio):
-    try:
-        combined_input = symptoms
-        if followup and followup.strip():
-            combined_input += (
-                " " + followup
-            )
-        result = run_pipeline(
-            combined_input,
-            audio
-        )
-        response = result["response"]
-        pubmed = "\n".join(
-            result["pubmed"]
-        )
-        molecule_info = ""
-        molecule_html = ""
-        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:
-        return (
-            f"Error: {str(e)}",
-            "",
-            "",
-            ""
-        )
-# ====================================
-# UI
-# ====================================
-custom_css = """
-body {
-    background:
-    linear-gradient(
-        to right,
-        #141e30,
-        #243b55
-    );
-    color:white;
-}
-"""
-with gr.Blocks(
-    theme=gr.themes.Soft(),
-    css=custom_css
-) as demo:
-    gr.Markdown(
-        """
-# 💊 AI Biomedical Assistant
-Real-Time Biomedical Intelligence System
-"""
-    )
-    symptoms_input = gr.Textbox(
-        label="🩺 Symptoms",
-        placeholder="Example: nausea, chest pain..."
-    )
-    followup_input = gr.Textbox(
-        label="💬 Follow-Up Information"
-    )
-    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.launch()

 def create_molecule_viewer(smiles):
+    return f"""
     <iframe
         width="100%"
+        height="500"
         frameborder="0"
         srcdoc='
         <html>
         <head>
         </head>
+        <body style="margin:0">
         <div
             id="viewer"
         <script>
+        let viewer = $3Dmol.createViewer(
+            document.getElementById("viewer"),
+            {{
+                backgroundColor: "white"
+            }}
+        );
         viewer.addModel(
             "{smiles}",
             {{
                 stick: {{}},
                 sphere: {{
+                    scale:0.3
                 }}
             }}
         );
         </body>
         </html>
         '>
     </iframe>
+    """