Update app.py

app.py

@@ -63,7 +63,44 @@ optimize_pipeline_(pipe,
 default_prompt_t2v = "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."
 default_negative_prompt = "色调艳丽, 过曝, 静态, 细节模糊不清, 字幕, 风格, 作品, 画作, 画面, 静止, 整体发灰, 最差质量, 低质量, JPEG压缩残留, 丑陋的, 残缺的, 多余的手指, 画得不好的手部, 画得不好的脸部, 畸形的, 毁容的, 形态畸形的肢体, 手指融合, 静止不动的画面, 杂乱的背景, 三条腿, 背景人很多, 倒着走"
 
+from huggingface_hub import HfApi, upload_file
+import os
+import uuid
+
+def upload_to_hf(video_path, summary_text):
+    api = HfApi()
+    unique_folder = f"WANT2Vvideo_{uuid.uuid4().hex[:8]}"
+    logging.info(f"Creating new HF folder: {unique_folder} in repo {HF_MODEL}")
+
+    # Upload video
+    video_filename = os.path.basename(video_path)
+    video_hf_path = f"{unique_folder}/{video_filename}"
+    upload_file(
+        path_or_fileobj=video_path,
+        path_in_repo=video_hf_path,
+        repo_id=HF_MODEL,
+        repo_type="model",  # Specify that it's a model repo
+        token=os.environ.get("HUGGINGFACE_HUB_TOKEN"),
+    )
+    logging.info(f"✅ Uploaded video to HF: {video_hf_path}")
+
+    # Upload summary.txt
+    summary_file = f"/tmp/summary.txt"
+    with open(summary_file, "w", encoding="utf-8") as f:
+        f.write(summary_text)
+
+    summary_hf_path = f"{unique_folder}/summary.txt"
+    upload_file(
+        path_or_fileobj=summary_file,
+        path_in_repo=summary_hf_path,
+        repo_id=HF_MODEL,
+        repo_type="model",  # Specify that it's a model repo
+        token=os.environ.get("HUGGINGFACE_HUB_TOKEN"),
+    )
+    logging.info(f"✅ Uploaded summary to HF: {summary_hf_path}")
+    return unique_folder
 
+    
 def get_duration(
     prompt,
     negative_prompt,
@@ -151,6 +188,7 @@ def generate_video(
         video_path = tmpfile.name
 
     export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
+    hf_folder = upload_to_hf(video_path, prompt)
 
     return video_path, current_seed