import gradio as gr

from config import css
from tabs.images.ui import image_tab
# from tabs.tts.ui import audio_tab
# from tabs.sound.ui import sound_tab


with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
    # Header
    with gr.Column(elem_classes="center-content"):
        gr.Markdown("""
        # 🚀 AAI: All AI
        Unleash your creativity with our multi-modal AI platform.
        [![Sync code to HF Space](https://github.com/mantrakp04/aai/actions/workflows/hf-space.yml/badge.svg)](https://github.com/mantrakp04/aai/actions/workflows/hf-space.yml)
        """)
    
    # Tabs
    with gr.Tabs():
        # with gr.Tab(label="🔊 Sound"):
        #     sound_tab()
        with gr.Tab(label="🖼️ Image"):
            image_tab()
        # with gr.Tab(label="🎵 Audio"):
        #     audio_tab()
        # with gr.Tab(label="🎥 Video"):
        #     video_tab()
        # with gr.Tab(label="📝 Text"):
        #     text_tab()


demo.launch(
    share=False,
    debug=True,
)