Create app.py

app.py

@@ -0,0 +1,61 @@
+import gradio as gr
+import requests
+from PIL import Image
+import io
+
+# ------------------------------------------------------------------
+# Helper: fetch list of available models
+# ------------------------------------------------------------------
+def get_models():
+    r = requests.get("https://image.pollinations.ai/models", timeout=10)
+    models = [m.strip() for m in r.text.splitlines() if m.strip()]
+    extra = ["kontext", "gptimage"]
+    for m in extra:
+        if m not in models:
+            models.append(m)
+    return models
+
+# ------------------------------------------------------------------
+# Image-generation function
+# ------------------------------------------------------------------
+def generate_image(prompt, model):
+    url = f"https://image.pollinations.ai/prompt/{prompt}?nologo=true&seed=random"
+    if model:
+        url += f"&model={model}"
+    r = requests.get(url, timeout=30)
+    img = Image.open(io.BytesIO(r.content))
+    return img
+
+# ------------------------------------------------------------------
+# Gradio UI
+# ------------------------------------------------------------------
+models = get_models()
+
+with gr.Blocks(theme="soft") as demo:
+    gr.Markdown("# 🖼️ Pollinations Image Generator")
+    
+    with gr.Row():
+        with gr.Column(scale=3):
+            prompt_box = gr.Textbox(
+                label="Prompt",
+                placeholder="A cat wearing sunglasses on the moon...",
+                lines=2
+            )
+            model_dropdown = gr.Dropdown(
+                choices=models,
+                value=models[0],
+                label="Model"
+            )
+            generate_btn = gr.Button("Generate", variant="primary")
+        
+        with gr.Column(scale=3):
+            output_img = gr.Image(type="pil", label="Generated Image")
+    
+    generate_btn.click(
+        fn=generate_image,
+        inputs=[prompt_box, model_dropdown],
+        outputs=output_img
+    )
+
+if __name__ == "__main__":
+    demo.launch()