Remove demo files and fix image serving for proper download support

- Remove demo images (disaster_girl.jpg, grumpy.png, wednesday.png)
- Remove hardcoded examples from UI
- Change gallery type from 'pil' to 'filepath' for proper image serving
- Save generated images to outputs/ directory with unique filenames
- Fix image download issues on mobile devices and right-click functionality
- Add .gitignore to exclude IDE configs and generated outputs

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

.gitignore

@@ -0,0 +1,46 @@
+# IDE configuration files
+pyrightconfig.json
+.vscode/
+
+# Generated outputs
+outputs/
+
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+env/
+venv/
+ENV/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Model files (if you don't want to track large files)
+*.pt
+*.pth
+*.bin
+*.safetensors
+
+# OS files
+.DS_Store
+Thumbs.db
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/

app.py

@@ -368,7 +368,8 @@ def update_history(new_images, history):
 def use_history_as_input(evt: gr.SelectData):
     """Sets the selected history image as the new input image."""
     if evt.value is not None:
-        return gr.update(value=[(evt.value,)])
+        # For filepath gallery, return the path directly in a list
+        return gr.update(value=[evt.value])
     return gr.update()
     
 def encode_image(pil_image):
@@ -417,10 +418,17 @@ def suggest_next_scene_prompt(images):
     if images is not None:
         for item in images:
             try:
-                if isinstance(item[0], Image.Image):
-                    pil_images.append(item[0].convert("RGB"))
-                elif isinstance(item[0], str):
-                    pil_images.append(Image.open(item[0]).convert("RGB"))
+                if isinstance(item, str):
+                    # Direct file path from filepath gallery
+                    pil_images.append(Image.open(item).convert("RGB"))
+                elif isinstance(item, tuple) and len(item) > 0:
+                    # Tuple format (legacy support)
+                    if isinstance(item[0], Image.Image):
+                        pil_images.append(item[0].convert("RGB"))
+                    elif isinstance(item[0], str):
+                        pil_images.append(Image.open(item[0]).convert("RGB"))
+                elif isinstance(item, Image.Image):
+                    pil_images.append(item.convert("RGB"))
                 elif hasattr(item, "name"):
                     pil_images.append(Image.open(item.name).convert("RGB"))
             except Exception:
@@ -464,10 +472,17 @@ def infer(
     if images is not None:
         for item in images:
             try:
-                if isinstance(item[0], Image.Image):
-                    pil_images.append(item[0].convert("RGB"))
-                elif isinstance(item[0], str):
-                    pil_images.append(Image.open(item[0]).convert("RGB"))
+                if isinstance(item, str):
+                    # Direct file path from filepath gallery
+                    pil_images.append(Image.open(item).convert("RGB"))
+                elif isinstance(item, tuple) and len(item) > 0:
+                    # Tuple format (legacy support)
+                    if isinstance(item[0], Image.Image):
+                        pil_images.append(item[0].convert("RGB"))
+                    elif isinstance(item[0], str):
+                        pil_images.append(Image.open(item[0]).convert("RGB"))
+                elif isinstance(item, Image.Image):
+                    pil_images.append(item.convert("RGB"))
                 elif hasattr(item, "name"):
                     pil_images.append(Image.open(item.name).convert("RGB"))
             except Exception:
@@ -484,7 +499,7 @@ def infer(
     
 
     # Generate the image
-    image = pipe(
+    images_pil = pipe(
         image=pil_images if len(pil_images) > 0 else None,
         prompt=prompt,
         height=height,
@@ -496,13 +511,19 @@ def infer(
         num_images_per_prompt=num_images_per_prompt,
     ).images
 
-    # Return images, seed, and make button visible
-    return image, seed, gr.update(visible=True), gr.update(visible=True)
+    # Save images to temporary files for proper serving
+    output_paths = []
+    os.makedirs("outputs", exist_ok=True)
+    for idx, img in enumerate(images_pil):
+        output_path = f"outputs/output_{seed}_{idx}_{int(time.time()*1000)}.png"
+        img.save(output_path)
+        output_paths.append(output_path)
 
+    # Return image paths, seed, and make button visible
+    return output_paths, seed, gr.update(visible=True), gr.update(visible=True)
 
-# --- Examples and UI Layout ---
-examples = []
 
+# --- UI Layout ---
 css = """
 #col-container {
     margin: 0 auto;
@@ -531,9 +552,9 @@ with gr.Blocks(css=css) as demo:
         """)
         with gr.Row():
             with gr.Column():
-                input_images = gr.Gallery(label="Input Images", 
-                                          show_label=False, 
-                                          type="pil", 
+                input_images = gr.Gallery(label="Input Images",
+                                          show_label=False,
+                                          type="filepath",
                                           interactive=True)
 
                 prompt = gr.Text(
@@ -596,7 +617,7 @@ with gr.Blocks(css=css) as demo:
         
 
             with gr.Column():
-                result = gr.Gallery(label="Result", show_label=False, type="pil")
+                result = gr.Gallery(label="Result", show_label=False, type="filepath")
                 with gr.Row():
                     use_output_btn = gr.Button("↗️ Use as input", variant="secondary", size="sm", visible=False)
                     turn_video_btn = gr.Button("🎬 Turn into Video", variant="secondary", size="sm", visible=False)
@@ -613,14 +634,6 @@ with gr.Blocks(css=css) as demo:
                     visible=False
                 )
 
-        gr.Examples(examples=[
-            [["disaster_girl.jpg", "grumpy.png"], "Next Scene: the camera zooms in, showing the cat walking away from the fire"],
-            [["wednesday.png"], "Next Scene: The camera pulls back and rises to an elevated angle, revealing the full dance floor with the choreographed movements of all dancers as the central figure becomes part of the larger ensemble."],
-            ],
-                inputs=[input_images, prompt], 
-                outputs=[result, seed], 
-                fn=infer, 
-                cache_examples="lazy")