Update app.py

app.py

@@ -53,18 +53,30 @@ gmaps = googlemaps.Client(key=os.getenv("GOOGLE_API_KEY"))
 
 # Disease Prediction Code
 def load_data():
-    df = pd.read_csv("Training.csv")
-    tr = pd.read_csv("Testing.csv")
-    
+    try:
+        df = pd.read_csv("Training.csv")
+        tr = pd.read_csv("Testing.csv")
+    except FileNotFoundError as e:
+        raise RuntimeError("Data files not found. Please ensure `Training.csv` and `Testing.csv` are uploaded correctly.")
+
     disease_dict = {
-        # Disease encoding dictionary...
+        # Example disease encoding dictionary, update accordingly
+        'Fungal infection': 0, 'Allergy': 1, 'GERD': 2, 'Chronic cholestasis': 3, 'Drug Reaction': 4,
+        'Peptic ulcer diseae': 5, 'AIDS': 6, 'Diabetes': 7, 'Gastroenteritis': 8, 'Bronchial Asthma': 9,
+        'Hypertension': 10, 'Migraine': 11, 'Cervical spondylosis': 12, 'Paralysis': 13,
+        'Jaundice': 14, 'Malaria': 15, 'Chicken pox': 16, 'Dengue': 17, 'Typhoid': 18,
+        'Hepatitis A': 19, 'Hepatitis B': 20, 'Hepatitis C': 21, 'Hepatitis D': 22, 'Hepatitis E': 23,
+        'Alcoholic hepatitis': 24, 'Tuberculosis': 25, 'Common Cold': 26, 'Pneumonia': 27,
+        'Heart attack': 29, 'Varicose veins': 30, 'Hypothyroidism': 31, 'Hyperthyroidism': 32,
+        'Hypoglycemia': 33, 'Osteoarthritis': 34, 'Arthritis': 35
     }
+
     df.replace({'prognosis': disease_dict}, inplace=True)
     df = df.infer_objects(copy=False)
 
     tr.replace({'prognosis': disease_dict}, inplace=True)
     tr = tr.infer_objects(copy=False)
-    
+
     return df, tr, disease_dict
 
 df, tr, disease_dict = load_data()
@@ -161,7 +173,7 @@ def detect_emotion(user_input):
 def generate_suggestions(emotion):
     emotion_key = emotion.lower()
     suggestions = {
-        # Suggestions based on emotion...
+        # Replace with appropriate suggestions for each emotion
     }
     
     formatted_suggestions = [
@@ -275,12 +287,12 @@ textarea:focus, input:focus {
 # Gradio Application
 with gr.Blocks(css=custom_css) as app:
     gr.HTML("<h1>🌟 Well-Being Companion</h1>")
+    
     with gr.Row():
         user_input = gr.Textbox(label="Please Enter Your Message Here")
         location = gr.Textbox(label="Your Current Location Here")
         query = gr.Textbox(label="Search Health Professionals Nearby")
         
-    # New Row for Disease Prediction Symptoms
     with gr.Row():
         symptom1 = gr.Dropdown(choices=["None"] + l1, label="Symptom 1")
         symptom2 = gr.Dropdown(choices=["None"] + l1, label="Symptom 2")
@@ -294,7 +306,6 @@ with gr.Blocks(css=custom_css) as app:
     sentiment = gr.Textbox(label="Detected Sentiment")
     emotion = gr.Textbox(label="Detected Emotion")
     
-    # Suggestions Title
     gr.Markdown("Suggestions", elem_id="suggestions-title")
     
     suggestions = gr.DataFrame(headers=["Title", "Link"])  # Suggestions DataFrame