Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio_client import Client, handle_file
 from google import genai 
 from google.genai import types 
 import os
@@ -49,11 +50,10 @@ def unified_image_generator(
     prompt: str, 
     images: Optional[List[str]] = None,
     oauth_token: Optional[gr.OAuthToken] = None
-) -> str:
+) -> tuple:
     """
     Handles all image generation tasks based on the number of input images.
-    - 0 images: Text-to-image
-    - 1+ images: Image-to-image (single or multi-modal)
+    Returns: (output_image_path, video_button_visible, video_output_visible)
     """
     if not verify_pro_status(oauth_token):
         raise gr.Error("Access Denied. This service is for PRO users only.")
@@ -84,11 +84,50 @@ def unified_image_generator(
         pil_image = Image.open(BytesIO(image_data))
         with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmpfile:
             pil_image.save(tmpfile.name)
-            return tmpfile.name
+            output_path = tmpfile.name
+        
+        # Determine if video button should be shown (only if exactly 1 input image)
+        show_video_button = images and len(images) == 1
+        
+        # Return output image path, video button visibility, and hide video output
+        return output_path, gr.update(visible=show_video_button), gr.update(visible=False)
 
     except Exception as e:
         raise gr.Error(f"Image generation failed: {e}")
 
+def create_video_transition(
+    input_image_gallery: List[str],
+    output_image: str,
+    oauth_token: Optional[gr.OAuthToken] = None
+) -> tuple:
+    """
+    Creates a video transition between the input and output images.
+    Returns: (video_path, video_visible)
+    """
+    if not verify_pro_status(oauth_token):
+        raise gr.Error("Access Denied. This service is for PRO users only.")
+    
+    if not input_image_gallery or not output_image:
+        raise gr.Error("Both input and output images are required for video creation.")
+    
+    try:
+        yield gr.update(visible=True, value=None)
+        
+        video_client = Client("multimodalart/wan-2-2-first-last-frame", hf_token=oauth_token.token)
+        
+        input_image_path = input_image_gallery[0][0]
+        
+        result = video_client.predict(
+            start_image_pil=handle_file(input_image_path),
+            end_image_pil=handle_file(output_image),
+            prompt="Smooth transition between the two images",
+            api_name="/generate_video"
+        )
+        
+        return result
+        
+    except Exception as e:
+        raise gr.Error(f"Video creation failed: {e}")
 
 # --- Gradio App UI ---
 css = '''
@@ -103,8 +142,8 @@ css = '''
 .grid-container img{object-fit: contain}
 .grid-container {display: grid;grid-template-columns: repeat(2, 1fr)}
 .grid-container:has(> .gallery-item:only-child) {grid-template-columns: 1fr}
-
 '''
+
 with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     gr.HTML('''
     <img class="logo-dark" src='https://huggingface.co/spaces/multimodalart/nano-banana/resolve/main/nano_banana_pros.png' style='margin: 0 auto; max-width: 500px' />
@@ -135,6 +174,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
             with gr.Column(scale=1):
                 output_image = gr.Image(label="Output", interactive=False, elem_id="output", type="filepath")
                 use_image_button = gr.Button("♻️ Use this Image for Next Edit")
+                create_video_button = gr.Button("Create a video between the two 🎥", variant="primary", visible=False)
+                video_output = gr.Video(label="Generated Video", visible=False)
+                
         gr.Markdown("## Thank you for being a PRO! 🤗")
     
     login_button = gr.LoginButton()
@@ -142,9 +184,13 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     # --- Event Handlers ---
     gr.on(
         triggers=[generate_button.click, prompt_input.submit],
+        fn=lambda: [gr.update(visible=False), gr.update(visible=False)],
+        inputs=[],
+        outputs=[create_video_button, video_output],
+    ).then(
         fn=unified_image_generator,
         inputs=[prompt_input, image_input_gallery],
-        outputs=[output_image],
+        outputs=[output_image, create_video_button, video_output],
     )
 
     use_image_button.click(
@@ -152,8 +198,19 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
         inputs=[output_image],
         outputs=[image_input_gallery]
     )
+    
+    # Video creation handler
+    create_video_button.click(
+        fn=lambda: gr.update(visible=True),
+        inputs=[],
+        outputs=[video_output],
+    ).then(
+        fn=create_video_transition,
+        inputs=[image_input_gallery, output_image],
+        outputs=[video_output],
+    )
 
-    # --- Access Control Logic (UNCHANGED) ---
+    # --- Access Control Logic ---
     def control_access(
         profile: Optional[gr.OAuthProfile] = None,
         oauth_token: Optional[gr.OAuthToken] = None