Update app.py

app.py

@@ -612,6 +612,9 @@ css = """
 .seed-image-upload img { max-height: 120px !important; object-fit: contain; }
 .main-row { align-items: flex-start !important; }
 .controls-column { min-width: 280px; }
+.world-gallery { margin-bottom: 12px; }
+.world-gallery .gallery { gap: 8px !important; }
+.world-gallery .gallery-item { border-radius: 8px; overflow: hidden; }
 
 /* Mobile responsive styles */
 @media (max-width: 768px) {
@@ -635,6 +638,9 @@ def create_app():
         # Latest frame for display
         latest_frame = gr.State(None)
         latest_frame_count = gr.State(0)
+        
+        # Selected seed URL from examples
+        selected_seed_url = gr.State(None)
 
         gr.Markdown("""
         # 🌍 Waypoint 1 Small
@@ -671,15 +677,24 @@ def create_app():
                     lines=2,
                 )
 
-                with gr.Accordion("Seed Image", open=False):
-                    seed_dropdown = gr.Dropdown(
-                        choices=["Random"] + [f"Seed {i+1}" for i in range(len(SEED_FRAME_URLS))],
-                        value="Random",
-                        label="Preset Seeds",
-                        scale=1,
+                with gr.Accordion("World Selection", open=True):
+                    gr.Markdown("**Choose a starting world** (or leave blank for random):")
+                    
+                    # Gallery for world selection
+                    world_gallery = gr.Gallery(
+                        value=SEED_FRAME_URLS,
+                        label="Preset Worlds",
+                        columns=5,
+                        rows=1,
+                        height=100,
+                        object_fit="cover",
+                        allow_preview=False,
+                        elem_classes=["world-gallery"],
                     )
+                    
+                    gr.Markdown("**Or upload your own:**")
                     seed_image_upload = gr.Image(
-                        label="Or Upload Your Own",
+                        label="Custom World Image",
                         type="pil",
                         sources=["upload", "clipboard"],
                         height=120,
@@ -690,8 +705,14 @@ def create_app():
                 frame_display = gr.Number(label="Frame", value=0, interactive=False)
 
         # --- Event Handlers ---
-
-        def on_start(seed_choice, uploaded_image, prompt):
+        
+        def on_gallery_select(evt: gr.SelectData):
+            """Handle gallery selection - return the URL of selected world."""
+            if evt.index is not None and evt.index < len(SEED_FRAME_URLS):
+                return SEED_FRAME_URLS[evt.index]
+            return None
+
+        def on_start(selected_url, uploaded_image, prompt):
             """Start GPU session - generator that shows loading then first frame."""
             # Show loading state immediately
             loading_img = create_loading_image(text="Generating World ...")
@@ -706,14 +727,15 @@ def create_app():
             )
 
             # Determine seed image/url
+            # Priority: uploaded image > selected from gallery > random
             seed_image = None
             seed_url = None
 
             if uploaded_image is not None:
                 seed_image = uploaded_image
-            elif seed_choice and seed_choice != "Random":
-                idx = int(seed_choice.split()[-1]) - 1
-                seed_url = SEED_FRAME_URLS[idx]
+            elif selected_url is not None:
+                seed_url = selected_url
+            # else: random will be chosen in create_gpu_game_loop
 
             command_queue = Queue()
             gen = create_gpu_game_loop(
@@ -781,22 +803,21 @@ def create_app():
             except StopIteration:
                 return current_frame, current_count, current_frame, current_count
 
-        def on_reset(state, seed_choice, uploaded_image, prompt):
+        def on_reset(state, selected_url, uploaded_image, prompt):
             """Reset world with new seed."""
             if len(state) == 0:
                 return None, 0, None, 0
 
             gen, command_queue = state
 
-            # Priority: uploaded image > preset seed > random
+            # Priority: uploaded image > selected from gallery > random
             seed_image = None
             seed_url = None
 
             if uploaded_image is not None:
                 seed_image = uploaded_image
-            elif seed_choice and seed_choice != "Random":
-                idx = int(seed_choice.split()[-1]) - 1
-                seed_url = SEED_FRAME_URLS[idx]
+            elif selected_url is not None:
+                seed_url = selected_url
 
             command_queue.put(ResetCommand(seed_image=seed_image, seed_url=seed_url, prompt=prompt))
 
@@ -815,9 +836,15 @@ def create_app():
             return value
 
         # Wire up events
+        world_gallery.select(
+            fn=on_gallery_select,
+            inputs=[],
+            outputs=[selected_seed_url],
+        )
+        
         start_btn.click(
             fn=on_start,
-            inputs=[seed_dropdown, seed_image_upload, prompt_input],
+            inputs=[selected_seed_url, seed_image_upload, prompt_input],
             outputs=[session_state, latest_frame, latest_frame_count,
                     video_output, frame_display, start_btn, stop_btn],
         )
@@ -831,7 +858,7 @@ def create_app():
 
         reset_btn.click(
             fn=on_reset,
-            inputs=[session_state, seed_dropdown, seed_image_upload, current_prompt],
+            inputs=[session_state, selected_seed_url, seed_image_upload, current_prompt],
             outputs=[latest_frame, latest_frame_count, video_output, frame_display],
         )
 
@@ -877,4 +904,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()