Update Modules/Generate_Video.py

Modules/Generate_Video.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
 
 import os
+import uuid
 import random
 import tempfile
 from typing import Annotated
 
 import gradio as gr
 from huggingface_hub import InferenceClient
+from .File_System import ROOT_DIR
 
 from app import _log_call_end, _log_call_start, _truncate_for_log
 from ._docstrings import autodoc
@@ -22,9 +24,10 @@ TOOL_SUMMARY = (
 
 
 def _write_video_tmp(data_iter_or_bytes: object, suffix: str = ".mp4") -> str:
-    fd, fname = tempfile.mkstemp(suffix=suffix)
+    filename = f"video_{uuid.uuid4().hex[:8]}{suffix}"
+    path = os.path.join(ROOT_DIR, filename)
     try:
-        with os.fdopen(fd, "wb") as file:
+        with open(path, "wb") as file:
             if isinstance(data_iter_or_bytes, (bytes, bytearray)):
                 file.write(data_iter_or_bytes)
             elif hasattr(data_iter_or_bytes, "read"):
@@ -39,11 +42,11 @@ def _write_video_tmp(data_iter_or_bytes: object, suffix: str = ".mp4") -> str:
                 raise gr.Error("Unsupported video data type returned by provider.")
     except Exception:
         try:
-            os.remove(fname)
+            os.remove(path)
         except Exception:
             pass
         raise
-    return fname
+    return path
 
 
 @autodoc(