Update app.py

app.py

@@ -1,58 +1,45 @@
-import yaml
 from smolagents import CodeAgent, HfApiModel
+import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Simplified system prompt without any code-related instructions
+# Updated system prompt: provide only final, direct advice with no chain-of-thought or code.
 system_prompt = (
     "You are a health and lifestyle advisor specializing in the early detection and prevention of hypertension. "
-    "Provide direct and concise lifestyle advice based on the user's details. "
-    "Output a clear answer as plain text with no extra commentary."
+    "Diagnostic criteria: Normal BP is < 120/80 mmHg, Borderline BP is 120-139/80-89 mmHg, and Hypertension is > 140/90 mmHg. "
+    "Based solely on the user's details, provide only the final, direct, and concise lifestyle tips. "
+    "Do NOT include any internal reasoning, chain-of-thought, or any code snippets in your output. "
+    "Only output the final advice as plain text. For example, if the user mentions alcohol consumption, simply say: "
+    "'Reduce alcohol intake, as it can raise blood pressure.'"
 )
 
-def clean_response(text):
-    """
-    Simple cleanup function that removes extra whitespace and ensures proper formatting.
-    """
-    # 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)
+# Use only the final_answer tool.
+final_answer = FinalAnswerTool()
 
-# Initialize the model with simplified settings
+# Set up your model.
 model = HfApiModel(
-    max_tokens=1024,
+    max_tokens=2096,
     temperature=0.5,
-    model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B/',
-    token='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B/'
+    model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',  # Adjust if needed.
+    custom_role_conversions=None,
 )
 
-# Initialize CodeAgent with minimal configuration
+# Load prompt templates from the YAML file.
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+
+# Initialize the CodeAgent with the updated system prompt.
 agent = CodeAgent(
     model=model,
-    tools=[FinalAnswerTool()],
-    max_steps=1,
-    verbosity_level=0,
-    name="Health Advisor",
+    tools=[final_answer],
+    max_steps=6,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name="Hypertension Prevention Advisor",
     description=system_prompt,
     prompt_templates=prompt_templates
 )
 
-def run_agent(user_input):
-    """
-    Runs the agent and returns a clean, formatted response.
-    """
-    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
+# Launch the Gradio UI.
 GradioUI(agent).launch()