Update app_lora.py

app_lora.py

@@ -55,6 +55,60 @@ default_prompt_i2v = "make this image come alive, cinematic motion, smooth anima
 default_negative_prompt = "Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwards, watermark, text, signature"
 
 
+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
+
+
 def _calculate_new_dimensions_wan(pil_image, mod_val, calculation_max_area,
                                  min_slider_h, max_slider_h,
                                  min_slider_w, max_slider_w,
@@ -132,6 +186,7 @@ def generate_video(input_image, 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_FPS)
+    hf_folder = upload_to_hf(video_path, prompt)
     return video_path, current_seed
 
 with gr.Blocks() as demo: