Update app.py

app.py

@@ -241,6 +241,10 @@ CRITICAL RULES:
             search_match = re.search(r'SEARCH:\s*(.+?)(?:\n|$)', response)
             wiki_match = re.search(r'WIKIPEDIA:\s*(.+?)(?:\n|$)', response)
             python_match = re.search(r'PYTHON:\s*```(?:python)?\n?(.*?)```', response, re.DOTALL)
+            if not python_match:
+                python_match = re.search(r'```python\n(.*?)```', response, re.DOTALL)
+            if not python_match:
+                python_match = re.search(r'```\n(.*?)```', response, re.DOTALL)
 
             if answer_match:
                 answer = answer_match.group(1).strip()