Update app.py

app.py

@@ -1,81 +1,81 @@
-# 💡 Run this file with: streamlit run app.py
-#Install the required dependencies using the following command
-#pip install streamlit replicate duckduckgo-search
-
-
-import streamlit as st
-import replicate
-from duckduckgo_search import DDGS
-
-# ------------------- 🔐 Setup Replicate -------------------
-
-REPLICATE_API_TOKEN = "r8_0yAsouiplq8fxaxqVrqJ69tEF3CdgRD37L2Eq"  # <-- Replace this with your token
-client = replicate.Client(api_token=REPLICATE_API_TOKEN)
-GRANITE_MODEL_ID = "ibm-granite/granite-3.3-8b-instruct"
-
-# ------------------- 🔍 Optional Web Search -------------------
-
-def search_duckduckgo(query):
-    with DDGS() as ddgs:
-        results = ddgs.text(query, max_results=3)
-        return "\n".join([f"- {r['title']}: {r['href']}" for r in results])
-
-# ------------------- 🧠 LLM Query Function -------------------
-
-def query_replicate(prompt):
-    try:
-        output = client.run(GRANITE_MODEL_ID, input={"prompt": prompt, "max_new_tokens": 300})
-        return "".join(output)
-    except Exception as e:
-        return f"❌ Error from Granite: {str(e)}"
-
-# ------------------- 🤖 Agent Functions -------------------
-
-def workout_agent(prompt):
-    system_prompt = "You are a personal workout coach. Recommend home-based fitness routines based on user goals and preferences."
-    return query_replicate(f"{system_prompt}\n\nUser: {prompt}")
-
-def nutrition_agent(prompt):
-    system_prompt = "You are a friendly nutritionist. Suggest healthy, affordable meal ideas based on user input. Be simple and practical."
-    return query_replicate(f"{system_prompt}\n\nUser: {prompt}")
-
-def motivation_agent(prompt):
-    system_prompt = "You are a motivational fitness buddy. Share tips, inspiration, and strategies to stay consistent with healthy habits."
-    return query_replicate(f"{system_prompt}\n\nUser: {prompt}")
-
-# ------------------- 🧠 Smart Router -------------------
-
-def fitness_buddy_router(user_prompt):
-    workout_keywords = ["workout", "exercise", "routine", "stretch", "home gym", "training"]
-    nutrition_keywords = ["diet", "meal", "protein", "food", "nutrition", "eat", "snack"]
-    motivation_keywords = ["motivation", "lazy", "can't", "habit", "inspiration", "routine", "mental"]
-
-    prompt_lower = user_prompt.lower()
-
-    if any(word in prompt_lower for word in workout_keywords):
-        return workout_agent(user_prompt)
-    elif any(word in prompt_lower for word in nutrition_keywords):
-        return nutrition_agent(user_prompt)
-    elif any(word in prompt_lower for word in motivation_keywords):
-        return motivation_agent(user_prompt)
-    else:
-        return "🤖 Please clarify: are you asking about workouts, meals, or motivation?"
-
-# ------------------- 🎨 Streamlit UI -------------------
-
-st.set_page_config(page_title="Fitness Buddy", page_icon="💪")
-st.title("🏋️ Fitness Buddy AI")
-st.caption("Powered by IBM Granite via Replicate")
-
-st.markdown("**Ask me about:**\n- 🏋️ Home workouts\n- 🥗 Nutritious meal ideas\n- 💡 Fitness motivation and habit building")
-
-user_input = st.text_area("Ask Fitness Buddy:", height=150, placeholder="e.g., Suggest a workout for weight loss at home")
-
-if st.button("Ask AI"):
-    if user_input.strip():
-        with st.spinner("Thinking..."):
-            reply = fitness_buddy_router(user_input)
-        st.markdown("### 🤖 Fitness Buddy Says:")
-        st.write(reply)
-    else:
-        st.warning("Please enter a question or message first.")
+# 💡 Run this file with: streamlit run app.py
+#Install the required dependencies using the following command
+#pip install streamlit replicate duckduckgo-search
+
+
+import streamlit as st
+import replicate
+from duckduckgo_search import DDGS
+
+# ------------------- 🔐 Setup Replicate -------------------
+
+REPLICATE_API_TOKEN = "r8_fN4twLwm1FOIgMwxX0IpFmpVtUbPxEV3f1hNd"  # <-- Replace this with your token
+client = replicate.Client(api_token=REPLICATE_API_TOKEN)
+GRANITE_MODEL_ID = "ibm-granite/granite-3.3-8b-instruct"
+
+# ------------------- 🔍 Optional Web Search -------------------
+
+def search_duckduckgo(query):
+    with DDGS() as ddgs:
+        results = ddgs.text(query, max_results=3)
+        return "\n".join([f"- {r['title']}: {r['href']}" for r in results])
+
+# ------------------- 🧠 LLM Query Function -------------------
+
+def query_replicate(prompt):
+    try:
+        output = client.run(GRANITE_MODEL_ID, input={"prompt": prompt, "max_new_tokens": 300})
+        return "".join(output)
+    except Exception as e:
+        return f"❌ Error from Granite: {str(e)}"
+
+# ------------------- 🤖 Agent Functions -------------------
+
+def workout_agent(prompt):
+    system_prompt = "You are a personal workout coach. Recommend home-based fitness routines based on user goals and preferences."
+    return query_replicate(f"{system_prompt}\n\nUser: {prompt}")
+
+def nutrition_agent(prompt):
+    system_prompt = "You are a friendly nutritionist. Suggest healthy, affordable meal ideas based on user input. Be simple and practical."
+    return query_replicate(f"{system_prompt}\n\nUser: {prompt}")
+
+def motivation_agent(prompt):
+    system_prompt = "You are a motivational fitness buddy. Share tips, inspiration, and strategies to stay consistent with healthy habits."
+    return query_replicate(f"{system_prompt}\n\nUser: {prompt}")
+
+# ------------------- 🧠 Smart Router -------------------
+
+def fitness_buddy_router(user_prompt):
+    workout_keywords = ["workout", "exercise", "routine", "stretch", "home gym", "training"]
+    nutrition_keywords = ["diet", "meal", "protein", "food", "nutrition", "eat", "snack"]
+    motivation_keywords = ["motivation", "lazy", "can't", "habit", "inspiration", "routine", "mental"]
+
+    prompt_lower = user_prompt.lower()
+
+    if any(word in prompt_lower for word in workout_keywords):
+        return workout_agent(user_prompt)
+    elif any(word in prompt_lower for word in nutrition_keywords):
+        return nutrition_agent(user_prompt)
+    elif any(word in prompt_lower for word in motivation_keywords):
+        return motivation_agent(user_prompt)
+    else:
+        return "🤖 Please clarify: are you asking about workouts, meals, or motivation?"
+
+# ------------------- 🎨 Streamlit UI -------------------
+
+st.set_page_config(page_title="Fitness Buddy", page_icon="💪")
+st.title("🏋️ Fitness Buddy AI")
+st.caption("Powered by IBM Granite via Replicate")
+
+st.markdown("**Ask me about:**\n- 🏋️ Home workouts\n- 🥗 Nutritious meal ideas\n- 💡 Fitness motivation and habit building")
+
+user_input = st.text_area("Ask Fitness Buddy:", height=150, placeholder="e.g., Suggest a workout for weight loss at home")
+
+if st.button("Ask AI"):
+    if user_input.strip():
+        with st.spinner("Thinking..."):
+            reply = fitness_buddy_router(user_input)
+        st.markdown("### 🤖 Fitness Buddy Says:")
+        st.write(reply)
+    else:
+        st.warning("Please enter a question or message first.")