| 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. | |
| [](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, | |
| ) | |