Update run_generator.py

run_generator.py

@@ -11,20 +11,17 @@ import subprocess
 import random
 import argparse
 from typing import Sequence, Mapping, Any, Union
-import shutil # ✨ 복사본 생성을 위해 shutil 임포트
+import shutil
 
-# --- 0. 기본 설정 및 인수 파싱 ---\n",
+# --- 0. 기본 설정 및 인수 파싱 ---
 def parse_args():
     parser = argparse.ArgumentParser(description="ComfyUI Video Generation Script with All Controls from 1.py")
     parser.add_argument("--positive_prompt", type=str, required=True); parser.add_argument("--negative_prompt", type=str, required=True)
     parser.add_argument("--width", type=int, required=True); parser.add_argument("--height", type=int, required=True)
     parser.add_argument("--length", type=int, required=True); parser.add_argument("--upscale_ratio", type=float, required=True)
-
-    # ✨ 수정: CFG 인자 분리
     parser.add_argument("--steps", type=int, default=4)
-    parser.add_argument("--cfg_high", type=float, default=1.0) # High
-    parser.add_argument("--cfg_low", type=float, default=1.0)  # Low
-
+    parser.add_argument("--cfg_high", type=float, default=1.0)
+    parser.add_argument("--cfg_low", type=float, default=1.0)
     parser.add_argument("--sampler_name_high", type=str, default="euler"); parser.add_argument("--scheduler_high", type=str, default="simple")
     parser.add_argument("--sampler_name_low", type=str, default="euler"); parser.add_argument("--scheduler_low", type=str, default="simple")
     parser.add_argument("--noise_seed", type=int, default=-1); parser.add_argument("--split_point_percent", type=float, default=50.0)
@@ -103,10 +100,41 @@ def import_custom_nodes() -> None:
 def main():
     args = parse_args()
     print("🚀 동영상 생성을 시작합니다 (Full Control Mode, VRAM Optimized)...\n")
-    # ✨ SyntaxError 수정: f-string 끝의 불필요한 \ 제거
-    subprocess.run(f"rm -rf {COMFYUI_BASE_PATH}/output/*", shell=True, check=True)
-    os.makedirs(f"{COMFYUI_BASE_PATH}/output/temp", exist_ok=True); os.makedirs(f"{COMFYUI_BASE_PATH}/output/up", exist_ok=True)
+
+    # 🚨🚨🚨 수정된 부분 시작: .mp4, .mkv, .webm 파일은 제외하고 output 폴더 정리 🚨🚨🚨
+    output_dir = os.path.join(COMFYUI_BASE_PATH, 'output')
+    print(f"  - 이전 출력물 정리 중... (Output: {output_dir})")
+    
+    deleted_count = 0
+    try:
+        # output 폴더 내의 모든 파일과 폴더 목록을 가져옵니다.
+        for item_name in os.listdir(output_dir):
+            item_path = os.path.join(output_dir, item_name)
+            
+            # 💡 조건: 비디오 파일 확장자(.mp4, .mkv, .webm)는 삭제하지 않고 보존합니다.
+            if item_name.lower().endswith(('.mp4', '.mkv', '.webm')):
+                print(f"    - 🗄️ 비디오 파일 '{item_name}'은 보존합니다.")
+                continue
+            
+            # 파일 또는 링크인 경우 삭제
+            if os.path.isfile(item_path) or os.path.islink(item_path):
+                os.unlink(item_path)
+                deleted_count += 1
+            # 폴더인 경우 재귀적으로 삭제
+            elif os.path.isdir(item_path):
+                shutil.rmtree(item_path)
+                deleted_count += 1
+                
+        print(f"  ✅ 정리 완료. 보존된 비디오 외 {deleted_count}개의 항목이 삭제되었습니다.")
+    except Exception as e:
+        print(f"  ❌ 출력 폴더 정리 중 오류 발생: {e}")
+    # 🚨🚨🚨 수정된 부분 끝 🚨🚨🚨
+    
+    # 임시 폴더 재생성 (정리 과정에서 삭제되었을 수 있음)
+    os.makedirs(f"{COMFYUI_BASE_PATH}/output/temp", exist_ok=True); 
+    os.makedirs(f"{COMFYUI_BASE_PATH}/output/up", exist_ok=True)
     os.makedirs(f"{COMFYUI_BASE_PATH}/output/interpolated", exist_ok=True)
+    
     add_comfyui_directory_to_sys_path()
     try: from utils.extra_config import load_extra_path_config
     except ImportError: print("⚠️ ComfyUI의 extra_model_paths.yaml 로딩 실패 (무시하고 진행)"); load_extra_path_config = lambda x: None
