import gradio as gr
import spaces
import torch

# The space detects this decorator at startup to allocate ZeroGPU resources
@spaces.GPU(duration=30)
def generate_image(prompt):
    # Short-lived, synchronous GPU computation
    # The GPU remains assigned only while this function is running
    result = perform_gpu_inference(prompt) 
    return result

with gr.Blocks() as demo:
    gr.Markdown("## ZeroGPU Inference Demo")
    input_text = gr.Textbox(label="Prompt")
    output_img = gr.Image(label="Result")
    submit_btn = gr.Button("Generate")

    submit_btn.click(generate_image, inputs=input_text, outputs=output_img)

demo.launch()