Update app.py

app.py

@@ -1,4 +1,9 @@
+import os
 import streamlit as st
+from groq import Groq
+GROQ_API_KEY = "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"
+# Initialize Groq Client
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 # Disease and Treatment Information Data
 diseases_info = {
@@ -22,14 +27,18 @@ diseases_info = {
         "symptoms": ["shortness of breath", "wheezing", "chest tightness", "coughing"],
         "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.",
-    },
     "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.",
+    },
+    "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.",
+    },
     "anxiety": {
         "symptoms": ["excessive worry", "restlessness", "fatigue", "difficulty concentrating", "irritability"],
         "treatment": "Cognitive-behavioral therapy (CBT), medications like SSRIs, relaxation techniques.",
@@ -38,70 +47,68 @@ diseases_info = {
         "symptoms": ["persistent sadness", "loss of interest", "fatigue", "changes in appetite or sleep patterns"],
         "treatment": "Antidepressants, therapy (CBT), exercise, and social support.",
     },
-    "arthritis": {
-        "symptoms": ["joint pain", "swelling", "stiffness", "reduced range of motion"],
-        "treatment": "Pain relievers, anti-inflammatory drugs, physical therapy, and joint protection.",
-    },
-    "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.",
+    # Additional Diseases
+    "migraine": {
+        "symptoms": ["severe headache", "nausea", "sensitivity to light and sound"],
+        "treatment": "Prescription medications, rest in a dark room, and lifestyle adjustments.",
     },
-      "blood cancer": {
-        "symptoms": ["weightloss", "hiarfall", "weakness", "bedrest"],
-        "treatment": " over-the-counter medications, and antiviral drugs if prescribed.",
+    "eczema": {
+        "symptoms": ["itchy skin", "red rashes", "dry skin", "cracked or scaly patches"],
+        "treatment": "Moisturizers, corticosteroid creams, antihistamines, and avoiding irritants.",
     },
 }
 
-
 # Function to display disease info
 def display_disease_info(disease_name):
     disease_name = disease_name.lower().replace(" ", "_")
     if disease_name in diseases_info:
         disease = diseases_info[disease_name]
-        result = f"\n**Disease:** {disease_name.replace('_', ' ').title()}"
-        result += f"\n**Symptoms:** {', '.join(disease['symptoms'])}"
-        result += f"\n**Treatment:** {disease['treatment']}"
+        result = f"**Disease:** {disease_name.replace('_', ' ').title()}\n"
+        result += f"**Symptoms:** {', '.join(disease['symptoms'])}\n"
+        result += f"**Treatment:** {disease['treatment']}"
         return result
     else:
-        return "\nSorry, no information available for this disease."
+        return "Sorry, no information available for this disease."
 
-# Function to handle the chat bot
+# Function to interact with Groq API for Chat Bot
 def chat_bot(user_input):
-    if 'cha hal aa' in user_input.lower():
-        if 'are you healthcare app' in user_input.lower():
-            return "na rai mazak na kar maa app"
-        return "Khair aa, tu budha!"
-    elif 'exit' in user_input.lower():
-        return "Goodbye!"
-    else:
-        return "I'm not sure how to respond to that. Type 'exit' to end the chat."
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": user_input}],
+            model="llama-3.3-70b-versatile",
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        return f"Error: Unable to process your request. {str(e)}"
 
 # Streamlit App Interface
 def main():
     # Title of the app
-    st.title("Disease Information and Chat Bot")
-    
+    st.title("Health Guide Chat Bot")
+
     # Disease Input
     disease_name = st.text_input("Enter Disease Name", "")
     if disease_name:
         st.subheader("Disease Information")
         disease_info = display_disease_info(disease_name)
         st.text_area("Disease Info", disease_info, height=150)
-    
+
     # Chat Bot Section
     st.subheader("Chat with Bot")
-    user_input = st.text_input("Enter a Message (e.g., How are you?)", "")
+    user_input = st.text_input("Enter a Message for the Chat Bot", "")
     if user_input:
         response = chat_bot(user_input)
         st.text_area("Chat Bot Response", response, height=100)
-    
-    # Add some helpful instructions
-    st.markdown("""
+
+    # Add instructions
+    st.markdown(
+        """
         - **Enter a disease name** in the input box to get information about symptoms and treatments.
-        - **Ask the chat bot** something like "How are you?" to interact with it.
-        - Type "exit" to end the chat session.
-    """)
+        - **Chat with the bot** to get general health advice or answers to your questions.
+        """
+    )
 
 # Run the app
 if __name__ == "__main__":
     main()
+