@@ -126,10 +154,9 @@ def main():
         clipvisionencode=NODE_CLASS_MAPPINGS["CLIPVisionEncode"]()
         pathchsageattentionkj=NODE_CLASS_MAPPINGS["PathchSageAttentionKJ"]()
 
-        # --- ✨ 1단계: CLIP Vision 로직 추가 및 메모리 해제 ---\n",
+        # --- ✨ 1단계: CLIP Vision 로직 추가 및 메모리 해제 ---
         print("\n1단계: 데이터 로딩 및 초기 Latent 생성 중...");
         print(f"  - CLIP 로딩: {args.clip_name}");
-        # ✨ SyntaxError 수정: 불필요한 \ 제거
         cliploader_460 = cliploader.load_clip(clip_name=args.clip_name, type="wan", device="default");
         cliptextencode_462 = cliptextencode.encode(text=args.positive_prompt, clip=get_value_at_index(cliploader_460, 0));
         cliptextencode_463 = cliptextencode.encode(text=args.negative_prompt, clip=get_value_at_index(cliploader_460, 0));
@@ -137,7 +164,7 @@ def main():
 
         imageresizekjv2_401 = imageresizekjv2.resize(
             width=args.width, height=args.height,
-            upscale_method=args.input_resize_algo, # ✨ 변경됨
+            upscale_method=args.input_resize_algo,
             image=get_value_at_index(loadimage_88, 0), keep_proportion="crop",
             pad_color="0, 0, 0", crop_position="center", divisible_by=2,
             unique_id=random.randint(1, 2**64)
@@ -149,7 +176,7 @@ def main():
         clipvisionencode_cv = clipvisionencode.encode(
             crop="none",
             clip_vision=get_value_at_index(clipvisionloader_cv, 0),
-            image=get_value_at_index(imageresizekjv2_401, 0) # 리사이즈된 이미지 사용
+            image=get_value_at_index(imageresizekjv2_401, 0)
         );
         clip_vision_output = get_value_at_index(clipvisionencode_cv, 0)
 
@@ -164,7 +191,7 @@ def main():
             positive=get_value_at_index(cliptextencode_462, 0),
             negative=get_value_at_index(cliptextencode_463, 0),
             vae=get_value_at_index(vaeloader_temp, 0),
-            clip_vision_output=clip_vision_output, # ✨ 추가된 인수
+            clip_vision_output=clip_vision_output,
             start_image=get_value_at_index(imageresizekjv2_401, 0)
         );
 
@@ -183,7 +210,7 @@ def main():
         if args.lora_high_2_name != "None": print(f"  - H LoRA 2: {args.lora_high_2_name}"); model_for_patching, clip = loraloader.load_lora(lora_name=args.lora_high_2_name, strength_model=args.lora_high_2_strength_model, strength_clip=args.lora_high_2_strength_clip, model=model_for_patching, clip=clip)
         shifted_model = get_value_at_index(modelsamplingsd3.patch(shift=args.shift, model=model_for_patching), 0); final_model = shifted_model;
 
-        # ✨ 수정: cfg=args.cfg_high 사용
+        # 수정: cfg=args.cfg_high 사용
         ksampleradvanced_466 = ksampleradvanced.sample(add_noise="enable", noise_seed=final_seed, steps=args.steps, cfg=args.cfg_high, sampler_name=args.sampler_name_high, scheduler=args.scheduler_high, start_at_step=0, end_at_step=split_step, return_with_leftover_noise="enable", model=final_model, positive=get_value_at_index(wanimagetovideo_464, 0), negative=get_value_at_index(wanimagetovideo_464, 1), latent_image=get_value_at_index(wanimagetovideo_464, 2));
 
         if to_bool(args.sageattention): del pathchsageattentionkj_124
@@ -195,7 +222,7 @@ def main():
         if args.lora_low_2_name != "None": print(f"  - L LoRA 2: {args.lora_low_2_name}"); model_for_patching, clip = loraloader.load_lora(lora_name=args.lora_low_2_name, strength_model=args.lora_low_2_strength_model, strength_clip=args.lora_low_2_strength_clip, model=model_for_patching, clip=clip)
         shifted_model = get_value_at_index(modelsamplingsd3.patch(shift=args.shift, model=model_for_patching), 0); final_model = shifted_model;
 
