Update app.py

app.py

@@ -16,11 +16,15 @@ if not api_key:
 client = Groq(api_key=api_key)
 
 # Function to get chatbot response
-def get_health_response(user_question):
-    # Check if the question is related to health, fitness, or diet
-    health_keywords = ["health", "fitness", "exercise", "diet", "nutrition", "weight loss", "workout", "calories", "meal plan"]
+def get_fitness_response(user_question):
+    # Define keywords related to fitness, weight loss, and health
+    fitness_keywords = [
+        "health", "fitness", "exercise", "diet", "nutrition", "weight loss", "workout", "calories", 
+        "meal plan", "cardio", "strength training", "fat loss", "BMI", "hydration", "metabolism", 
+        "protein intake", "muscle gain", "recovery", "supplements", "healthy habits", "intermittent fasting"
+    ]
     
-    if any(keyword in user_question.lower() for keyword in health_keywords):
+    if any(keyword in user_question.lower() for keyword in fitness_keywords):
         # Call Groq API to generate a response
         completion = client.chat.completions.create(
             model="deepseek-r1-distill-llama-70b",
@@ -32,16 +36,16 @@ def get_health_response(user_question):
         )
         return completion.choices[0].message['content']
     else:
-        return "⚠️ This chatbot only answers health, fitness, and diet-related questions."
+        return "⚠️ This chatbot only answers fitness, health, and weight loss-related questions."
 
 # Streamlit UI
 st.set_page_config(page_title="Health & Fitness Chatbot", layout="centered")
 st.title("💪 Health & Fitness Chatbot")
-st.write("Ask me about health, fitness, weight loss, and diet plans!")
+st.write("Ask me anything about fitness, weight loss, workouts, and nutrition!")
 
 # User Input
 user_input = st.text_input("Type your question:")
 
 if user_input:
-    response = get_health_response(user_input)
+    response = get_fitness_response(user_input)
     st.write(response)