Update app.py

app.py

@@ -125,9 +125,20 @@ model = LiteLLMModel(
     api_key=os.getenv("OPENAI_API_KEY"),
     api_base=os.getenv("CUSTOM_OPENAI_API_BASE"),
     temperature=0.1,
-    frequency_penalty=0.2
+    frequency_penalty=0.2,
+    messages=[
+        {
+            "role": "system",
+            "content": """ALWAYS format code responses with:
+            ```py
+            # Your code
+            ```
+            Use markdown for text and strict triple-backtick for code blocks"""
+        }
+    ]
 )
 
+
 text_limit = 20000
 ti_tool = TextInspectorTool(model, text_limit)
 browser = SimpleTextBrowser(**BROWSER_CONFIG)
@@ -179,18 +190,37 @@ WEB_TOOLS = [
     ArchiveSearchTool(browser),
     TextInspectorTool(model, text_limit),
 ]
+
+from smolagents.parsers import CodeParser
+import re
+
+class RobustCodeParser(CodeParser):
+    def extract_code(self, response: str) -> str:
+        try:
+            return super().extract_code(response)
+        except ValueError:
+            # Fallback pattern matching
+            code_match = re.search(r"```(?:python|py)?\n(.*?)\n```", response, re.DOTALL)
+            if code_match:
+                return code_match.group(1).strip()
+            raise ValueError(f"Invalid code format in response:\n{response}")
+
+# Replace in agent creation:
+
 # Agent creation in a factory function
 def create_agent():
-    """Creates a fresh agent instance for each session"""
     return CodeAgent(
         model=model,
         tools=[visualizer] + WEB_TOOLS,
-        max_steps=5,
-        verbosity_level=2,
+        max_steps=7,  # Increased from 5
+        verbosity_level=3,  # Higher debug info
         additional_authorized_imports=AUTHORIZED_IMPORTS,
-        planning_interval=4,
+        planning_interval=3,
+        code_block_delimiters=("```py", "```"),  # Explicit code formatting [github.com]
+        code_clean_pattern=r"^[\s\S]*?(```py\n[\s\S]*?\n```)",  # Improved regex
+        enforce_code_format=True,
+        parser=RobustCodeParser() # Explicitly use RobustCodeParser here!
     )
-
 document_inspection_tool = TextInspectorTool(model, 20000)
 
 def stream_to_gradio(