Update app.py

app.py

@@ -1,106 +1,57 @@
 import yaml
-import re
 from smolagents import CodeAgent, HfApiModel
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Updated system prompt: Only output the final, direct advice in plain text.
+# Simplified system prompt without any code-related instructions
 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. "
-    "Do NOT include any internal reasoning, chain-of-thought, intermediate steps, or code snippets. "
-    "Output exactly one final answer as plain text with no extra commentary."
+    "Provide direct and concise lifestyle advice based on the user's details. "
+    "Output a clear answer as plain text with no extra commentary."
 )
 
-def remove_code_snippets(text):
+def clean_response(text):
     """
-    Removes code blocks, inline code, chain-of-thought, error messages, and debugging/step logs from the output.
+    Simple cleanup function that removes extra whitespace and ensures proper formatting.
     """
-    # Remove triple-backtick code blocks.
-    text = re.sub(r"```[\s\S]+?```", "", text, flags=re.DOTALL)
-    # Remove inline code enclosed in single backticks.
-    text = re.sub(r"`[^`]+`", "", text)
-    # Remove any text between <think> and </think> tags.
-    text = re.sub(r"<think>[\s\S]*?</think>", "", text, flags=re.DOTALL)
-    # Remove block from "Make sure to include code with the correct pattern" until "<end_code>"
-    text = re.sub(r"Make sure to include code with the correct pattern[\s\S]*?<end_code>", "", text, flags=re.DOTALL)
-    
-    # Split the text into lines and filter out debug/error/step log lines.
-    lines = text.splitlines()
-    filtered_lines = []
-    for line in lines:
-        # Skip lines that are clearly debug or error messages.
-        if re.search(r"Step \d+", line):
-            continue
-        if "Duration:" in line:
-            continue
-        if "Error in code parsing:" in line:
-            continue
-        if "💥 Error" in line:
-            continue
-        if "Reached max steps" in line:
-            continue
-        if "Make sure to provide correct code blobs" in line:
-            continue
-        if "Here is your code snippet:" in line:
-            continue
-        if "Code:" in line:
-            continue
-        filtered_lines.append(line)
-    
-    cleaned_text = "\n".join(filtered_lines)
-    return cleaned_text.strip()
-
-# Use only the final_answer tool.
-final_answer = FinalAnswerTool()
+    # Remove extra whitespace
+    text = ' '.join(text.split())
+    # Split into paragraphs for readability
+    paragraphs = text.split('\n\n')
+    cleaned_paragraphs = [p.strip() for p in paragraphs if p.strip()]
+    return '\n\n'.join(cleaned_paragraphs)
+
+# Load prompt templates from YAML
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
 
-# Set up the model with a reduced token limit.
+# Initialize the model with simplified settings
 model = HfApiModel(
     max_tokens=1024,
     temperature=0.5,
-    model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
-    custom_role_conversions=None,
+    model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B'
 )
 
-# Load prompt templates from YAML.
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-# Ensure the final_answer key exists in prompt_templates to prevent KeyError.
-if "final_answer" not in prompt_templates:
-    prompt_templates["final_answer"] = {"pre_messages": "", "post_messages": ""}
-
-# Initialize CodeAgent with a low verbosity level to reduce extra debug output.
+# Initialize CodeAgent with minimal configuration
 agent = CodeAgent(
     model=model,
-    tools=[final_answer],
+    tools=[FinalAnswerTool()],
     max_steps=1,
     verbosity_level=0,
-    grammar=None,
-    planning_interval=None,
-    name="Hypertension Prevention Advisor",
+    name="Health Advisor",
     description=system_prompt,
     prompt_templates=prompt_templates
 )
 
 def run_agent(user_input):
     """
-    Runs the agent and then removes any internal chain-of-thought, error messages,
-    debug logs, and code snippet sections before returning the final plain-text answer.
+    Runs the agent and returns a clean, formatted response.
     """
-    raw_response = agent.run(user_input)
-    print("Raw Agent Response:", raw_response)  # Debugging output (optional)
-    
-    if not raw_response.strip():
-        return "I'm sorry, but I couldn't generate a response. Please try again."
-    if "final_answer" not in raw_response.lower():
-        return "Error: The response did not use the `final_answer` tool. Please try again."
-    
-    clean_response = remove_code_snippets(raw_response)
-    words = clean_response.split()
-    if len(set(words)) < 5:
-        return "I'm unable to generate a meaningful response. Please refine your query."
-    return clean_response
-
-# Launch the Gradio UI.
-GradioUI(agent).launch()
+    try:
+        response = agent.run(user_input)
+        return clean_response(response)
+    except Exception as e:
+        return f"I apologize, but I couldn't process your request. Please try again."
+
+# Launch the Gradio UI
+GradioUI(agent).launch()