import gradio as gr

from config import css
from tabs.image_tab import image_tab
from tabs.audio_tab import audio_tab
from tabs.video_tab import video_tab
from tabs.text_tab import text_tab

def main():
    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="🖼️ 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,
    )

if __name__ == "__main__":
    main()