import gradio as gr

from launch.image_generation import image_generation_ui
from launch.model_generation import model_generation_ui
from launch.story_generation import story_generation_ui
# from launch.image_edition import image_edition_ui


with gr.Blocks() as demo:
    with gr.Group():
        with gr.Tab("Generate Story"):
            story_generation_ui()

        # with gr.Tab("2D Character and Assets"):
        #     with gr.Tab("Edit Image"):
        #         image_edition_ui()
        with gr.Tab("Generate Image and Remove Background"):
            input_image, processed_image = image_generation_ui()

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

demo.launch()