Robust deployment: lazy data loading and package structure

app.py

@@ -2,51 +2,45 @@ import gradio as gr
 from agent.core import MedicalAgent
 from data.loader import fetch_med_dialog_sample, format_dialogue_context
 import os
-from dotenv import load_dotenv
 
-load_dotenv()
+# Globals
+agent = None
+medical_context = ""
 
-# Initialize the agent
-# Note: User needs to provide GEMINI_API_KEY in .env
-try:
-    agent = MedicalAgent()
-    # Fetch some initial context to 'prime' the agent
-    initial_data = fetch_med_dialog_sample(3)
-    medical_context = format_dialogue_context(initial_data)
-except Exception as e:
-    print(f"Warning: Agent initialization failed (likely missing API key): {e}")
-    agent = None
-    medical_context = ""
+def get_agent():
+    global agent, medical_context
+    if agent is None:
+        try:
+            agent = MedicalAgent()
+            # Try to get a sample, but don't crash if it fails
+            try:
+                initial_data = fetch_med_dialog_sample(2)
+                medical_context = format_dialogue_context(initial_data)
+            except:
+                medical_context = "No additional context available."
+        except Exception as e:
+            print(f"Error initializing agent: {e}")
+    return agent
 
 def medical_chat_interface(message, history):
-    if not agent:
-        return "Error: Agent not initialized. Please ensure GEMINI_API_KEY is set in your .env file."
+    current_agent = get_agent()
+    if not current_agent:
+        return "Error: Agent not initialized. Please check your GEMINI_API_KEY in Secrets."
     
     try:
-        response = agent.get_response(message, context=medical_context)
+        response = current_agent.get_response(message, context=medical_context)
         return response
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-# Custom Theme for a Professional Medical Look
-theme = gr.themes.Soft(
-    primary_hue="blue",
-    secondary_hue="slate",
-    neutral_hue="slate",
-    font=[gr.themes.GoogleFont("Inter"), "ui-sans-serif", "system-ui", "sans-serif"],
-).set(
-    body_background_fill="*neutral_50",
-    block_background_fill="white",
-    block_border_width="1px",
-)
+# Custom Theme
+theme = gr.themes.Soft(primary_hue="blue")
 
 with gr.Blocks(theme=theme, title="Medical AI Agent") as demo:
     gr.Markdown(
         """
         # 🏥 Medical AI Assistant
-        Developed by **[shahnewazkabirrafi017-hub](https://github.com/shahnewazkabirrafi017-hub)**
-        
-        This agent is grounded in the **OpenMed** dataset collection and powered by **Gemini 3 Flash**.
+        Powered by **Gemini 3 Flash** and **OpenMed** datasets.
         
         *⚠️ Disclaimer: This is an AI tool for educational purposes. Consult a doctor for medical advice.*
         """
@@ -54,7 +48,7 @@ with gr.Blocks(theme=theme, title="Medical AI Agent") as demo:
     
     chatbot = gr.ChatInterface(
         fn=medical_chat_interface,
-        examples=["What are common symptoms of iron deficiency?", "How can I improve my sleep hygiene?", "Explain what hypertension is in simple terms."]
+        examples=["What are common symptoms of iron deficiency?", "Explain hypertension in simple terms."]
     )
 
 if __name__ == "__main__":