issue fixed 007

app.py

@@ -1,7 +1,6 @@
 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
@@ -65,17 +64,6 @@ 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):
@@ -362,49 +350,26 @@ 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:
-            raw_filename = match.group(1).strip().rstrip("`")
+            filename = match.group(1).strip().rstrip("`")
             code = match.group(2).strip()
-            # folder = determine_folder(raw_filename)
-            # full_path = f"{folder}/{raw_filename}"
-            zipf.writestr(raw_filename, code)
-            # file_list.append(full_path)
+            zipf.writestr(filename, code)
             file_count += 1
 
-        # ✅ 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
+            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
 
     zip_buffer.seek(0)
-    response = Response(
+    return 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__":