import gradio as gr
import torch
from modules import ui

def infer(prompt):
    return f"Generated image from: {prompt}"

with gr.Blocks() as demo:
    gr.Markdown("# Stable Diffusion WebUI")
    text_input = gr.Textbox(label="Enter Prompt")
    output = gr.Image()
    btn = gr.Button("Generate")
    btn.click(fn=infer, inputs=text_input, outputs=output)

demo.launch()