Update app.py

app.py

@@ -28,11 +28,12 @@ torch.backends.cuda.preferred_linalg_library="cusolver"
 
 import cv2
 import gc
-
+import subprocess
+# --- NEW: Import for SFTP ---
+import paramiko
 from image_gen_aux import UpscaleWithModel
 
 import gradio as gr
-import numpy as np
 import random
 import yaml
 from pathlib import Path
@@ -43,7 +44,7 @@ from huggingface_hub import hf_hub_download
 import shutil
 
 from diffusers import StableDiffusionXLImg2ImgPipeline, AutoencoderKL
-
+import imageio
 from inference import (
     create_ltx_video_pipeline,
     create_latent_upsampler,
@@ -108,6 +109,27 @@ print(f"Target inference device: {target_inference_device}")
 pipeline_instance.to(target_inference_device)
 if latent_upsampler_instance: latent_upsampler_instance.to(target_inference_device)
 
+
+def upload_to_sftp(local_filepath):
+    if not all([FTP_HOST, FTP_USER, FTP_PASS, FTP_DIR]):
+        print("SFTP credentials not set. Skipping upload.")
+        return
+    try:
+        transport = paramiko.Transport((FTP_HOST, 22))
+        transport.connect(username=FTP_USER, password=FTP_PASS)
+        sftp = paramiko.SFTPClient.from_transport(transport)
+        remote_filename = os.path.basename(local_filepath)
+        remote_filepath = os.path.join(FTP_DIR, remote_filename)
+        print(f"Uploading {local_filepath} to {remote_filepath}...")
+        sftp.put(local_filepath, remote_filepath)
+        print("Upload successful.")
+        sftp.close()
+        transport.close()
+    except Exception as e:
+        print(f"SFTP upload failed: {e}")
+        gr.Warning(f"SFTP upload failed: {e}")
+
+        
 def calculate_new_dimensions(orig_w, orig_h):
     if orig_w == 0 or orig_h == 0: return int(768), int(768)
     if orig_w >= orig_h:
@@ -279,6 +301,7 @@ def generate(prompt, negative_prompt, clips_list, input_image_filepath, input_vi
         for idx, frame in enumerate(video_np):
             progress(idx / len(video_np), desc="Saving video clip...")
             video_writer.append_data(frame)
+    upload_to_sftp(output_path)
     updated_clips_list = clips_list + [output_video_path]
     counter_text = f"Clips created: {len(updated_clips_list)}"
     return output_video_path, seed_ui, gr.update(visible=True), updated_clips_list, counter_text