Update app.py

app.py

@@ -1,66 +1,128 @@
 # app.py
 """
 The main Gradio user interface for Project Asclepius.
-This file defines the layout and connects UI components to the backend logic.
+This file defines the complete, multi-tab layout and connects the UI components
+to the powerful asynchronous backend logic in the orchestrator. It is designed
+for a clean, professional, and responsive user experience.
 """
 import gradio as gr
+from PIL import Image
+
+# Import the master orchestrator which contains all our application logic
 from modules import orchestrator
 
-# --- UI Layout and Components ---
+# --- UI Configuration and Styling ---
+
+# Load custom CSS for a polished, professional look
+try:
+    with open("static/style.css", "r", encoding="utf-8") as f:
+        custom_css = f.read()
+except FileNotFoundError:
+    print("Warning: style.css not found. Using default Gradio styles.")
+    custom_css = ""
+
+# Block Title and Header
+APP_TITLE = "🧠 Project Asclepius: The Cognitive Medical Nexus"
+APP_DESCRIPTION = (
+    "Welcome. This is an AI-powered system that synthesizes public medical data from trusted sources "
+    "(like the National Library of Medicine) to provide clear, evidence-based information. "
+    "**This application does not provide medical advice, diagnosis, or treatment.**"
+)
 
-# Load custom CSS for a polished look
-with open("static/style.css", "r", encoding="utf-8") as f:
-    custom_css = f.read()
+# --- Gradio Application Layout ---
 
-# Define the main application blocks
-with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
-    # Header
-    with gr.Row():
-        # You can add your own logo.png to the static folder
-        # gr.Image("static/logo.png", width=100, show_label=False, show_download_button=False)
-        gr.Markdown("# 🧠 Project Asclepius: The Cognitive Medical Nexus")
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"), css=custom_css, title="Project Asclepius") as demo:
 
-    gr.Markdown("An AI-powered system that synthesizes public medical data to provide clear, evidence-based information. **This is not a diagnostic tool.**")
+    # Main Header
+    gr.Markdown(f"# {APP_TITLE}")
+    gr.Markdown(APP_DESCRIPTION)
 
     # Define the main tabs for different functionalities
     with gr.Tabs():
-        # --- TAB 1: Symptom Synthesizer ---
-        with gr.TabItem("Symptom Synthesizer"):
+
+        # =====================================================================
+        # TAB 1: Symptom & Condition Synthesizer
+        # =====================================================================
+        with gr.TabItem("Symptom & Condition Synthesizer"):
             with gr.Row():
                 with gr.Column(scale=2):
                     symptom_input = gr.Textbox(
                         lines=5,
                         label="Describe Symptoms or Ask a Medical Question",
-                        placeholder="e.g., 'I have a sharp pain in my chest and feel dizzy' or 'What are the latest treatments for migraines?'"
+                        placeholder="e.g., 'I have a sharp pain behind my right eye and I'm sensitive to light' or 'What are the latest treatments for psoriatic arthritis?'"
+                    )
+                    image_input = gr.Image(
+                        type="pil",
+                        label="Optional: Upload a Relevant Image",
+                        info="For visual symptoms like a skin rash or swelling. The AI will describe what it sees objectively."
                     )
-                    # Optional image input can be added here
-                    # image_input = gr.Image(type="pil", label="Upload a relevant image (e.g., skin rash)", optional=True)
-                
                 with gr.Column(scale=1):
-                    submit_btn = gr.Button("Synthesize Information", variant="primary")
+                    synthesis_submit_btn = gr.Button("Synthesize Information", variant="primary", scale=1)
+
+            gr.Markdown("---")
+            synthesis_output = gr.Markdown(label="Synthesized Report", value="Your synthesized report will appear here...")
 
+        # =====================================================================
+        # TAB 2: Drug Interaction & Safety Analyzer
+        # =====================================================================
+        with gr.TabItem("Drug Interaction & Safety Analyzer"):
+            with gr.Row():
+                with gr.Column(scale=2):
+                    drug_input = gr.Textbox(
+                        lines=5,
+                        label="Enter a List of Medications (separated by commas)",
+                        placeholder="e.g., lisinopril, metformin, atorvastatin, aspirin, ozempic"
+                    )
+                with gr.Column(scale=1):
+                    interaction_submit_btn = gr.Button("Analyze Drug Safety", variant="primary", scale=1)
+            
             gr.Markdown("---")
-            synthesis_output = gr.Markdown(label="Synthesized Report")
+            interaction_output = gr.Markdown(label="Drug Safety Report", value="Your drug safety and interaction report will appear here...")
 
-        # --- TAB 2: Research Deep Dive (Placeholder) ---
-        with gr.TabItem("Research Deep Dive"):
-            gr.Markdown("## Feature Coming Soon\nThis section will allow you to do a deep dive on a specific disease, gene, or drug, pulling in comprehensive research data.")
-            # Add inputs and outputs for this tab here
+        # =====================================================================
+        # TAB 3: About & Disclaimer
+        # =====================================================================
+        with gr.TabItem("About & Disclaimer"):
+            gr.Markdown(
+                """
+                ## About Project Asclepius
+                Project Asclepius is a proof-of-concept demonstrating how modern AI can be combined with high-quality, open-source medical APIs to make complex information more accessible. It is built using the Gradio framework, powered by Google's Gemini large language model, and leverages real-time data from:
+                - **PubMed:** For biomedical literature and research abstracts.
+                - **ClinicalTrials.gov:** For information on active clinical studies.
+                - **OpenFDA:** For drug adverse event reports and recalls.
+                - **RxNorm:** For standardizing drug names and checking for interactions.
 
-        # --- TAB 3: Drug Interaction Analyzer (Placeholder) ---
-        with gr.TabItem("Drug Interaction Analyzer"):
-            gr.Markdown("## Feature Coming Soon\nThis tool will analyze potential interactions and safety concerns for a list of medications.")
-            # Add inputs and outputs for this tab here
+                This tool is for informational and educational purposes only.
 
-    # --- Event Handlers ---
-    submit_btn.click(
+                ## **⚠️ Full Disclaimer**
+                **This tool is NOT a substitute for professional medical advice, diagnosis, or treatment.** The information provided is generated by an AI model synthesizing publicly available data and may contain inaccuracies, be incomplete, or be misinterpreted. The AI is not a doctor.
+                
+                **ALWAYS seek the advice of your physician or other qualified health provider** with any questions you may have regarding a medical condition. Never disregard professional medical advice or delay in seeking it because of something you have read on this platform. Reliance on any information provided by this application is solely at your own risk.
+                """
+            )
+
+    # --- Wire the UI Components to the Backend Orchestrator ---
+
+    # Handler for the Symptom Synthesizer tab
+    synthesis_submit_btn.click(
         fn=orchestrator.run_symptom_synthesis,
-        inputs=[symptom_input], # Add image_input here if using
+        inputs=[symptom_input, image_input],
         outputs=[synthesis_output],
         api_name="symptom_synthesis",
-        show_progress="full" # Provides a great user experience
+        # Provide excellent user feedback during the async process
+        show_progress="full"
+    )
+
+    # Handler for the Drug Interaction Analyzer tab
+    interaction_submit_btn.click(
+        fn=orchestrator.run_drug_interaction_analysis,
+        inputs=[drug_input],
+        outputs=[interaction_output],
+        api_name="drug_interaction_analysis",
+        show_progress="full"
     )
 
-# Launch the application
+# --- Launch the Application ---
 if __name__ == "__main__":
+    # The `share=True` argument can be used to create a public link if running locally
     demo.launch(debug=True)