-        # ✨ 수정: cfg=args.cfg_low 사용
+        # 수정: cfg=args.cfg_low 사용
         ksampleradvanced_465 = ksampleradvanced.sample(add_noise="disable", noise_seed=final_seed, steps=args.steps, cfg=args.cfg_low, sampler_name=args.sampler_name_low, scheduler=args.scheduler_low, start_at_step=split_step, end_at_step=10000, return_with_leftover_noise="disable", model=final_model, positive=get_value_at_index(wanimagetovideo_464, 0), negative=get_value_at_index(wanimagetovideo_464, 1), latent_image=get_value_at_index(ksampleradvanced_466, 0));
 
         if to_bool(args.sageattention): del pathchsageattentionkj_129
@@ -228,7 +255,7 @@ def main():
                     if loaded_images is None: print("    - (경고) 건너뛸 수 없는 이미지가 로드되었습니다, 이 배치를 건너뜁니다."); continue
                     imageupscale_chunk = imageupscalewithmodel.upscale(upscale_model=get_value_at_index(upscalemodelloader_384, 0), image=loaded_images);
                     imagescale_chunk = imagescaleby.upscale(
-                        upscale_method=args.output_resize_algo, # ✨ 변경됨
+                        upscale_method=args.output_resize_algo,
                         scale_by=scale_by_ratio,
                         image=get_value_at_index(imageupscale_chunk, 0)
                     );
@@ -237,7 +264,7 @@ def main():
                 del upscalemodelloader_384; clear_memory(); combine_input_dir_for_ffmpeg = f"{COMFYUI_BASE_PATH}/output/up"; print("5단계 완료.")
         else: print("\n5단계: 업스케일링 건너뜀 (비율 1.0).")
 
-        # --- ✨ 6단계: RIFE 청크 로직 수정 (Overlap 적용) ---\n",
+        # --- ✨ 6단계: RIFE 청크 로직 수정 (Overlap 적용) ---
         print("\n6단계: 비디오 결합 준비 중..."); final_frame_rate = float(args.frame_rate); ffmpeg_input_dir = combine_input_dir_for_ffmpeg
         if to_bool(args.interpolation):
             print("  - 프레임 보간 (RIFE)을 활성화합니다."); interpolated_dir = f"{COMFYUI_BASE_PATH}/output/interpolated"; source_dir = combine_input_dir_for_ffmpeg
@@ -277,7 +304,7 @@ def main():
 
                 if loaded_images is None:
                     print("    - (경고) 건너뛸 수 없는 이미지가 로드되었습니다, 이 배치를 ���너뜁니다.");
-                    current_frame_idx += chunk_size # 다음 청크로 이동
+                    current_frame_idx += chunk_size
                     is_first_chunk = False
                     continue
 
@@ -291,7 +318,7 @@ def main():
 
                 images_to_save = rife_chunk_result_tensor
                 if not is_first_chunk:
-                    # ✨ 첫 번째가 아닌 모든 청크는 겹치는 첫 프레임을 제거 (텐서 슬라이싱)
+                    # 첫 번째가 아닌 모든 청크는 겹치는 첫 프레임을 제거 (텐서 슬라이싱)
                     print(f"    - (Overlap) 중복 프레임 1개 제거 후 저장")
                     images_to_save = rife_chunk_result_tensor[1:]
 
@@ -307,7 +334,7 @@ def main():
         # --- 6단계 수정 완료 ---
 
         print(f"  - 최종 비디오를 FFmpeg ({args.video_encoder})로 결합합니다..."); print(f"  - 입력 폴더: '{ffmpeg_input_dir}'")
-        input_pattern = os.path.join(ffmpeg_input_dir, "example_%05d_.png") # 파일명 패턴 수정됨
+        input_pattern = os.path.join(ffmpeg_input_dir, "example_%05d_.png")
         timestamp = time.strftime("%Y%m%d-%H%M%S"); output_filename = f"AnimateDiff_{timestamp}.mp4"; output_path = os.path.join(COMFYUI_BASE_PATH, "output", output_filename)
         ffmpeg_cmd = ["ffmpeg", "-framerate", str(final_frame_rate), "-i", input_pattern]
         encoder_choice = args.video_encoder
@@ -328,7 +355,7 @@ def main():
         except Exception as e: print(f"  ❌ 오류: FFmpeg 실행 중 예상치 못한 오류 발생: {e}"); raise
         print("✅ 모든 단계 완료.")
 
-        # --- ✨ UnboundLocalError 해결 및 복사 로직 (최종) ---\n",
+        # --- ✨ UnboundLocalError 해결 및 복사 로직 (최종) ---
         latest_video = None
 
         if os.path.exists(output_path):
@@ -360,3 +387,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+