Update app.py

app.py

@@ -56,10 +56,13 @@ optimize_pipeline_(pipe, image=[Image.new("RGB", (1024, 1024)), Image.new("RGB",
 
 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:
+def _generate_video_segment(input_image_path: str, output_image_path: str, prompt: str) -> str:
     """Generates a single video segment using the external service."""
-    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})
+    from gradio.context import LocalContext
+    request = LocalContext.request.get(None)
+    print(f"Request available in _generate_video_segment: {request is not None} {request}")    
+    print("x-ip", request.headers["x-ip-token"])    
+    video_client = Client("multimodalart/wan-2-2-first-last-frame")
     result = video_client.predict(
         start_image_pil=handle_file(input_image_path),
         end_image_pil=handle_file(output_image_path),
@@ -152,6 +155,11 @@ def infer_camera_edit(
 
 def create_video_between_images(input_image, output_image, prompt: str, request: gr.Request) -> str:
     """Create a video between the input and output images."""
+    from gradio.context import LocalContext
+    request = LocalContext.request.get(None)
+    print(f"Request available in create_video_between_images: {request is not None} {request}")    
+    print("x-ip", request.headers["x-ip-token"])    
+    
     if input_image is None or output_image is None:
         raise gr.Error("Both input and output images are required to create a video.")