import gradio as gr
from geo_painting import GeoPainting

geo_painting = GeoPainting()


def generate_image(input_promp, control_image):
    return geo_painting.generate_painting(input_promp, control_image)


input_promp = gr.Textbox(label="Input promp")
control_image = gr.Image(label="Select a control image")
output_image = gr.Image(label="Generated geospatial image")

iface = gr.Interface(generate_image, [input_promp, control_image], output_image)
iface.launch()