Refactor UI structure and add image tab

- Refactored the UI structure to improve code organization and maintainability.
- Added a new image tab to the UI, allowing users to interact with image-related functionalities.

.github/workflows/hf-space.yml

@@ -15,4 +15,4 @@ jobs:
         - name: Sync to hf space
           env:
             HF_TOKEN: ${{ secrets.HF_TOKEN }}
-          run: git push https://mantrakp:$HF_TOKEN@huggingface.co/spaces/mantrakp/demo main
+          run: git push https://mantrakp:$HF_TOKEN@huggingface.co/spaces/mantrakp/aai main

README.md

@@ -1,6 +1,6 @@
 ---
 title: Aai
-emoji: 👁
+emoji: 🚀
 colorFrom: gray
 colorTo: pink
 sdk: gradio

app.py

@@ -0,0 +1,49 @@
+import gradio as gr
+
+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.
+        """)
+
+    # 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=True)

requirements.txt

@@ -1,3 +1,4 @@
 torch
 torchvision
 torchaudio
+gradio

src/ui/__init__.py

@@ -0,0 +1 @@
+from .images import image_tab

src/ui/images.py

@@ -0,0 +1,14 @@
+import os
+
+import gradio as gr
+
+from .tabs import (
+    flux_tab,
+)
+
+def image_tab():
+    with gr.Tabs():
+        with gr.Tab(label="Flux"):
+            flux_tab()
+        with gr.Tab(label="SDXL"):
+            gr.Label("Coming soon!")

src/ui/tabs/__init__.py

@@ -0,0 +1 @@
+from .images.flux_tab import flux_tab

src/ui/tabs/images/flux_tab.py

@@ -0,0 +1,8 @@
+import gradio as gr
+
+def flux_tab():
+    with gr.Column():
+        prompt = gr.Textbox(lines=5, label="Prompt")
+    with gr.Column():
+        with gr.Group():
+            output_image = gr.Image()