Upload app.py with huggingface_hub

app.py

@@ -1,6 +1,10 @@
 # -*- coding: utf-8 -*-
 import os
 import warnings
+import io
+import zipfile
+import tempfile
+from pathlib import Path
 
 import gradio as gr
 
@@ -26,7 +30,7 @@ else:
     print("GROQ_API_KEY not set. The app will work in demo mode with placeholder messages.")
 
 # ----------------------------------------------------------------------
-# Helper: list of supported languages (common set)
+# Supported languages (common set)
 # ----------------------------------------------------------------------
 SUPPORTED_LANGUAGES = [
     "Python",
@@ -55,7 +59,6 @@ SUPPORTED_LANGUAGES = [
 def improve_code(original_code: str, language: str) -> str:
     """
     Sends the original code to the LLM and returns an improved version.
-    The language selector is used to guide the model.
     """
     if not original_code.strip():
         return "Please provide some code to improve."
@@ -83,24 +86,49 @@ def improve_code(original_code: str, language: str) -> str:
         return f"Error communicating with Groq: {e}"
 
 # ----------------------------------------------------------------------
-# Function: generate full web code from an idea
+# Helper: create a zip file with given files
 # ----------------------------------------------------------------------
-def generate_web_code(idea: str, language: str) -> str:
+def _create_zip(file_dict: dict) -> str:
+    """
+    file_dict: mapping of relative file path -> file content (string)
+    Returns the path to the created zip file.
+    """
+    tmp_dir = tempfile.mkdtemp()
+    zip_path = Path(tmp_dir) / "generated_project.zip"
+
+    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zipf:
+        for rel_path, content in file_dict.items():
+            # Ensure parent directories exist inside the zip
+            zipf.writestr(rel_path, content)
+
+    return str(zip_path)
+
+# ----------------------------------------------------------------------
+# Function: generate full web project from an idea
+# ----------------------------------------------------------------------
+def generate_web_code(idea: str, language: str):
     """
     Generates a complete web project (HTML/CSS/JS or a single‑file script)
     based on the user's idea and the selected language.
+    Returns a tuple: (preview_text, zip_file_path)
     """
     if not idea.strip():
-        return "Please describe the web idea you want to build."
+        return ("Please describe the web idea you want to build.", None)
 
     if not groq_client:
-        return "No Groq API key configured. Set the GROQ_API_KEY environment variable."
+        return ("No Groq API key configured. Set the GROQ_API_KEY environment variable.", None)
 
     prompt = (
         f"You are a senior full‑stack developer. Based on the following description, "
         f"write a complete, runnable {language} web application. Include all necessary "
         "files (HTML, CSS, JavaScript, or a single script) and comment the code. "
-        "Return the code only, without extra explanation.\n\n"
+        "Also provide a minimal requirements.txt, a README.md explaining how to run the project, "
+        "and a Dockerfile that builds and runs the app. Return the files in the following format:\n"
+        "=== filename ===\n"
+        "<file content>\n"
+        "=== filename ===\n"
+        "<file content>\n"
+        "Do not add any extra explanation.\n\n"
         f"Description:\n{idea}"
     )
 
@@ -112,9 +140,34 @@ def generate_web_code(idea: str, language: str) -> str:
             max_tokens=4000,
         )
         generated = response.choices[0].message.content
-        return generated.strip() if generated else "No content returned from Groq."
+        if not generated:
+            return ("No content returned from Groq.", None)
+
+        # Parse the generated sections
+        sections = {}
+        current_file = None
+        for line in generated.splitlines():
+            if line.startswith("=== ") and line.endswith(" ==="):
+                current_file = line.strip("= ").strip()
+                sections[current_file] = ""
+            elif current_file:
+                sections[current_file] += line + "\n"
+
+        # Ensure we have at least a main file for preview
+        preview_key = None
+        for key in sections:
+            if key.lower().endswith((".html", ".py", ".js")):
+                preview_key = key
+                break
+        preview_text = sections.get(preview_key, "Generated files:\n" + "\n".join(sections.keys()))
+
+        # Create zip with all files
+        zip_path = _create_zip(sections)
+
+        return (preview_text, zip_path)
+
     except Exception as e:
-        return f"Error communicating with Groq: {e}"
+        return (f"Error communicating with Groq: {e}", None)
 
 # ----------------------------------------------------------------------
 # Gradio UI
@@ -169,16 +222,21 @@ with gr.Blocks() as demo:
                         lines=12,
                     )
                 with gr.Column():
-                    generated_output = gr.Textbox(
-                        label="Generated Code",
+                    generated_preview = gr.Textbox(
+                        label="Generated Files Preview",
                         placeholder="The generated code will appear here",
                         lines=12,
                     )
+                    download_btn = gr.File(
+                        label="Download Project Zip",
+                        type="file",
+                        visible=False,
+                    )
             generate_btn = gr.Button("Generate")
             generate_btn.click(
                 fn=generate_web_code,
                 inputs=[idea_input, lang_select_generate],
-                outputs=generated_output,
+                outputs=[generated_preview, download_btn],
             )
 
 demo.launch()