Update app.py

app.py

@@ -1,138 +1,155 @@
 import gradio as gr
 import os
 import requests
+from datetime import datetime
 
 # Mistral API setup
-MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY", "your_api_key_here")  # Replace with your actual API key
+MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY", "your_api_key_here")
 MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions"
 
-def query_mistral(prompt, max_tokens=1024, temperature=0.7):
-    """Query Mistral AI API"""
-    headers = {
-        "Authorization": f"Bearer {MISTRAL_API_KEY}",
-        "Content-Type": "application/json"
+# Local medical knowledge base
+MEDICAL_KNOWLEDGE = {
+    "headache": {
+        "assessment": "Could be tension-type, migraine, or other common headache",
+        "recommendations": [
+            "Rest in a quiet, dark room",
+            "Apply cold compress to forehead",
+            "Hydrate well (at least 8 glasses water/day)",
+            "Try gentle neck stretches",
+            "Consider OTC pain relievers (ibuprofen 200-400mg or paracetamol 500mg)"
+        ],
+        "warnings": [
+            "If sudden/severe 'thunderclap' headache",
+            "If with fever, stiff neck, or confusion",
+            "If following head injury"
+        ]
+    },
+    "fever": {
+        "assessment": "Common symptom of infection or inflammation",
+        "recommendations": [
+            "Rest and increase fluid intake",
+            "Use light clothing and keep room cool",
+            "Sponge with lukewarm water if uncomfortable",
+            "Paracetamol 500mg every 6 hours (max 4 doses/day)",
+            "Monitor temperature every 4 hours"
+        ],
+        "warnings": [
+            "Fever >39°C lasting >3 days",
+            "If with rash, difficulty breathing, or seizures",
+            "In infants <3 months with any fever"
+        ]
+    },
+    "sore throat": {
+        "assessment": "Often viral, but could be strep throat",
+        "recommendations": [
+            "Gargle with warm salt water 3-4 times daily",
+            "Drink warm liquids (tea with honey)",
+            "Use throat lozenges",
+            "Stay hydrated",
+            "Rest your voice"
+        ],
+        "warnings": [
+            "Difficulty swallowing or breathing",
+            "White patches on tonsils",
+            "Fever >38.5°C with throat pain"
+        ]
     }
+}
 
-    payload = {
-        "model": "mistral-tiny",
-        "messages": [{"role": "user", "content": prompt}],
-        "temperature": temperature,
-        "max_tokens": max_tokens
-    }
+def get_local_advice(symptom):
+    """Get structured medical advice from local knowledge base"""
+    symptom_lower = symptom.lower()
+    for key in MEDICAL_KNOWLEDGE:
+        if key in symptom_lower:
+            knowledge = MEDICAL_KNOWLEDGE[key]
+            response = "🔍 [Assessment]\n- " + knowledge["assessment"] + "\n\n"
+            response += "💡 [Self-Care Recommendations]\n" + "\n".join(["• " + item for item in knowledge["recommendations"]]) + "\n\n"
+            response += "⚠️ [When to Seek Medical Care]\n" + "\n".join(["• " + item for item in knowledge["warnings"]]) + "\n\n"
+            response += "📅 [Follow-up]\n• Re-evaluate in 2-3 days if not improving\n• See doctor if symptoms worsen or persist beyond 5 days"
+            return response
+    return None
 
+def query_mistral(prompt):
+    """Try API first, fallback to local knowledge"""
     try:
+        headers = {"Authorization": f"Bearer {MISTRAL_API_KEY}", "Content-Type": "application/json"}
+        payload = {
+            "model": "mistral-tiny",
+            "messages": [{"role": "user", "content": prompt}],
+            "temperature": 0.7,
+            "max_tokens": 1024
+        }
         response = requests.post(MISTRAL_API_URL, headers=headers, json=payload, timeout=10)
         response.raise_for_status()
         return response.json()['choices'][0]['message']['content']
-    except requests.exceptions.RequestException as e:
-        return f"I apologize, I'm currently unable to access my medical knowledge base. Please try again later."
-    except Exception as e:
-        return f"An unexpected error occurred. Please try rephrasing your question."
-
-def generate_medical_response(name, message):
-    """Generate structured medical response"""
-    prompt = f"""You are Dr. Alex, an experienced physician. {name} asks: "{message}"
+    except:
+        return None
 
-    Provide a detailed response with these sections (use bullet points):
-    
-    [Assessment]
-    - Brief professional assessment of the described condition
+def generate_response(name, message):
+    """Generate medical response with fallback"""
+    # First try local knowledge base
+    local_response = get_local_advice(message)
+    if local_response:
+        return f"Hello {name},\n\n{local_response}\n\nWishing you good health,\nDr. Alex"
     
-    [Recommended Actions] (3-5 points)
-    - Specific self-care measures
-    - Home remedies
-    - Lifestyle adjustments
+    # Then try API
+    prompt = f"""As Dr. Alex, provide structured advice for {name} who reports: "{message}"
+
+    Format response with these sections:
     
-    [Medical Treatments] (if applicable)
-    - OTC medications (with dosage examples)
-    - When to consider prescription treatments
+    [Assessment] - Brief professional evaluation
+    [Self-Care Recommendations] - 3-5 specific actionable steps
+    [When to Seek Medical Care] - Clear warning signs
+    [Follow-up] - Monitoring advice
     
-    [Warning Signs] (Red Flags)
-    - Symptoms that require immediate medical attention
+    Use bullet points, professional yet compassionate tone."""
     
-    [Follow-up Advice]
-    - When to revisit the issue
-    - Recommended tests or specialist consultation if needed
+    api_response = query_mistral(prompt)
+    if api_response:
+        return api_response
     
-    Keep the tone professional yet compassionate. Use simple language. 
-    Provide concrete, actionable advice. Limit to 250 words."""
+    # Final fallback
+    common_conditions = "\n".join([f"- {cond}" for cond in MEDICAL_KNOWLEDGE.keys()])
+    return f"""Hello {name},
 
-    return query_mistral(prompt)
+I'm currently unable to access detailed medical databases. For general advice:
 
-def chatbot_response(message, history, name):
-    """Personal medical assistant response function"""
-    message_lower = message.lower()
-    
-    # Simple greetings
-    if any(word in message_lower for word in ['hello', 'hi', 'hey']):
-        return f"""Hello {name}! I'm Dr. Alex. How can I assist you today?
+Common conditions I can advise on:
+{common_conditions}
 
-Please remember:
-- I can provide general health information
-- For emergencies, contact local emergency services
-- Always consult your personal physician for specific medical advice"""
-    
-    elif any(word in message_lower for word in ['thank', 'thanks']):
-        return f"You're very welcome, {name}! Wishing you good health. Is there anything else you'd like to discuss today?"
-    
-    # Medical responses
-    return generate_medical_response(name, message)
+For immediate concerns:
+• Contact your local healthcare provider
+• In emergencies, call your local emergency number
+
+Please try asking about one of these common conditions or consult a healthcare professional.
+
+Best regards,
+Dr. Alex"""
 
 # Gradio Interface
-with gr.Blocks(title="Dr. Alex - Personal Medical Consultant", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 👨‍⚕️ Dr. Alex - Your Personal Medical Consultant")
+with gr.Blocks(title="Dr. Alex Medical Advisor", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🩺 Dr. Alex - General Health Advisor")
+    name = gr.Textbox(label="Your Name", placeholder="Enter your name")
     
-    # Get user's name for personalization
-    name = gr.Textbox(label="Your Name", placeholder="Please enter your name for personalized care")
+    with gr.Tab("Health Consultation"):
+        gr.Markdown("## Describe your health concern")
+        msg = gr.Textbox(label="Symptoms/Question", lines=3)
+        submit_btn = gr.Button("Get Medical Advice")
+        output = gr.Textbox(label="Dr. Alex's Advice", lines=15, interactive=False)
     
-    with gr.Tabs():
-        with gr.TabItem("Consult Dr. Alex"):
-            gr.Markdown("## Describe your health concerns")
-            chatbot = gr.ChatInterface(
-                fn=lambda msg, hist: chatbot_response(msg, hist, name.value),
-                examples=[
-                    "I have fever and body aches since yesterday",
-                    "Best home remedies for sore throat",
-                    "How to manage mild back pain?"
-                ],
-                title="Private Consultation with Dr. Alex"
-            )
-        
-        with gr.TabItem("Structured Symptom Checker"):
-            gr.Markdown(f"## {name.value}, let's thoroughly evaluate your symptoms")
-            with gr.Row():
-                with gr.Column():
-                    age = gr.Number(label="Your Age", minimum=1, maximum=120)
-                    gender = gr.Radio(choices=["Male", "Female", "Other"], label="Gender")
-                    symptoms = gr.Textbox(label="Describe your symptoms in detail", lines=3, 
-                                       placeholder="Include location, nature of pain, triggers, etc.")
-                    duration = gr.Dropdown(
-                        choices=["Less than 24 hours", "1-3 days", "4-7 days", "1-2 weeks", "2-4 weeks", "More than 1 month"],
-                        label="Duration of symptoms"
-                    )
-                    severity = gr.Slider(1, 10, step=1, label="Pain/Severity level (1-10)")
-                    aggravating_factors = gr.Textbox(label="What makes it worse?", lines=2)
-                    relieving_factors = gr.Textbox(label="What provides relief?", lines=2)
-                    analyze_btn = gr.Button("Get Detailed Medical Advice", variant="primary")
-                
-                with gr.Column():
-                    analysis_output = gr.Textbox(label="Dr. Alex's Professional Evaluation", lines=15, interactive=False)
-            
-            analyze_btn.click(
-                lambda age, gender, symptoms, duration, severity, aggravating, relieving: generate_medical_response(
-                    name.value,
-                    f"As a {age}-year-old {gender}, I've had: {symptoms} for {duration}. Severity: {severity}/10. Gets worse with {aggravating} and better with {relieving}."
-                ),
-                inputs=[age, gender, symptoms, duration, severity, aggravating_factors, relieving_factors],
-                outputs=analysis_output
+    submit_btn.click(
+        fn=lambda name, msg: generate_response(name, msg),
+        inputs=[name, msg],
+        outputs=output
+    )
+    
+    gr.Markdown("### Common Health Topics")
+    with gr.Row():
+        for condition in MEDICAL_KNOWLEDGE:
+            gr.Button(condition.capitalize()).click(
+                fn=lambda name, cond=condition: generate_response(name, cond),
+                inputs=[name],
+                outputs=output
             )
 
-    gr.Markdown("""
-    <div style='background:#f8f9fa;padding:20px;border-radius:10px;margin-top:20px;'>
-    <h3>Important Disclaimer</h3>
-    <p>This AI provides general health information only, not medical advice. Always consult a qualified healthcare provider for personal medical concerns. In emergencies, contact your local emergency services immediately.</p>
-    </div>
-    """)
-
-# Launch app
 demo.launch()