back to 6 grid

app.py

@@ -526,7 +526,7 @@ def get_voice_config(voice_profile: str) -> Tuple[str, VoiceSettings]:
 
     if "rachel" in vp or "female" in vp:
         return (
-            "21m00Tcm4TlvDq8ikWAM",
+            "21m00Tcm4TlvDq8ikWAM",  # Rachel
             VoiceSettings(
                 stability=0.5,
                 similarity_boost=0.9,
@@ -536,7 +536,7 @@ def get_voice_config(voice_profile: str) -> Tuple[str, VoiceSettings]:
         )
 
     return (
-        "pNInz6obpgDQGcFmaJgB",
+        "pNInz6obpgDQGcFmaJgB",  # Adam
         VoiceSettings(
             stability=0.6,
             similarity_boost=0.8,
@@ -759,7 +759,7 @@ def mcp_agent_pipeline(
 
 
 # =============================================================================
-# GALLERY LOADING
+# GALLERY (6 FIXED SLOTS)
 # =============================================================================
 
 def load_gallery_videos() -> List[str]:
@@ -771,26 +771,22 @@ def load_gallery_videos() -> List[str]:
         glob.glob(f"{gallery_output_dir}/*.mp4"),
         key=os.path.getmtime,
         reverse=True,
-    )
+    )[:6]
+
+    videos: List[str] = [None] * 6  # type: ignore
+    for i, path in enumerate(existing_videos):
+        videos[i] = path
 
-    return existing_videos
+    return videos
 
 
 # =============================================================================
 # GRADIO UI
 # =============================================================================
 
-custom_css = """
-#video-gallery {
-  max-height: 540px;
-  overflow-y: auto;
-}
-"""
-
 with gr.Blocks(
     title="AIQuoteClipGenerator - MCP + Gemini Edition",
     theme=gr.themes.Soft(),
-    css=custom_css,
 ) as demo:
     gr.Markdown(
         """
@@ -802,17 +798,17 @@ with gr.Blocks(
     """
     )
 
-    # Scrollable grid gallery
-    with gr.Accordion("📸 Example Gallery – All Generated Videos", open=True):
-        gr.Markdown("Scroll to explore all the clips you've generated so far.")
-        gallery = gr.Gallery(
-            label=None,
-            show_label=False,
-            elem_id="video-gallery",
-            columns=3,
-            height=540,
-            preview=False,
-        )
+    # 6-slot gallery grid (3x2)
+    with gr.Accordion("📸 Example Gallery – Recent Videos", open=True):
+        gr.Markdown("See what has been generated. Auto-updates after each run.")
+        with gr.Row():
+            gallery_video1 = gr.Video(height=300, show_label=False)
+            gallery_video2 = gr.Video(height=300, show_label=False)
+            gallery_video3 = gr.Video(height=300, show_label=False)
+        with gr.Row():
+            gallery_video4 = gr.Video(height=300, show_label=False)
+            gallery_video5 = gr.Video(height=300, show_label=False)
+            gallery_video6 = gr.Video(height=300, show_label=False)
 
     gr.Markdown("---")
     gr.Markdown("## 🎯 Generate Your Own Quote Video")
@@ -923,8 +919,14 @@ with gr.Blocks(
         v3 = videos[2] if len(videos) > 2 else None
 
         gallery_vids = load_gallery_videos()
+        g1 = gallery_vids[0] if len(gallery_vids) > 0 else None
+        g2 = gallery_vids[1] if len(gallery_vids) > 1 else None
+        g3 = gallery_vids[2] if len(gallery_vids) > 2 else None
+        g4 = gallery_vids[3] if len(gallery_vids) > 3 else None
+        g5 = gallery_vids[4] if len(gallery_vids) > 4 else None
+        g6 = gallery_vids[5] if len(gallery_vids) > 5 else None
 
-        return status, v1, v2, v3, gallery_vids
+        return status, v1, v2, v3, g1, g2, g3, g4, g5, g6
 
     generate_btn.click(
         process_and_display,
@@ -941,14 +943,33 @@ with gr.Blocks(
             video1,
             video2,
             video3,
-            gallery,
+            gallery_video1,
+            gallery_video2,
+            gallery_video3,
+            gallery_video4,
+            gallery_video5,
+            gallery_video6,
         ],
     )
 
+    # Load gallery when app starts
+    def initial_gallery():
+        vids = load_gallery_videos()
+        vids += [None] * (6 - len(vids))
+        return vids[:6]
+
     demo.load(
-        load_gallery_videos,
-        outputs=[gallery],
+        initial_gallery,
+        outputs=[
+            gallery_video1,
+            gallery_video2,
+            gallery_video3,
+            gallery_video4,
+            gallery_video5,
+            gallery_video6,
+        ],
     )
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=["/data/gallery_videos"])
+    demo.launch(allowed_paths=["/data/gallery_videos"])
+