Update app.py

app.py

@@ -67,12 +67,12 @@ def load_data():
         'GERD': 2, 
         'Chronic cholestasis': 3, 
         'Drug Reaction': 4,
-        'Peptic ulcer disease': 5,  # corrected spelling
+        'Peptic ulcer disease': 5,
         'AIDS': 6, 
-        'Diabetes': 7,  # corrected spelling (removed extra space)
+        'Diabetes': 7, 
         'Gastroenteritis': 8, 
         'Bronchial Asthma': 9,
-        'Hypertension': 10,  # corrected spelling (removed extra space)
+        'Hypertension': 10, 
         'Migraine': 11, 
         'Cervical spondylosis': 12, 
         'Paralysis (brain hemorrhage)': 13,
@@ -81,7 +81,7 @@ def load_data():
         'Chicken pox': 16, 
         'Dengue': 17, 
         'Typhoid': 18, 
-        'Hepatitis A': 19,  # corrected spelling (capitalized)
+        'Hepatitis A': 19,
         'Hepatitis B': 20, 
         'Hepatitis C': 21, 
         'Hepatitis D': 22, 
@@ -90,15 +90,15 @@ def load_data():
         'Tuberculosis': 25, 
         'Common Cold': 26, 
         'Pneumonia': 27, 
-        'Dimorphic hemorrhoids (piles)': 28,  # corrected spelling
+        'Dimorphic hemorrhoids (piles)': 28,
         'Heart attack': 29, 
         'Varicose veins': 30, 
         'Hypothyroidism': 31, 
         'Hyperthyroidism': 32,
         'Hypoglycemia': 33, 
-        'Osteoarthritis': 34,  # corrected spelling
+        'Osteoarthritis': 34, 
         'Arthritis': 35,
-        '(vertigo) Paroxysmal Positional Vertigo': 36,  # corrected spelling
+        '(vertigo) Paroxysmal Positional Vertigo': 36, 
         'Acne': 37, 
         'Urinary tract infection': 38,
         'Psoriasis': 39, 
@@ -318,102 +318,70 @@ def app_function(user_input, location, query, symptoms, history):
         disease_results
     )
 
-# CSS Styling
-custom_css = """
-body {
-    font-family: 'Roboto', sans-serif;
-    background-color: #3c6487; 
-    color: white;
-}
-h1 {
-    background: #ffffff;
-    color: #000000;
-    border-radius: 8px;
-    padding: 10px;
-    font-weight: bold;
-    text-align: center;
-    font-size: 2.5rem;
-}
-textarea, input {
-    background: transparent;
-    color: black;
-    border: 2px solid orange;
-    padding: 8px;
-    font-size: 1rem;
-    caret-color: black;
-    outline: none;
-    border-radius: 8px;
-}
-textarea:focus, input:focus {
-    background: transparent;
-    color: black;
-    border: 2px solid orange;
-    outline: none;
-}
-.df-container {
-    background: white;
-    color: black;
-    border: 2px solid orange;
-    border-radius: 10px;
-    padding: 10px;
-    font-size: 14px;
-    max-height: 400px;
-    height: auto;
-    overflow-y: auto;
-}
-#suggestions-title {
-    text-align: center !important;
-    font-weight: bold !important;
-    color: white !important; 
-    font-size: 4.2rem !important; 
-    margin-bottom: 20px !important; 
-}
-.gr-button {
-    background-color: #ae1c93; 
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1), 0 2px 4px rgba(0, 0, 0, 0.06);
-    transition: background-color 0.3s ease;
-}
-.gr-button:hover {
-    background-color: #8f167b;
-}
-.gr-button:active {
-    background-color: #7f156b;
-}
-"""
-
-# Gradio Application
-with gr.Blocks(css=custom_css) as app:
-    gr.HTML("<h1>🌟 Well-Being Companion</h1>")
+# Disease Prediction Interface
+def disease_app_function(name, symptom1, symptom2, symptom3, symptom4, symptom5):
+    if not name.strip():
+        return "Please enter the patient's name."
     
