Update app.py

app.py

@@ -30,32 +30,40 @@ def extract_react_preview(code: str) -> str:
     """
 
 def run_app(user_input, enhance_prompt, action, state, progress=gr.Progress()):
-    progress(0, desc="Starting Gemini generation...")
-
-    if not state:
-        progress(0.1, desc="Calling Gemini API...")
-        code_bundle = generate_app(user_input)
-        state = {"bundle": code_bundle, "history": []}
-    else:
-        code_bundle = state["bundle"]
-
-    if action == "Enhance" and enhance_prompt:
-        progress(0.5, desc="Enhancing with Gemini...")
-        enhanced = enhance_app(code_bundle, enhance_prompt)
-        state["bundle"] = enhanced
-        state["history"].append(enhance_prompt)
-        code_bundle = enhanced
-
-    progress(0.7, desc="Creating zip archive...")
-    zip_bytes = save_as_zip(code_bundle)
+    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."""
+
+    response = model.generate_content(prompt_text)
+    parts = response.text.split("```")
+
+    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)
 
-    progress(0.9, desc="Generating preview...")
-    preview_html = extract_react_preview(code_bundle.get("files", {}).get("App.jsx", ""))
-
-    progress(1.0, desc="Done!")
-    return code_bundle.get("full_code", ""), "generated_app.zip", preview_html, code_bundle.get("files", {}), state
+    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_")
@@ -98,16 +106,6 @@ with gr.Blocks(title="Auto App Builder") as iface:
         outputs=[file_selector, file_viewer]
     )
 
-    btn_enh.click(
-        fn=run_app,
-        inputs=[inp, enh, gr.State("Enhance"), 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],