app.py

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

with gr.Blocks() as demo:
    with gr.Group():
        with gr.Tab("Generate Story"):
            story_generation_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()