issue fixed 005

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 import torch
+import datetime
 from flask import Flask, request, render_template, jsonify, Response
 from flasgger import Swagger, swag_from
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
@@ -64,6 +65,17 @@ def is_incomplete(text):
 
 search_tool = DuckDuckGoSearchRun()
 
+def determine_folder(name):
+    ext = name.split('.')[-1].lower()
+    if ext in {"js", "jsx", "ts", "tsx", "html", "css"}:
+        return "frontend"
+    elif ext in {"py", "java", "cs", "go"}:
+        return "backend"
+    elif ext in {"sql", "prisma", "db"}:
+        return "database"
+    else:
+        return "misc"
+
 # ✅ Chat Completion using Phi-4
 # @GPU
 # def generate_full_reply(message, history):
@@ -350,26 +362,49 @@ def chat_stream_doc():
         return jsonify({"error": "No files found in generated output."}), 500
 
     zip_buffer = BytesIO()
+    file_list = []
+    file_count = 0
+
+
+
     with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zipf:
-        file_count = 0
         for match in matches:
-            filename = match.group(1).strip()
+            raw_filename = match.group(1).strip().rstrip("`")
             code = match.group(2).strip()
-            zipf.writestr(filename, code)
+            folder = determine_folder(raw_filename)
+            full_path = f"{folder}/{raw_filename}"
+            zipf.writestr(full_path, code)
+            file_list.append(full_path)
             file_count += 1
 
-            if file_count == 0:
-                print("⚠️ No files written to ZIP. Check LLM output format.")
-                print("LLM Reply:\n", reply)
-                return jsonify({"error": "No files found in generated output."}), 500
+        # ✅ Add a README.md
+        readme = (
+            "# Project Scaffold\n"
+            f"**Frontend:** {frontend}\n"
+            f"**Backend:** {backend}\n"
+            f"**Database:** {database}\n\n"
+            f"**Generated on:** {datetime.datetime.now().isoformat()}\n"
+            f"**Files:** {file_count}\n\n"
+            "This project was auto-generated using AI from a requirement document.\n"
+        )
+        zipf.writestr("README.md", readme)
+        file_list.append("README.md")
+
+        if file_count == 0:
+            return jsonify({"error": "No files found in generated output."}), 500
 
     zip_buffer.seek(0)
-    return Response(
+    response = Response(
         zip_buffer,
         mimetype='application/zip',
         headers={"Content-Disposition": "attachment; filename=generated_project.zip"}
     )
 
+    # Optional: attach log
+    response.headers["X-Generated-Files"] = ", ".join(file_list)
+
+    return response
+
 
 # ✅ Warm-up
 if __name__ == "__main__":