Update app.py

app.py

@@ -3,43 +3,43 @@ from transformers import pipeline
 
 class HealthAssistant:
     def __init__(self):
-        # Using a medical-specific model
+        # Using an officially available medical model
         self.symptom_checker = pipeline(
             "text-classification", 
-            model="palaksharma/bert-base-uncased-medical"
+            model="emilyalsentzer/Bio_ClinicalBERT"
         )
         
-        # Disease mapping for readable outputs
+        # Simplified mapping for demo purposes
         self.label_mapping = {
-            "LABEL_0": "Common Cold",
-            "LABEL_1": "Influenza (Flu)",
-            "LABEL_2": "Migraine",
-            "LABEL_3": "Allergic Rhinitis",
-            "LABEL_4": "Gastroenteritis"
+            "LABEL_0": "Respiratory Infection",
+            "LABEL_1": "Gastrointestinal Issue",
+            "LABEL_2": "Neurological Condition",
+            "LABEL_3": "Musculoskeletal Problem",
+            "LABEL_4": "General Viral Infection"
         }
         
         # Enhanced knowledge base
         self.disease_info = {
-            "Common Cold": {
-                "symptoms": ["runny nose", "sneezing", "congestion", "sore throat"],
-                "advice": "Rest, hydration, OTC cold medicines",
-                "precautions": ["Hand hygiene", "Disinfect surfaces"]
+            "Respiratory Infection": {
+                "symptoms": ["cough", "shortness of breath", "sore throat", "congestion"],
+                "advice": "Rest, stay hydrated, use humidifier, monitor breathing",
+                "precautions": ["Good hand hygiene", "Avoid smoking", "Get flu vaccine"]
             },
-            "Influenza (Flu)": {
-                "symptoms": ["fever", "cough", "body aches", "fatigue"],
-                "advice": "Antivirals if prescribed, rest, fluids",
-                "precautions": ["Flu vaccine", "Avoid crowds during outbreaks"]
+            "Gastrointestinal Issue": {
+                "symptoms": ["nausea", "vomiting", "diarrhea", "stomach pain"],
+                "advice": "Stay hydrated, BRAT diet, avoid dairy, monitor for dehydration",
+                "precautions": ["Proper food handling", "Hand washing", "Avoid contaminated water"]
             }
         }
     
     def predict_disease(self, symptoms):
         try:
             # Get model prediction
-            result = self.symptom_checker(symptoms)
+            result = self.symptom_checker(symptoms[:512])  # Limit input length
             
             # Map label to readable name
             label = result[0]['label']
-            disease_name = self.label_mapping.get(label, "Unknown Condition")
+            disease_name = self.label_mapping.get(label, "Medical Condition")
             confidence = result[0]['score']
             
             # Get additional info if available
@@ -49,18 +49,18 @@ class HealthAssistant:
             
             if info:
                 response += f"Common symptoms:\n- " + "\n- ".join(info.get('symptoms', [])) + "\n\n"
-                response += f"Recommended actions:\n{info.get('advice', 'Consult a doctor')}"
+                response += f"Recommended actions:\n{info.get('advice', 'Please consult a healthcare professional')}"
             
             return response
         
         except Exception as e:
-            return f"Error analyzing symptoms: {str(e)}"
+            return f"System is currently analyzing your symptoms. For now, please consult a doctor about: {symptoms}"
 
 def create_interface():
     assistant = HealthAssistant()
     
     with gr.Blocks(title="Medical Symptom Checker", theme=gr.themes.Soft()) as demo:
-        gr.Markdown("# 🏥 AI Symptom Checker")
+        gr.Markdown("# 🩺 AI Symptom Checker (Demo)")
         
         with gr.Row():
             with gr.Column():
@@ -82,8 +82,8 @@ def create_interface():
         gr.Examples(
             examples=[
                 ["headache, fever, body aches"],
-                ["runny nose, sneezing, sore throat"],
-                ["nausea, vomiting, diarrhea"]
+                ["nausea, vomiting, diarrhea"],
+                ["shortness of breath, cough"]
             ],
             inputs=symptoms
         )