import gradio as gr
import spaces

from src.ui import (
    image_tab,
)


css = """
@import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600&display=swap');
body {
    font-family: 'Poppins', sans-serif !important;
}
.center-content {
    text-align: center;
    max-width: 600px;
    margin: 0 auto;
    padding: 20px;
}
.center-content h1 {
    font-weight: 600;
    margin-bottom: 1rem;
}
.center-content p {
    margin-bottom: 1.5rem;
}
"""


# Main Gradio app
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"):
            gr.Label("Coming soon!")
        with gr.Tab(label="🎬 Video"):
            gr.Label("Coming soon!")
        with gr.Tab(label="📄 Text"):
            gr.Label("Coming soon!")
    
demo.launch(
    share=False,
    debug=True,
)