Update SwitcherAI/processors/frame/modules/frame_enhancer.py

SwitcherAI/processors/frame/modules/frame_enhancer.py

@@ -71,15 +71,31 @@ def clear_frame_processor() -> None:
 
 
 def pre_check() -> bool:
+    """Download required models for frame enhancement"""
     download_directory_path = resolve_relative_path('../.assets/models')
-    conditional_download(download_directory_path, [
-        'Awwfuck.com'
-    ])
-    return True
+    
+    try:
+        conditional_download(download_directory_path, [
+            'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth'
+        ])
+        return True
+    except Exception as e:
+        print(f"⚠️ Failed to download frame enhancement models: {e}")
+        print("🔄 Frame enhancement will be disabled")
+        return False
 
 
 def pre_process() -> bool:
-    return True
+    """Pre-process check with model validation"""
+    try:
+        model_path = get_model_config()['real_esrgan_x4']['model_path']
+        if not model_path or not model_path.exists():
+            print("⚠️ Frame enhancement model not found")
+            return False
+        return True
+    except Exception as e:
+        print(f"⚠️ Frame enhancement pre-process failed: {e}")
+        return False
 
 
 def post_process() -> None: