Update app.py

app.py

@@ -10,6 +10,7 @@ from diffusers import AnimateDiffPipeline, EulerDiscreteScheduler
 from diffusers.utils import export_to_video
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
+from tenacity import retry, stop_after_attempt, wait_fixed, retry_if_exception_type
 
 # Thiết lập logging
 logging.basicConfig(level=logging.INFO)
@@ -51,6 +52,18 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(
 )
 pipe.safety_checker = None
 
+# Hàm tải lên Filebin với retry
+@retry(
+    stop=stop_after_attempt(3),  # Thử lại 3 lần
+    wait=wait_fixed(2),  # Chờ 2 giây giữa các lần thử
+    retry=retry_if_exception_type(requests.exceptions.RequestException)  # Thử lại nếu gặp lỗi mạng
+)
+def upload_to_filebin(file_path, file_name):
+    with open(file_path, "rb") as f:
+        response = requests.post(f"https://filebin.net/{uuid.uuid4()}/{file_name}", files={"file": (file_name, f)})
+        response.raise_for_status()
+        return response.json().get("url")  # Lấy URL công khai từ phản hồi JSON
+
 # Hàm tạo video
 def generate_image(prompt, base="Realistic", motion="", step=1, progress=gr.Progress()):
     global step_loaded, base_loaded, motion_loaded
@@ -121,16 +134,13 @@ def generate_image(prompt, base="Realistic", motion="", step=1, progress=gr.Prog
 
         logger.info(f"✅ Video sẵn sàng tại {video_path}")
 
-        # Tải video lên transfer.sh để lấy URL công khai
+        # Tải video lên Filebin để lấy URL công khai
         try:
-            with open(video_path, "rb") as f:
-                response = requests.put(f"https://transfer.sh/{name}.mp4", data=f)
-                response.raise_for_status()
-                public_url = response.text.strip()
-                logger.info(f"✅ Public URL: {public_url}")
-                return public_url  # Trả về URL công khai
+            public_url = upload_to_filebin(video_path, f"{name}.mp4")
+            logger.info(f"✅ Public URL: {public_url}")
+            return public_url  # Trả về URL công khai
         except Exception as e:
-            logger.error(f"❌ Không thể tải lên transfer.sh: {e}")
+            logger.error(f"❌ Không thể tải lên Filebin: {e}")
             # Fallback: Trả về đường dẫn tệp để Gradio hiển thị
             return video_path