Update app.py

app.py

@@ -3,55 +3,27 @@ from smolagents import CodeAgent, HfApiModel
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Simple system prompt without any code-related instructions
+# 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 direct and concise lifestyle advice based on the user's details. "
     "Output a clear answer as plain text with no extra commentary."
 )
 
-class SimpleAgent(CodeAgent):
-    def run(self, user_input, **kwargs):
-        """
-        Override the run method to bypass code parsing completely
-        """
-        messages = [
-            {"role": "system", "content": self.description},
-            {"role": "user", "content": user_input}
-        ]
-        
-        response = self.model.complete(messages=messages, max_tokens=1024)
-        return self.clean_response(response)
-    
-    def clean_response(self, text):
-        """
-        Clean up the response text without any code parsing
-        """
-        # Remove potential error messages
-        if "Error in code parsing" in text:
-            text = text.split("Error in code parsing")[0]
-        
-        # Remove code block markers and step information
-        text = text.replace("```", "")
-        text = text.replace("<end_code>", "")
-        
-        # Remove step information
-        lines = []
-        for line in text.split('\n'):
-            if not any(x in line.lower() for x in ["step", "duration:", "input-tokens:", "output-tokens:"]):
-                lines.append(line)
-        
-        # Clean up and join
-        cleaned_lines = [line.strip() for line in lines if line.strip()]
-        return '\n\n'.join(cleaned_lines)
+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)
-    # Remove any code-related sections
-    for key in list(prompt_templates.keys()):
-        if 'code' in key.lower():
-            prompt_templates.pop(key)
 
 # Initialize the model with simplified settings
 model = HfApiModel(
@@ -60,8 +32,8 @@ model = HfApiModel(
     model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B'
 )
 
-# Initialize our custom agent instead of CodeAgent
-agent = SimpleAgent(
+# Initialize CodeAgent with minimal configuration
+agent = CodeAgent(
     model=model,
     tools=[FinalAnswerTool()],
     max_steps=1,
@@ -77,9 +49,9 @@ def run_agent(user_input):
     """
     try:
         response = agent.run(user_input)
-        return response
+        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()
+GradioUI(agent).launch()