Update app.py

app.py

@@ -1,10 +1,10 @@
 import yaml
 import re
-from smolagents import HfApiModel
+from smolagents import CodeAgent, HfApiModel
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# 🚀 Updated system prompt: No code extraction allowed.
+# 🚀 Updated system prompt: No code recognition or extraction allowed.
 system_prompt = (
     "You are a health and lifestyle advisor specializing in the early detection and prevention of hypertension. "
     "Provide only the final, direct, and concise lifestyle advice based solely on the user's details. "
@@ -22,7 +22,7 @@ def remove_code_snippets(text):
 # ✅ Use only the final_answer tool.
 final_answer = FinalAnswerTool()
 
-# 🎯 Updated Model Setup
+# 🎯 Model setup
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -34,10 +34,10 @@ model = HfApiModel(
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# 🔄 Initialize GenericAgent instead of CodeAgent
-agent = GenericAgent(
+# 🔄 Initialize CodeAgent (but block code extraction)
+agent = CodeAgent(
     model=model,
-    tools=[final_answer],  # Restricted to final answer only.
+    tools=[final_answer],  # Restricted to final advice only.
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -51,7 +51,7 @@ agent = GenericAgent(
 def run_agent(user_input):
     """Runs the agent and ensures no code snippets are in the output."""
     raw_response = agent.run(user_input)
-    clean_response = remove_code_snippets(raw_response)
+    clean_response = remove_code_snippets(raw_response)  # Filter out code snippets
     return clean_response
 
 # 🚀 Launch the Gradio UI.