Spaces:

offerpk3
/

auto-app-builder

Sleeping

App Files Files Community

offerpk3 commited on Jul 2, 2025

Commit

bd789e3

verified ·

1 Parent(s): a08f0a2

Update generator.py

Browse files

Files changed (1) hide show

generator.py +24 -129

generator.py CHANGED Viewed

@@ -1,129 +1,24 @@
-import os
-import gradio as gr
-from generator import generate_app, enhance_app
-from helpers import save_as_zip
-import uuid
-import time
-def extract_react_preview(code: str) -> str:
-    # Wrap code in a full HTML React preview template
-    return f"""
-    <!DOCTYPE html>
-    <html>
-    <head>
-        <meta charset='UTF-8' />
-        <script crossorigin src=\"https://unpkg.com/react@18/umd/react.development.js\"></script>
-        <script crossorigin src=\"https://unpkg.com/react-dom@18/umd/react-dom.development.js\"></script>
-        <script src=\"https://unpkg.com/@babel/standalone/babel.min.js\"></script>
-        <style>
-            body {{ font-family: sans-serif; padding: 20px; }}
-        </style>
-    </head>
-    <body>
-        <div id='root'></div>
-        <script type='text/babel'>
-        {code}
-        ReactDOM.createRoot(document.getElementById('root')).render(<App />);
-        </script>
-    </body>
-    </html>
-    """
-def run_app(user_input, enhance_prompt, action, state, progress=gr.Progress()):
-    import google.generativeai as genai
-    from dotenv import load_dotenv
-    load_dotenv()
-    progress(0, desc="Authenticating with Gemini...")
-    genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-    model = genai.GenerativeModel("gemini-pro")
-    progress(0.2, desc="Sending prompt to Gemini...")
-    prompt_text = f"""Create a full React app based on this idea:
-{user_input}
-Return only relevant files like App.jsx, index.html, styles, utils or components in plain code blocks with filenames."""
-    try:
-        response = model.generate_content(prompt_text)
-        print("🔍 FULL GEMINI RESPONSE RAW >>>\n", response)
-        print("🔍 Gemini TEXT PREVIEW >>>\n", getattr(response, "text", "(no .text found)"))
-        if not hasattr(response, "text") or not response.text.strip():
-            raise ValueError("Gemini response is empty.")
-        parts = response.text.split("```")
-    except Exception as e:
-        print("❌ Gemini error:", e)
-        return (
-            "❌ Error: Could not generate app from Gemini.",
-            None,
-            "<p style='color:red;'>Gemini API failed or returned empty. Try again.</p>",
-            {},
-            {}
-        )
-    files = {}
-    current_file = None
-    for part in parts:
-        if part.strip().startswith("jsx") or part.strip().startswith("javascript") or part.strip().startswith("html"):
-            lines = part.strip().split("\n")
-            if len(lines) > 1 and lines[1].startswith("// filename:"):
-                current_file = lines[1].replace("// filename:", "").strip()
-                files[current_file] = "\n".join(lines[2:])
-    app_code = files.get("App.jsx", next(iter(files.values()), ""))
-    preview_html = extract_react_preview(app_code)
-    zip_bytes = save_as_zip({"files": files})
-    with open("generated_app.zip", "wb") as f:
-        f.write(zip_bytes)
-    state = {"bundle": {"full_code": app_code, "files": files}, "history": []}
-    return app_code, "generated_app.zip", preview_html, files, state
-with gr.Blocks(title="Auto App Builder") as iface:
-    gr.Markdown("# ⚡ Auto App Builder (Gemini Powered)\n_Powered by Google Gemini API_")
-    with gr.Row():
-        inp = gr.Textbox(label="App Idea", placeholder="e.g. Inventory system with login", lines=2)
-        btn = gr.Button("🚀 Generate App")
-    with gr.Row():
-        with gr.Column():
-            code_out = gr.Code(label="📄 Generated Code")
-            enh = gr.Textbox(label="Enhancement Prompt", placeholder="e.g. Add login, PDF export, dark mode...", lines=1)
-            btn_enh = gr.Button("✨ Enhance")
-            zip_file = gr.File(label="⬇️ Download App ZIP", interactive=True)
-            gr.Markdown("### 📁 File Explorer")
-            file_selector = gr.Dropdown(choices=[], label="Select File")
-            file_viewer = gr.Code(label="📂 File Content")
-        with gr.Column():
-            gr.Markdown("### 🔍 Live React Preview")
-            preview = gr.HTML(label="React App Preview")
-    state = gr.State()
-    def update_file_tree(code, zip_name, preview_path, files, state):
-        if isinstance(files, dict) and len(files) > 0:
-            keys = list(files.keys())
-            default_key = keys[0] if keys else None
-            return gr.update(choices=keys, value=default_key), files.get(default_key, "")
-        else:
-            return gr.update(choices=[], value=None), ""
-    btn.click(
-        fn=run_app,
-        inputs=[inp, enh, gr.State(""), state],
-        outputs=[code_out, zip_file, preview, file_selector, state]
-    ).then(
-        fn=lambda code, zip_name, preview_path, state_obj: update_file_tree(code, zip_name, preview_path, state_obj.get("bundle", {}).get("files", {}), state_obj),
-        inputs=[code_out, zip_file, preview, state],
-        outputs=[file_selector, file_viewer]
-    )
-    file_selector.change(
-        lambda name, state: state["bundle"]["files"].get(name, "") if state and "bundle" in state else "",
-        inputs=[file_selector, state],
-        outputs=file_viewer
-    )
-iface.launch()

+def generate_app(prompt: str):
+    print("🧠 Simulating Gemini API call...")
+    return {
+        "full_code": f'''
+function App() {{
+  return <div>⚡ Hello, this is your generated app for: {prompt}</div>;
+}}
+export default App;
+''',
+        "files": {
+            "App.jsx": f'''
+function App() {{
+  return <div>⚡ Hello, this is your generated app for: {prompt}</div>;
+}}
+export default App;
+'''
+        }
+    }
+def enhance_app(existing_bundle: dict, enhancement_prompt: str):
+    updated_code = existing_bundle["full_code"] + f"\n// Enhancement applied: {enhancement_prompt}"
+    existing_bundle["full_code"] = updated_code
+    existing_bundle["files"]["App.jsx"] = updated_code
+    return existing_bundle