Update wan2_fast.py

wan2_fast.py

@@ -36,7 +36,58 @@ MODEL_ID = "FastVideo/FastWan2.2-TI2V-5B-FullAttn-Diffusers"
 
 
 
+HF_MODEL = os.environ.get("HF_UPLOAD_REPO", "rahul7star/VideoExplain")
+from huggingface_hub import HfApi, upload_file
+import os
+import uuid
+import logging
+
+import os
+import uuid
+import logging
+from datetime import datetime
+from huggingface_hub import HfApi, upload_file
+
+def upload_to_hf(video_path, summary_text):
+    api = HfApi()
+    
+    # Create a date-based folder (YYYY-MM-DD)
+    today_str = datetime.now().strftime("%Y-%m-%d")
+    date_folder = today_str
+    
+    # Generate a unique subfolder for this upload
+    unique_subfolder = f"Wan21-I2V-upload_{uuid.uuid4().hex[:8]}"
+    hf_folder = f"{date_folder}/{unique_subfolder}"
+    logging.info(f"Uploading files to HF folder: {hf_folder} in repo {HF_MODEL}")
+
+    # Upload video
+    video_filename = os.path.basename(video_path)
+    video_hf_path = f"{hf_folder}/{video_filename}"
+    upload_file(
+        path_or_fileobj=video_path,
+        path_in_repo=video_hf_path,
+        repo_id=HF_MODEL,
+        repo_type="model",
+        token=os.environ.get("HUGGINGFACE_HUB_TOKEN"),
+    )
+    logging.info(f"✅ Uploaded video to HF: {video_hf_path}")
+
+    # Upload summary.txt
+    summary_file = "/tmp/summary.txt"
+    with open(summary_file, "w", encoding="utf-8") as f:
+        f.write(summary_text)
+
+    summary_hf_path = f"{hf_folder}/summary.txt"
+    upload_file(
+        path_or_fileobj=summary_file,
+        path_in_repo=summary_hf_path,
+        repo_id=HF_MODEL,
+        repo_type="model",
+        token=os.environ.get("HUGGINGFACE_HUB_TOKEN"),
+    )
+    logging.info(f"✅ Uploaded summary to HF: {summary_hf_path}")
 
+    return hf_folder
 
 
 
@@ -281,6 +332,7 @@ def generate_video(prompt, height, width,
     with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
         video_path = tmpfile.name
     export_to_video(output_frames_list, video_path, fps=FIXED_OUTPUT_FPS)
+    hf_folder = upload_to_hf(video_path, prompt)
     return video_path, current_seed