Update generator.py

generator.py

@@ -1,17 +1,24 @@
-import os
-import google.generativeai as genai
-
-genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
-
-model = genai.GenerativeModel("gemini-pro")
-
 def generate_app(prompt: str):
-    system = "You are a full-stack developer. Build full React+Tailwind frontend and Express+MongoDB backend for:"
-    response = model.generate_content(f"{system}\n{prompt}")
-    return {"full_code": response.text, "manifest": prompt}
+    print("🧠 Simulating Gemini API call...")
+    return {
+        "full_code": '''
+function App() {
+  return <div>⚡ Hello, this is your generated app for: {prompt}</div>;
+}
+export default App;
+''',
+        "files": {
+            "App.jsx": '''
+function App() {
+  return <div>⚡ Hello, this is your generated app for: {prompt}</div>;
+}
+export default App;
+'''
+        }
+    }
 
-def enhance_app(bundle, enhance_prompt: str):
-    system = "You are improving an existing app. Current code is below. Modify it as follows:"
-    prompt = f"{system}\n\n{bundle['full_code']}\n\nEnhance by: {enhance_prompt}"
-    response = model.generate_content(prompt)
-    return {"full_code": response.text, "manifest": bundle["manifest"] + " + " + enhance_prompt}
+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