Create app.py

app.py

@@ -0,0 +1,67 @@
+# app.py
+# -----------------------
+# Imports
+# -----------------------
+import google.generativeai as genai
+import imgkit
+import gradio as gr
+import uuid
+import os
+
+# -----------------------
+# Configure Gemini API ONCE at startup
+# -----------------------
+genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))  # use HF secret for API key
+model = genai.GenerativeModel("gemini-2.0-flash")  # initialized once
+
+# -----------------------
+# Function: Generate Mockup Image
+# -----------------------
+def generate_mockup(user_prompt):
+    """
+    Generate PNG mockup directly from user description
+    """
+    system_prompt = (
+        "You are a Power Apps mockup generator. "
+        "Convert user descriptions into complete HTML + CSS code that looks like a Power Apps mobile screen. "
+        "Do NOT include JSON or explanations. Only return valid HTML + CSS."
+    )
+
+    try:
+        # Generate HTML using the already configured model
+        response = model.generate_content(system_prompt + "\n" + user_prompt)
+        html_code = response.text.strip()
+
+        # Save HTML to a unique file
+        unique_id = str(uuid.uuid4())
+        html_file = f"mockup_{unique_id}.html"
+        img_file = f"mockup_{unique_id}.png"
+
+        with open(html_file, "w", encoding="utf-8") as f:
+            f.write(html_code)
+
+        # Convert HTML to PNG
+        imgkit.from_file(html_file, img_file)
+
+        return img_file
+    except Exception as e:
+        return None
+
+# -----------------------
+# Gradio UI
+# -----------------------
+with gr.Blocks() as demo:
+    gr.Markdown("## ⚡ Power Apps Mockup Generator (Gemini + Gradio)")
+
+    user_input = gr.Textbox(
+        label="Describe your Power Apps mockup screen",
+        lines=4,
+        placeholder="Example: A login screen with username, password, and login button"
+    )
+    generate_btn = gr.Button("Generate Mockup")
+    image_output = gr.Image(label="Mockup Preview")
+
+    generate_btn.click(fn=generate_mockup, inputs=user_input, outputs=image_output)
+
+# Run app
+demo.launch(server_name="0.0.0.0", server_port=7860)