issue fixed 003

app.py

@@ -302,32 +302,58 @@ def chat_stream_doc():
 
     tech_stack = f"Frontend: {frontend}\nBackend: {backend}\nDatabase: {database}"
     prompt = (
-        f"You are a full-stack project generator.\n\n"
-        f"Requirement Document:\n{content}\n\n"
-        f"{tech_stack}\n\n"
-        f"Generate a complete project scaffold including:\n"
-        f"- Backend (API routes, models, server config)\n"
-        f"- Frontend (UI components, service calls)\n"
-        f"- Database schema\n"
-        f"Reply with full code organized as multiple files with filenames."
+        "You are a full-stack project code generator.\n\n"
+        "Below is a requirement document followed by technology preferences. Based on this, generate the full project scaffold.\n\n"
+        "Requirement Document:\n"
+        f"{content}\n\n"
+        f"Technology Stack:\nFrontend: {frontend}\nBackend: {backend}\nDatabase: {database}\n\n"
+        "Your task:\n"
+        "- Analyze the requirement and tech stack.\n"
+        "- Generate backend code: models, routes, and config.\n"
+        "- Generate frontend code: components, services.\n"
+        "- Define the database schema.\n\n"
+        "✅ Format the output as multiple files in this exact structure:\n\n"
+        "### File: <filename>\n"
+        "```<language>\n"
+        "<code content>\n"
+        "```\n\n"
+        "For example:\n\n"
+        "### File: app.py\n"
+        "```python\n"
+        "from flask import Flask\n"
+        "app = Flask(__name__)\n"
+        "@app.route('/')\n"
+        "def home():\n"
+        "    return 'Hello, world!'\n"
+        "```\n\n"
+        "### File: frontend/App.js\n"
+        "```javascript\n"
+        "import React from 'react';\n"
+        "function App() {\n"
+        "  return <h1>Hello from React</h1>;\n"
+        "}\n"
+        "export default App;\n"
+        "```\n\n"
+        "Now generate the complete project below 👇"
     )
 
     # ✅ Generate response from Phi-4
     reply = generate_full_reply(prompt, [])
 
     # ✅ Extract filenames and code from reply
-    file_pattern = r"### File: (.+?)\n```(.*?)```"
+    file_pattern = r"### File:\s*(.+?)\n```(?:\w+)?\n(.*?)```"
     matches = re.finditer(file_pattern, reply, re.DOTALL)
 
     if not matches:
+        print("⚠️ No file matches found in reply.")
+        print("Raw reply:\n", reply)
         return jsonify({"error": "No files found in generated output."}), 500
 
     zip_buffer = BytesIO()
     with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zipf:
-        for match in matches:
-            name = match.group(1).strip()
-            code = match.group(2).strip()
-            zipf.writestr(name, code)
+        for filename, code in matches:
+            print(f"✅ Writing file: {filename}")
+            zipf.writestr(filename.strip(), code.strip())
 
     zip_buffer.seek(0)
     return Response(