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() | |