import gradio as gr

from launch.image_generation import image_generation_ui
from launch.model_generation import model_generation_ui


with gr.Blocks() as demo:
    with gr.Group():
        with gr.Tab("Generate Image and Remove Background"):
            input_image, processed_image = image_generation_ui()

        with gr.Tab("Generate 3D Model"):
            output_model_obj, output_model_glb = model_generation_ui(
                processed_image)

demo.launch()