app.py

import gradio as gr
from main import generate_image


def ui(prompt, style):
    return generate_image(prompt, style)


title = "Gemini Text-to-Image Generator"
description = "Generate high-quality AI images using Google's Gemini API"

with gr.Blocks(title=title) as demo:
    gr.Markdown(f"# {title}")
    gr.Markdown(description)

    with gr.Row():
        prompt = gr.Textbox(label="Enter Image Prompt", placeholder="A futuristic city at sunset")
        style = gr.Dropdown(["Realistic", "Anime", "Cartoon", "Cyberpunk", "Oil Painting"], value="Realistic", label="Style")

    output = gr.Image(label="Generated Image")

    btn = gr.Button("Generate")
    btn.click(fn=ui, inputs=[prompt, style], outputs=output)


demo.launch(debug=True, share=True)