Update app.py

app.py

@@ -19,7 +19,7 @@ def query(payload):
 
 # Extract only Python code from text
 def extract_python_code(text):
-    code_pattern = r"```python(.*?)```"
+    code_pattern = r"```(?:python)?(.*?)```"
     match = re.search(code_pattern, text, re.DOTALL)
     if match:
         return match.group(1).strip()