Update app.py

app.py

@@ -1,26 +1,16 @@
-# app.py
-# -----------------------
-# Imports
-# -----------------------
-import google.generativeai as genai
-import imgkit
-import gradio as gr
+from html2image import Html2Image
 import uuid
 import os
+import google.generativeai as genai
+import gradio as gr
 
-# -----------------------
-# 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
+# Configure Gemini API
+genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))
+model = genai.GenerativeModel("gemini-2.0-flash")
+
+hti = Html2Image()
 
-# -----------------------
-# 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. "
@@ -28,11 +18,10 @@ def generate_mockup(user_prompt):
     )
 
     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 filenames
         unique_id = str(uuid.uuid4())
         html_file = f"mockup_{unique_id}.html"
         img_file = f"mockup_{unique_id}.png"
@@ -40,19 +29,16 @@ def generate_mockup(user_prompt):
         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)
+        # Convert HTML → PNG
+        hti.screenshot(html_file=html_file, save_as=img_file)
 
         return img_file
     except Exception as e:
-        return None
+        return f"Error: {e}"
 
-# -----------------------
 # 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,
@@ -60,8 +46,6 @@ with gr.Blocks() as demo:
     )
     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)
+demo.launch(server_name="0.0.0.0", server_port=7860)