Update app.py

app.py

@@ -61,47 +61,47 @@ optimize_pipeline_(
 MAX_SEED = np.iinfo(np.int32).max
 
 
-def _generate_video_segment(
-    input_image_path: str,
-    output_image_path: str,
-    prompt: str,
-    request: gr.Request
-) -> str:
-    """
-    Generate a single video segment between two frames by calling an external
-    Wan 2.2 image-to-video service hosted on Hugging Face Spaces.
+# def _generate_video_segment(
+#     input_image_path: str,
+#     output_image_path: str,
+#     prompt: str,
+#     request: gr.Request
+# ) -> str:
+#     """
+#     Generate a single video segment between two frames by calling an external
+#     Wan 2.2 image-to-video service hosted on Hugging Face Spaces.
 
-    This helper function is used internally when the user asks to create
-    a video between the input and output images.
+#     This helper function is used internally when the user asks to create
+#     a video between the input and output images.
 
-    Args:
-        input_image_path (str):
-            Path to the starting frame image on disk.
-        output_image_path (str):
-            Path to the ending frame image on disk.
-        prompt (str):
-            Text prompt describing the camera movement / transition.
-        request (gr.Request):
-            Gradio request object, used here to forward the `x-ip-token`
-            header to the downstream Space for authentication/rate limiting.
+#     Args:
+#         input_image_path (str):
+#             Path to the starting frame image on disk.
+#         output_image_path (str):
+#             Path to the ending frame image on disk.
+#         prompt (str):
+#             Text prompt describing the camera movement / transition.
+#         request (gr.Request):
+#             Gradio request object, used here to forward the `x-ip-token`
+#             header to the downstream Space for authentication/rate limiting.
 
-    Returns:
-        str:
-            A string returned by the external service, usually a URL or path
-            to the generated video.
-    """
-    x_ip_token = request.headers['x-ip-token']
-    video_client = Client(
-        "multimodalart/wan-2-2-first-last-frame",
-        headers={"x-ip-token": x_ip_token}
-    )
-    result = video_client.predict(
-        start_image_pil=handle_file(input_image_path),
-        end_image_pil=handle_file(output_image_path),
-        prompt=prompt,
-        api_name="/generate_video",
-    )
-    return result[0]["video"]
+#     Returns:
+#         str:
+#             A string returned by the external service, usually a URL or path
+#             to the generated video.
+#     """
+#     x_ip_token = request.headers['x-ip-token']
+#     video_client = Client(
+#         "multimodalart/wan-2-2-first-last-frame",
+#         headers={"x-ip-token": x_ip_token}
+#     )
+#     result = video_client.predict(
+#         start_image_pil=handle_file(input_image_path),
+#         end_image_pil=handle_file(output_image_path),
+#         prompt=prompt,
+#         api_name="/generate_video",
+#     )
+#     return result[0]["video"]
 
 
 def build_camera_prompt(
@@ -266,58 +266,58 @@ def infer_camera_edit(
     return result, seed, prompt
 
 
-def create_video_between_images(
-    input_image: Optional[Image.Image],
-    output_image: Optional[np.ndarray],
-    prompt: str,
-    request: gr.Request,
-    api_name=False
-) -> str:
-    """
-    Create a short transition video between the input and output images via the 
-    Wan 2.2 first-last-frame Space.
+# def create_video_between_images(
+#     input_image: Optional[Image.Image],
+#     output_image: Optional[np.ndarray],
+#     prompt: str,
+#     request: gr.Request,
+#     api_name=False
+# ) -> str:
+#     """
+#     Create a short transition video between the input and output images via the 
+#     Wan 2.2 first-last-frame Space.
 
-    Args:
-        input_image (PIL.Image.Image | None):
-            Starting frame image (the original / previous view).
-        output_image (numpy.ndarray | None):
-            Ending frame image - the output image with the the edited camera angles.
-        prompt (str):
-            The camera movement prompt used to describe the transition.
-        request (gr.Request):
-            Gradio request object, used to forward the `x-ip-token` header
-            to the video generation app.
+#     Args:
+#         input_image (PIL.Image.Image | None):
+#             Starting frame image (the original / previous view).
+#         output_image (numpy.ndarray | None):
+#             Ending frame image - the output image with the the edited camera angles.
+#         prompt (str):
+#             The camera movement prompt used to describe the transition.
+#         request (gr.Request):
+#             Gradio request object, used to forward the `x-ip-token` header
+#             to the video generation app.
 
-    Returns:
-        str:
-            a path pointing to the generated video.
+#     Returns:
+#         str:
+#             a path pointing to the generated video.
 
-    Raises:
-        gr.Error:
-            If either image is missing or if the video generation fails.
-    """
-    if input_image is None or output_image is None:
-        raise gr.Error("Both input and output images are required to create a video.")
-
-    try:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
-            input_image.save(tmp.name)
-            input_image_path = tmp.name
-
-        output_pil = Image.fromarray(output_image.astype('uint8'))
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
-            output_pil.save(tmp.name)
-            output_image_path = tmp.name
-
-        video_path = _generate_video_segment(
-            input_image_path,
-            output_image_path,
-            prompt if prompt else "Camera movement transformation",
-            request
-        )
-        return video_path
-    except Exception as e:
-        raise gr.Error(f"Video generation failed: {e}")
+#     Raises:
+#         gr.Error:
+#             If either image is missing or if the video generation fails.
+#     """
+#     if input_image is None or output_image is None:
+#         raise gr.Error("Both input and output images are required to create a video.")
+
+#     try:
+#         with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
+#             input_image.save(tmp.name)
+#             input_image_path = tmp.name
+
+#         output_pil = Image.fromarray(output_image.astype('uint8'))
+#         with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
+#             output_pil.save(tmp.name)
+#             output_image_path = tmp.name
+
+#         video_path = _generate_video_segment(
+#             input_image_path,
+#             output_image_path,
+#             prompt if prompt else "Camera movement transformation",
+#             request
+#         )
+#         return video_path
+#     except Exception as e:
+#         raise gr.Error(f"Video generation failed: {e}")
 
 
 # --- UI ---
@@ -622,7 +622,7 @@ with gr.Blocks() as demo:
 
 #    run_event.then(lambda img, *_: img, inputs=[result], outputs=[prev_output])
 
-    gr.api(infer_camera_edit, api_name="infer_edit_camera_angles")
+#    gr.api(infer_camera_edit, api_name="infer_edit_camera_angles")
 #    gr.api(create_video_between_images, api_name="create_video_between_images")
 
-demo.launch(mcp_server=True, show_api=True)
+demo.launch(mcp_server=True)