Update app.py

app.py

@@ -1,135 +1,88 @@
-# Disease Information Dictionary (extended with more diseases)
-diseases = {
+# Disease and Treatment Information Data
+diseases_info = {
     "flu": {
         "symptoms": ["fever", "cough", "sore throat", "body aches", "fatigue"],
-        "treatment": "Rest, fluids, pain relievers, and anti-viral medications if prescribed."
+        "treatment": "Rest, fluids, pain relievers, and anti-viral medications if prescribed.",
     },
     "headache": {
         "symptoms": ["head pain", "nausea", "sensitivity to light"],
-        "treatment": "Rest, hydration, over-the-counter painkillers, and avoid triggers."
+        "treatment": "Rest, hydration, over-the-counter painkillers, and avoid triggers.",
     },
     "diabetes": {
         "symptoms": ["increased thirst", "frequent urination", "fatigue", "blurred vision"],
-        "treatment": "Insulin therapy, lifestyle changes, and dietary management."
+        "treatment": "Insulin therapy, lifestyle changes, and dietary management.",
+    },
+    "cold": {
+        "symptoms": ["sore throat", "runny nose", "cough", "sneezing", "mild body aches"],
+        "treatment": "Rest, hydration, decongestants, and over-the-counter pain relief.",
     },
     "asthma": {
         "symptoms": ["shortness of breath", "wheezing", "chest tightness", "coughing"],
-        "treatment": "Inhalers, bronchodilators, corticosteroids, and avoiding triggers."
+        "treatment": "Inhalers, bronchodilators, corticosteroids, and avoiding triggers.",
     },
     "pneumonia": {
         "symptoms": ["cough with mucus", "fever", "shortness of breath", "chest pain"],
-        "treatment": "Antibiotics for bacterial pneumonia, rest, fluids, and pain relievers."
-    },
-    "diabetes type 1": {
-        "symptoms": ["frequent urination", "increased thirst", "fatigue", "unexplained weight loss"],
-        "treatment": "Insulin injections, blood sugar monitoring, and a healthy diet."
-    },
-    "diabetes type 2": {
-        "symptoms": ["increased thirst", "frequent urination", "fatigue", "blurred vision"],
-        "treatment": "Lifestyle changes, oral medications, and insulin therapy if needed."
+        "treatment": "Antibiotics for bacterial pneumonia, rest, fluids, and pain relievers.",
     },
     "hypertension": {
         "symptoms": ["headaches", "dizziness", "shortness of breath", "nosebleeds"],
-        "treatment": "Medications like beta-blockers, ACE inhibitors, lifestyle changes including exercise and diet."
-    },
-    "arthritis": {
-        "symptoms": ["joint pain", "swelling", "stiffness", "reduced range of motion"],
-        "treatment": "Pain relievers, anti-inflammatory drugs, physical therapy, and joint protection."
-    },
-    "cancer": {
-        "symptoms": ["unexplained weight loss", "fatigue", "pain", "skin changes"],
-        "treatment": "Chemotherapy, radiation, surgery, and immunotherapy."
-    },
-    "alzheimer's disease": {
-        "symptoms": ["memory loss", "confusion", "difficulty performing familiar tasks"],
-        "treatment": "Cognitive therapy, medications to slow progression, and support for caregivers."
-    },
-    "migraine": {
-        "symptoms": ["severe headache", "nausea", "sensitivity to light and sound"],
-        "treatment": "Pain relievers, anti-nausea medications, and lifestyle adjustments."
+        "treatment": "Medications like beta-blockers, ACE inhibitors, lifestyle changes including exercise and diet.",
     },
-    "stroke": {
-        "symptoms": ["sudden numbness", "weakness", "confusion", "difficulty speaking or understanding speech"],
-        "treatment": "Emergency care, rehabilitation, and medications to prevent further strokes."
+    "anxiety": {
+        "symptoms": ["excessive worry", "restlessness", "fatigue", "difficulty concentrating", "irritability"],
+        "treatment": "Cognitive-behavioral therapy (CBT), medications like SSRIs, relaxation techniques.",
     },
-    # Add additional diseases here...
-    "abdominal aortic aneurysm": {
-        "symptoms": ["pulsating feeling near the navel", "severe back or abdominal pain"],
-        "treatment": "Surgery or endovascular aneurysm repair (EVAR)."
+    "depression": {
+        "symptoms": ["persistent sadness", "loss of interest", "fatigue", "changes in appetite or sleep patterns"],
+        "treatment": "Antidepressants, therapy (CBT), exercise, and social support.",
     },
-    "acute lymphoblastic leukemia": {
-        "symptoms": ["fatigue", "fever", "bone pain", "paleness"],
-        "treatment": "Chemotherapy, radiation, stem cell transplant."
-    },
-    "acute myeloid leukemia": {
-        "symptoms": ["fever", "fatigue", "easy bruising", "shortness of breath"],
-        "treatment": "Chemotherapy, stem cell transplant, and targeted therapy."
-    },
-    "acromegaly": {
-        "symptoms": ["enlarged hands and feet", "facial changes", "joint pain"],
-        "treatment": "Surgery, radiation therapy, and medications to control growth hormone."
-    },
-    "actinomycosis": {
-        "symptoms": ["painful lumps", "fever", "abscesses"],
-        "treatment": "Antibiotics, typically penicillin."
-    },
-    "addison's disease": {
-        "symptoms": ["fatigue", "low blood pressure", "weight loss", "skin darkening"],
-        "treatment": "Hormone replacement therapy, particularly corticosteroids."
+    "arthritis": {
+        "symptoms": ["joint pain", "swelling", "stiffness", "reduced range of motion"],
+        "treatment": "Pain relievers, anti-inflammatory drugs, physical therapy, and joint protection.",
     },
-    "adhd": {
-        "symptoms": ["difficulty focusing", "impulsiveness", "hyperactivity"],
-        "treatment": "Medications (stimulants), behavior therapy, and lifestyle changes."
+    "covid-19": {
+        "symptoms": ["fever", "dry cough", "fatigue", "loss of taste or smell", "shortness of breath"],
+        "treatment": "Rest, fluids, over-the-counter medications, and antiviral drugs if prescribed.",
     },
-    "aids": {
-        "symptoms": ["rapid weight loss", "fever", "night sweats", "fatigue"],
-        "treatment": "Antiretroviral therapy (ART), supportive care."
+    # Adding the larger list of diseases
+    "abdominal_aortic_aneurysm": {
+        "symptoms": ["pulsating feeling in the abdomen", "pain in the back or abdomen", "dizziness"],
+        "treatment": "Surgical repair or monitoring depending on severity.",
     },
-    "albinism": {
-        "symptoms": ["very light skin", "white or very light hair", "vision problems"],
-        "treatment": "No cure, but management includes sun protection and corrective lenses."
+    "acute_lymphoblastic_leukemia": {
+        "symptoms": ["fatigue", "fever", "pale skin", "easy bruising", "swollen lymph nodes"],
+        "treatment": "Chemotherapy, bone marrow transplant.",
     },
-    # You can continue adding diseases similarly from the list.
+    # Add more diseases here similarly...
 }
 
-# Function to display disease information
 def display_disease_info(disease_name):
-    disease_name = disease_name.lower()
-    if disease_name in diseases:
-        disease = diseases[disease_name]
-        print(f"\nDisease: {disease_name.title()}")
-        print("Symptoms:")
-        for symptom in disease["symptoms"]:
-            print(f"  - {symptom}")
-        print(f"\nTreatment: {disease['treatment']}")
+    """Display disease info if available."""
+    disease_name = disease_name.lower().replace(" ", "_")
+    if disease_name in diseases_info:
+        disease = diseases_info[disease_name]
+        print(f"Disease: {disease_name.replace('_', ' ').title()}")
+        print(f"Symptoms: {', '.join(disease['symptoms'])}")
+        print(f"Treatment: {disease['treatment']}")
     else:
-        print("Disease not found. Please check the name and try again.")
+        print("Sorry, no information available for this disease.")
 
-# Function to handle chat feature
 def chat_bot():
-    print("\nChat Bot is active. Type 'exit' to quit.\n")
+    """Chat bot functionality."""
+    print("Chat Bot: Hello! How can I assist you today?")
     while True:
-        user_input = input("You: ").strip().lower()
-        
-        # Responding to the "cha hal aa?" query with "khair aa"
-        if user_input == "cha hal aa?":
-            print("Bot: khair aa")
-        
-        # Check if the user wants to know about a disease
-        elif user_input in diseases:
-            display_disease_info(user_input)
+        user_input = input("You: ").lower()
         
-        # Exit the chat bot if the user types 'exit'
-        elif user_input == "exit":
-            print("Exiting chat...")
+        if 'how are you' in user_input:
+            print("Chat Bot: Khair aa, thank you for asking!")
+        elif 'exit' in user_input:
+            print("Chat Bot: Goodbye!")
             break
-        
-        # If the input is unrecognized
         else:
-            print("Bot: Sorry, I didn't understand that. Please ask about a disease or type 'exit' to quit.")
+            print("Chat Bot: I'm not sure how to respond to that. Type 'exit' to end the chat.")
 
-# Main menu to choose between disease info or chat
 def main():
+    """Main program function."""
     print("Welcome to the Disease Information and Chat Bot!\n")
     
     while True:
@@ -138,10 +91,14 @@ def main():
         print("2. Chat with Bot")
         print("3. Exit")
         
-        option = input("Enter your choice (1/2/3): ").strip()
+        try:
+            option = input("Enter your choice (1/2/3): ").strip()
+        except EOFError:
+            print("\nError: Input is required. Exiting...")
+            break
         
         if option == "1":
-            disease_name = input("Enter the name of the disease: ").strip().lower()
+            disease_name = input("Enter the name of the disease: ").strip()
             display_disease_info(disease_name)
         
         elif option == "2":
@@ -154,6 +111,5 @@ def main():
         else:
             print("Invalid choice. Please try again.")
 
-# Run the program
 if __name__ == "__main__":
     main()