Update app.py

app.py

@@ -16,15 +16,16 @@ system_prompt = (
 # 🛑 Function to remove any detected code snippets.
 def remove_code_snippets(text):
     """Removes potential code snippets from the output."""
-    pattern = r"```.*?```|`[^`]+`"  # Matches both block and inline code
-    return re.sub(pattern, "", text, flags=re.DOTALL).strip()
+    pattern = r"```[\s\S]+?```|`[^`]+`|(?:\bimport\b|\bdef\b|\bclass\b|\breturn\b|\bprint\b|\bif\b|\belse\b|\btry\b|\bexcept\b|\bwhile\b|\bfor\b)"
+    clean_text = re.sub(pattern, "", text, flags=re.DOTALL | re.IGNORECASE).strip()
+    return clean_text if clean_text else "Invalid response generated. Please try again."
 
 # ✅ Use only the final_answer tool.
 final_answer = FinalAnswerTool()
 
-# 🎯 Model setup
+# 🎯 Model setup with reduced token limit to prevent long outputs
 model = HfApiModel(
-    max_tokens=2096,
+    max_tokens=1024,  # Reduced from 2096 to prevent long responses
     temperature=0.5,
     model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
     custom_role_conversions=None,
@@ -34,11 +35,11 @@ model = HfApiModel(
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# 🔄 Initialize CodeAgent (but block code extraction)
+# 🔄 Initialize CodeAgent (blocking code extraction)
 agent = CodeAgent(
     model=model,
     tools=[final_answer],  # Restricted to final advice only.
-    max_steps=6,
+    max_steps=4,  # Reduced from 6 to minimize infinite loops
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
@@ -49,9 +50,28 @@ agent = CodeAgent(
 
 # 🛑 Run agent with filtering.
 def run_agent(user_input):
-    """Runs the agent and ensures no code snippets are in the output."""
+    """Runs the agent, ensuring valid output and blocking infinite loops."""
     raw_response = agent.run(user_input)
-    clean_response = remove_code_snippets(raw_response)  # Filter out code snippets
+
+    # Debugging: Print raw response
+    print("Raw Agent Response:", raw_response)
+
+    # Prevent infinite loops and blank responses
+    if not raw_response.strip():
+        return "I'm sorry, but I couldn't generate a response. Please try again."
+
+    # Ensure only final_answer is used
+    if "final_answer" not in raw_response.lower():
+        return "Error: The response did not use the `final_answer` tool. Please try again."
+
+    # Remove code snippets
+    clean_response = remove_code_snippets(raw_response)
+
+    # Detect and prevent repeated phrases (possible infinite loop indicator)
+    words = clean_response.split()
+    if len(set(words)) < 5:  # Arbitrary threshold for detecting repetition
+        return "I'm unable to generate a meaningful response. Please refine your query."
+
     return clean_response
 
 # 🚀 Launch the Gradio UI.