import gradio as gr
from only_gradio_server import process_images

# Create Gradio interface
iface = gr.Interface(fn=process_images,
                     inputs=[gr.Image(type='filepath', label='Input Image 1'),
                             gr.Image(type='filepath', label='Input Image 2', image_mode="RGBA"),
                             gr.Textbox(label='Replace Object Name')],
                     outputs='image',
                     title="Image Processing",
                     description="Object to Object Replacement")

# Launch Gradio interface
iface.launch()