Update app.py

app.py

@@ -3,27 +3,55 @@ from smolagents import CodeAgent, HfApiModel
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Simplified system prompt without any code-related instructions
+# Simple 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."
 )
 
-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)
+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)
 
 # 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(
@@ -32,8 +60,8 @@ model = HfApiModel(
     model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B'
 )
 
-# Initialize CodeAgent with minimal configuration
-agent = CodeAgent(
+# Initialize our custom agent instead of CodeAgent
+agent = SimpleAgent(
     model=model,
     tools=[FinalAnswerTool()],
     max_steps=1,
@@ -49,9 +77,9 @@ def run_agent(user_input):
     """
     try:
         response = agent.run(user_input)
-        return clean_response(response)
+        return 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()