refact: added request id

handler.py

@@ -7,6 +7,7 @@ import tempfile
 import gc
 import time
 import threading
+import uuid
 from transformers import VideoLlavaProcessor, VideoLlavaForConditionalGeneration
 
 class EndpointHandler:
@@ -93,7 +94,7 @@ class EndpointHandler:
             # We print the error but do NOT raise it, ensuring the user still gets their result
             print(f"Webhook failed: {str(e)}")
 
-    def _process_video(self, inputs, video_url, parameters, callback_url=None):
+    def _process_video(self, inputs, video_url, parameters, callback_url=None, request_id=None):
         """
         Core video processing logic. Used by both sync and async paths.
         If callback_url is provided, sends result via webhook.
@@ -188,6 +189,7 @@ class EndpointHandler:
             # 9. SEND WEBHOOK (if callback_url provided)
             if callback_url:
                 webhook_data = {
+                    "request_id": request_id,
                     "input_prompt": inputs,
                     "video_url": video_url,
                     "result": response_payload
@@ -206,6 +208,7 @@ class EndpointHandler:
             # Send error via webhook if callback_url provided
             if callback_url:
                 webhook_data = {
+                    "request_id": request_id,
                     "input_prompt": inputs,
                     "video_url": video_url,
                     "result": error_payload
@@ -229,24 +232,28 @@ class EndpointHandler:
         video_url = data.get("video", None)
         parameters = data.get("parameters", {})
         
+        # Generate unique request ID
+        request_id = str(uuid.uuid4())
+        
         # Validation
         if not video_url:
-            return {"error": "Missing 'video' URL.", "status": "failed"}
+            return {"error": "Missing 'video' URL.", "status": "failed", "request_id": request_id}
 
         # --- ASYNC MODE: Return early, process in background ---
         if callback_url:
-            print(f"Async mode: will send result to {callback_url}")
+            print(f"Async mode: request_id={request_id}, will send result to {callback_url}")
             
             # Spawn background thread for processing
             thread = threading.Thread(
                 target=self._process_video,
-                args=(inputs, video_url, parameters, callback_url),
+                args=(inputs, video_url, parameters, callback_url, request_id),
                 daemon=True  # Daemon thread won't block process exit
             )
             thread.start()
             
             # Return immediately with acknowledgment
             return [{
+                "request_id": request_id,
                 "status": "accepted",
                 "message": "Processing started. Result will be sent to callback_url.",
                 "callback_url": callback_url
@@ -254,5 +261,6 @@ class EndpointHandler:
         
         # --- SYNC MODE: Process and return result ---
         else:
-            result = self._process_video(inputs, video_url, parameters)
+            result = self._process_video(inputs, video_url, parameters, request_id=request_id)
+            result["request_id"] = request_id
             return [result]