Update app.py

app.py

@@ -281,7 +281,8 @@ def create_file(filename, prompt, response, should_save=True):
     if not should_save:
         return
     base_filename, ext = os.path.splitext(filename)
-    has_python_code = bool(re.search(r"```python([\s\S]*?)```", response))
+    has_python_code = re.search(r"```python([\s\S]*?)```", prompt.strip() + '\r\n' + response)
+    has_python_code = bool(re.search(r"```python([\s\S]*?)```", prompt.strip() + '\r\n' + response))
     if ext in ['.txt', '.htm', '.md']:
         with open(f"{base_filename}.md", 'w') as file:
             content = prompt.strip() + '\r\n' + response