-    with gr.Row():
-        user_input = gr.Textbox(label="Please Enter Your Message Here", placeholder="Type your message...")
-        location = gr.Textbox(label="Your Current Location Here", placeholder="Enter location...")
-        query = gr.Textbox(label="Search Health Professionals Nearby", placeholder="What are you looking for...")
-        
-    with gr.Row():
-        symptom1 = gr.Dropdown(choices=["None"] + l1, label="Symptom 1")
-        symptom2 = gr.Dropdown(choices=["None"] + l1, label="Symptom 2")
-        symptom3 = gr.Dropdown(choices=["None"] + l1, label="Symptom 3")
-        symptom4 = gr.Dropdown(choices=["None"] + l1, label="Symptom 4")
-        symptom5 = gr.Dropdown(choices=["None"] + l1, label="Symptom 5")
-
-    submit = gr.Button(value="Submit", variant="primary")
-
-    chatbot = gr.Chatbot(label="Chat History")
-    sentiment = gr.Textbox(label="Detected Sentiment")
-    emotion = gr.Textbox(label="Detected Emotion")
+    symptoms_selected = [s for s in [symptom1, symptom2, symptom3, symptom4, symptom5] if s != "None"]
     
-    gr.Markdown("Suggestions", elem_id="suggestions-title")
+    if len(symptoms_selected) < 3:
+        return "Please select at least 3 symptoms for accurate prediction."
+
+    results = []
+    for model_name, (model, acc) in trained_models.items():
+        prediction = predict_disease(model, symptoms_selected)
+        result = f"{model_name} Prediction: Predicted Disease: **{prediction}**"
+        result += f" (Accuracy: {acc * 100:.2f}%)"
+        results.append(result)
     
-    suggestions = gr.DataFrame(headers=["Title", "Link"])  
-    professionals = gr.DataFrame(label="Nearby Health Professionals", headers=["Name", "Address"])  
-    map_html = gr.HTML(label="Interactive Map")
-    disease_predictions = gr.Textbox(label="Disease Predictions")  
-
-    submit.click(
-        app_function,
-        inputs=[user_input, location, query, [symptom1, symptom2, symptom3, symptom4, symptom5], chatbot],
-        outputs=[chatbot, sentiment, emotion, suggestions, professionals, map_html, disease_predictions],
-    )
+    return "\n\n".join(results)
 
+# Gradio Interface Setup
+with gr.Blocks() as app:
+    gr.HTML("<h1>🌟 Well-Being Companion</h1>")
+    
+    with gr.Tab("Mental Health Companion"):
+        with gr.Row():
+            user_input = gr.Textbox(label="Please Enter Your Message Here", placeholder="Type your message...")
+            location = gr.Textbox(label="Your Current Location Here", placeholder="Enter location...")
+            query = gr.Textbox(label="Search Health Professionals Nearby", placeholder="What are you looking for...")
+            
+        with gr.Row():
+            symptom1 = gr.Dropdown(choices=["None"] + l1, label="Symptom 1")
+            symptom2 = gr.Dropdown(choices=["None"] + l1, label="Symptom 2")
+            symptom3 = gr.Dropdown(choices=["None"] + l1, label="Symptom 3")
+            symptom4 = gr.Dropdown(choices=["None"] + l1, label="Symptom 4")
+            symptom5 = gr.Dropdown(choices=["None"] + l1, label="Symptom 5")
+
+        submit_chatbot = gr.Button(value="Submit", variant="primary")
+        chatbot = gr.Chatbot(label="Chat History")
+        sentiment = gr.Textbox(label="Detected Sentiment")
+        emotion = gr.Textbox(label="Detected Emotion")
+
+        suggestions = gr.DataFrame(headers=["Title", "Link"])  
+        professionals = gr.DataFrame(label="Nearby Health Professionals", headers=["Name", "Address"])  
+        map_html = gr.HTML(label="Interactive Map")
+        disease_predictions = gr.Textbox(label="Disease Predictions")  
+
+        submit_chatbot.click(
+            app_function,
+            inputs=[user_input, location, query, [symptom1, symptom2, symptom3, symptom4, symptom5], chatbot],
+            outputs=[chatbot, sentiment, emotion, suggestions, professionals, map_html, disease_predictions],
+        )
+
+    with gr.Tab("Disease Prediction"):
+        name_box = gr.Textbox(label="Name of Patient", placeholder="Enter patient's name")
+        symptom_choices = [gr.Dropdown(choices=["None"] + l1, label=f"Symptom {i+1}") for i in range(5)]
+        submit_disease = gr.Button(value="Submit", variant="primary")
+
+        disease_output = gr.Textbox(label="Predicted Disease", placeholder="Prediction will appear here")
+
+        submit_disease.click(
+            disease_app_function,
+            inputs=[name_box] + symptom_choices,
+            outputs=disease_output
+        )
+
+# Launch the Gradio application
 app.launch()