issue fixed 001

app.py

@@ -316,8 +316,11 @@ def chat_stream_doc():
     reply = generate_full_reply(prompt, [])
 
     # ✅ Extract filenames and code from reply
-    file_pattern = r"(?<=^|\n)```(?P<lang>[a-z]*)\s*file=(?P<name>[^\n]+)\n(?P<code>[\s\S]+?)```"
-    matches = re.finditer(file_pattern, reply, re.MULTILINE)
+    file_pattern = r"### File: (.+?)\n```(.*?)```"
+    matches = re.finditer(file_pattern, reply, re.DOTALL)
+    for match in matches:
+        filename = match.group(1).strip()
+        code = match.group(2).strip()
 
     if not matches:
         return jsonify({"error": "No files found in generated output."}), 500