modified app

app.py

@@ -1,8 +1,11 @@
 import streamlit as st
 from groq import Groq
 
-# Set up the Groq API client (Replace with your valid API key)
-client = Groq(api_key="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")  # Replace with a secure method for storing API keys
+# Hardcoded API Key (Replace with a valid key)
+API_KEY = "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"
+
+# Initialize Groq client
+client = Groq(api_key=API_KEY)
 
 # Streamlit UI Setup
 st.title("🥗 Diet and Nutrition Chatbot")
@@ -11,7 +14,14 @@ st.write("Welcome! I’m here to help you with diet tips, meal planning, and nut
 # Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = [
-        {"role": "system", "content": "You are a helpful Diet and Nutrition Chatbot. Provide meal plans, healthy eating tips, information about food benefits, and encourage balanced diets. Tailor your suggestions to different dietary preferences when asked."}
+        {
+            "role": "system",
+            "content": (
+                "You are a highly knowledgeable chatbot specializing in diet and nutrition. "
+                "Answer only questions related to nutrition, meal planning, food benefits, and healthy eating. "
+                "Do not respond to questions outside of diet and nutrition. If asked an unrelated question, politely ask the user to focus on diet and nutrition topics."
+            )
+        }
     ]
 
 # Display chat history
@@ -24,29 +34,30 @@ for message in st.session_state.messages:
 user_input = st.chat_input("Ask me about healthy eating, diets, or meal planning...")
 
 if user_input:
-    # Validate the input: check if the question is related to diet or nutrition
-    keywords = ["diet", "nutrition", "meal", "food", "healthy", "calories", "protein", "vegetarian", "vegan", "keto", "weight"]
-    if any(keyword in user_input.lower() for keyword in keywords):
-        # Append user message
-        st.session_state.messages.append({"role": "user", "content": user_input})
-        with st.chat_message("user"):
-            st.write(user_input)
+    # Append user message
+    st.session_state.messages.append({"role": "user", "content": user_input})
+    with st.chat_message("user"):
+        st.write(user_input)
 
+    try:
         # Get AI Response
-        try:
-            response = client.chat.completions.create(
-                model="llama-3.3-70b-versatile",  # Adjust the model as per API availability
-                messages=st.session_state.messages
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",  # Adjust the model as per API availability
+            messages=st.session_state.messages
+        )
+        
+        bot_reply = response.choices[0].message.content
+
+        # Redirect if the AI response is off-topic
+        if "diet" not in bot_reply.lower() and "nutrition" not in bot_reply.lower():
+            bot_reply = (
+                "❗ I specialize in diet and nutrition. Please ask about meal plans, healthy eating, or food-related topics."
             )
-            bot_reply = response.choices[0].message.content
-        except Exception as e:
-            bot_reply = "Sorry, I'm having trouble processing your request right now. Please try again later."
-
-        # Append AI Response
-        st.session_state.messages.append({"role": "assistant", "content": bot_reply})
-        with st.chat_message("assistant"):
-            st.write(bot_reply)
-    else:
-        # Notify the user that their question is not related to the topic
-        with st.chat_message("assistant"):
-            st.write("❗ Please ask questions related to diet, nutrition, or healthy eating.")
+
+    except Exception as e:
+        bot_reply = "Sorry, I'm having trouble processing your request right now. Please try again later."
+
+    # Append AI Response
+    st.session_state.messages.append({"role": "assistant", "content": bot_reply})
+    with st.chat_message("assistant"):
+        st.write(bot_reply)