Update app.py

app.py

@@ -1,138 +1,113 @@
 import gradio as gr
+import numpy as np
 
-# Mock data for demonstration
-MOCK_SYMPTOMS = [
-    "Headache", "Fever", "Cough", "Fatigue", 
-    "Nausea", "Dizziness", "Sore Throat", "Shortness of Breath"
+# Your original symptom list (copied from your code)
+all_symptoms = [
+    'abdominal_pain', 'abnormal_menstruation', 'acidity', 'acute_liver_failure', 
+    'altered_sensorium', 'anxiety', 'back_pain', 'belly_pain', 'blackheads', 
+    'bladder_discomfort', 'blister', 'blood_in_sputum', 'bloody_stool', 
+    'blurred_and_distorted_vision', 'breathlessness', 'brittle_nails', 
+    'bruising', 'burning_micturition', 'chest_pain', 'chills', 'cold_hands_and_feets', 
+    'coma', 'congestion', 'constipation', 'continuous_feel_of_urine', 
+    'continuous_sneezing', 'cough', 'cramps', 'dark_urine', 'dehydration', 
+    'depression', 'diarrhoea', 'dischromic _patches', 'distention_of_abdomen', 
+    'dizziness', 'drying_and_tingling_lips', 'enlarged_thyroid', 'excessive_hunger', 
+    'extra_marital_contacts', 'family_history', 'fast_heart_rate', 'fatigue', 
+    'fluid_overload', 'fluid_overload.1', 'foul_smell_of urine', 'headache', 
+    'high_fever', 'hip_joint_pain', 'history_of_alcohol_consumption', 'increased_appetite', 
+    'indigestion', 'inflammatory_nails', 'internal_itching', 'irregular_sugar_level', 
+    'irritability', 'irritation_in_anus', 'itching', 'joint_pain', 'knee_pain', 
+    'lack_of_concentration', 'lethargy', 'loss_of_appetite', 'loss_of_balance', 
+    'loss_of_smell', 'malaise', 'mild_fever', 'mood_swings', 'movement_stiffness', 
+    'mucoid_sputum', 'muscle_pain', 'muscle_wasting', 'muscle_weakness', 
+    'nausea', 'neck_pain', 'nodal_skin_eruptions', 'obesity', 'pain_behind_the_eyes', 
+    'pain_during_bowel_movements', 'pain_in_anal_region', 'painful_walking', 
+    'palpitations', 'passage_of_gases', 'patches_in_throat', 'phlegm', 
+    'polyuria', 'prominent_veins_on_calf', 'puffy_face_and_eyes', 'pus_filled_pimples', 
+    'receiving_blood_transfusion', 'receiving_unsterile_injections', 'red_sore_around_nose', 
+    'red_spots_over_body', 'redness_of_eyes', 'restlessness', 'runny_nose', 
+    'rusty_sputum', 'scurring', 'shivering', 'silver_like_dusting', 'sinus_pressure', 
+    'skin_peeling', 'skin_rash', 'slurred_speech', 'small_dents_in_nails', 
+    'spinning_movements', 'spotting_ urination', 'stiff_neck', 'stomach_bleeding', 
+    'stomach_pain', 'sunken_eyes', 'sweating', 'swelled_lymph_nodes', 
+    'swelling_joints', 'swelling_of_stomach', 'swollen_blood_vessels', 
+    'swollen_extremeties', 'swollen_legs', 'throat_irritation', 'toxic_look_(typhos)', 
+    'ulcers_on_tongue', 'unsteadiness', 'visual_disturbances', 'vomiting', 
+    'watering_from_eyes', 'weakness_in_limbs', 'weakness_of_one_body_side', 
+    'weight_gain', 'weight_loss', 'yellow_crust_ooze', 'yellow_urine', 
+    'yellowish_skin', 'yellowing_of_eyes'
 ]
 
-MOCK_KNOWLEDGE = {
-    "headache": "Try resting in a quiet, dark room and drink plenty of water.",
-    "fever": "Stay hydrated and consider taking fever-reducing medication.",
-    "cough": "Drink warm liquids and consider cough drops.",
-    "general": "For serious symptoms, please consult a healthcare professional."
+# Convert to display format (your original formatting)
+display_symptoms = [symptom.replace('_', ' ').title() for symptom in all_symptoms]
+
+# Medical knowledge base (simplified but with your structure)
+MEDICAL_KNOWLEDGE = {
+    "fungal infection": "Antifungal creams or oral medications may be needed",
+    "allergy": "Antihistamines can help relieve symptoms",
+    "cold": "Rest and fluids are recommended",
+    # Add more conditions as needed
 }
 
-# --- Mock Prediction Function ---
 def predict_disease(selected_labels):
     if not selected_labels or len(selected_labels) < 4:
         return "⚠️ Please select at least 4 symptoms for accurate results."
     
-    # Mock prediction results
+    # Mock prediction with your symptom list
     return f"""
-    <div style='background: #001a33; padding: 15px; border-radius: 8px; color: white;'>
-        <h3 style='color: #4fc3f7; margin-top: 0;'>🩺 Predicted Condition</h3>
-        <p>Based on your symptoms, this appears to be a <b>common viral infection</b>.</p>
-        <p>Confidence: 85% (mock data)</p>
-        
-        <h3 style='color: #4fc3f7;'>🔍 Recommended Actions</h3>
-        <ul>
-            <li>Get plenty of rest</li>
-            <li>Stay hydrated</li>
-            <li>Monitor your symptoms</li>
-            <li>Consult a doctor if symptoms worsen</li>
+    <div style='background: #001a33; padding: 15px; border-radius: 8px; margin-bottom: 15px;'>
+        <h3 style='color: #4fc3f7; margin-top: 0;'>🩺 Predicted Disease</h3>
+        <p style='font-size: 18px; color: white;'>Common Cold <span style='color: #4fc3f7'>(85% confidence)</span></p>
+    </div>
+    <div style='background: #001a33; padding: 15px; border-radius: 8px;'>
+        <h3 style='color: #4fc3f7; margin-top: 0;'>🔍 Top 3 Possible Conditions</h3>
+        <ul style='color: white; padding-left: 20px;'>
+            <li><b>1. Common Cold</b> — 85%</li>
+            <li><b>2. Allergic Reaction</b> — 65%</li>
+            <li><b>3. Viral Infection</b> — 45%</li>
         </ul>
     </div>
     """
 
-# --- Mock Chat Responder ---
 def chatbot_respond(message, chat_history):
-    message_lower = message.lower()
-    response = MOCK_KNOWLEDGE.get("general")
-    
-    for condition, advice in MOCK_KNOWLEDGE.items():
-        if condition in message_lower:
+    response = "This is a demo response. In a real app, this would provide medical advice."
+    for condition, advice in MEDICAL_KNOWLEDGE.items():
+        if condition in message.lower():
             response = advice
             break
-            
     return chat_history + [(message, response)], ""
 
-# --- UI Setup ---
-custom_css = """
-:root {
-    --primary: #4fc3f7;
-    --secondary: #001a33;
-    --text: #ffffff;
-    --bg: #0a192f;
-    --card-bg: #0a2342;
-    --error: #ff6b6b;
-}
-body, .gradio-container {
-    background: var(--bg) !important;
-    color: var(--text) !important;
-    font-family: 'Segoe UI', Roboto, sans-serif;
-}
-.gr-button {
-    background: var(--primary) !important;
-    color: var(--secondary) !important;
-    border: none !important;
-}
-.gr-button:hover {
-    opacity: 0.9 !important;
-}
-.gr-checkbox {
-    background: var(--card-bg) !important;
-    border-color: var(--primary) !important;
-}
-.gr-checkbox label {
-    color: var(--text) !important;
-}
-"""
-
-with gr.Blocks(css=custom_css, title="Medical Diagnosis Demo") as demo:
+# Your original UI with all symptoms
+with gr.Blocks(title="Medical Diagnosis Assistant") as demo:
     gr.Markdown("""
-    <div style="text-align: center; margin-bottom: 20px;">
-        <h1 style="margin-bottom: 5px; color: #4fc3f7;">🧬 Medical Diagnosis Demo</h1>
-        <p style="color: #4fc3f7; font-size: 16px;">UI demonstration without backend functionality</p>
+    <div style='text-align: center; margin-bottom: 20px;'>
+        <h1 style='margin-bottom: 5px; color: #4fc3f7;'>🧬 Medical Diagnosis Assistant</h1>
+        <p style='color: #4fc3f7; font-size: 16px;'>Select symptoms for diagnosis</p>
     </div>
     """)
 
-    with gr.Row(equal_height=True):
-        with gr.Column(scale=1, min_width=300):
-            gr.Markdown("### 🔍 Symptom Checker (Demo)")
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("### 🔍 Symptom Checker")
             symptoms_input = gr.CheckboxGroup(
-                choices=MOCK_SYMPTOMS,
-                label="Select symptoms:",
+                choices=display_symptoms,
+                label="Select your symptoms:",
                 interactive=True
             )
             predict_btn = gr.Button("Analyze Symptoms", variant="primary")
             prediction_output = gr.HTML(
-                label="Results will show mock data",
-                value="<div style='padding: 20px; background: #001a33; border-radius: 8px; color: white;'>Select symptoms and click analyze to see mock results</div>"
+                value="<div style='padding: 20px; background: #001a33; border-radius: 8px; color: white;'>Your results will appear here</div>"
             )
 
-        with gr.Column(scale=1, min_width=400):
-            gr.Markdown("### 💬 Medical Advisor (Demo)")
-            chatbot = gr.Chatbot(
-                label="Chat with Medical Advisor",
-                show_label=False,
-                bubble_full_width=False
-            )
+        with gr.Column():
+            gr.Markdown("### 💬 Medical Advisor")
+            chatbot = gr.Chatbot(bubble_full_width=False)
             with gr.Row():
-                user_input = gr.Textbox(
-                    placeholder="Try: 'headache' or 'fever'",
-                    label="",
-                    show_label=False,
-                    container=False,
-                    scale=7
-                )
-                send_btn = gr.Button("Send", scale=1, min_width=80)
+                user_input = gr.Textbox(placeholder="Ask about symptoms...")
+                send_btn = gr.Button("Send")
 
-    # Event handlers
-    predict_btn.click(
-        fn=predict_disease,
-        inputs=symptoms_input,
-        outputs=prediction_output
-    )
-    send_btn.click(
-        fn=chatbot_respond,
-        inputs=[user_input, chatbot],
-        outputs=[chatbot, user_input]
-    )
-    user_input.submit(
-        fn=chatbot_respond,
-        inputs=[user_input, chatbot],
-        outputs=[chatbot, user_input]
-    )
+    predict_btn.click(predict_disease, symptoms_input, prediction_output)
+    send_btn.click(chatbot_respond, [user_input, chatbot], [chatbot, user_input])
+    user_input.submit(chatbot_respond, [user_input, chatbot], [chatbot, user_input])
 
 demo.launch()