Fix build errors: Update Gradio to 5.8.0, remove problematic dependencies, add graceful error handling

requirements.txt

@@ -15,10 +15,11 @@ ftfy
 dashscope
 imageio-ffmpeg
 
+# Remove flash-attn for now as it causes build issues on HF Spaces
 # flash_attn
-flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
+# flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
-gradio==5.33.0   
+gradio==5.8.0
 numpy>=1.23.5,<2
 einops
 moviepy==1.0.3
@@ -27,8 +28,10 @@ peft==0.14.0
 mutagen
 pydantic==2.10.6
 decord
-onnxruntime-gpu
-rembg[gpu]==2.0.65
+# Use CPU version for initial testing
+onnxruntime
+# rembg[gpu]==2.0.65
+rembg==2.0.65
 matplotlib
 timm
 segment-anything

wgp.py

@@ -17,7 +17,17 @@ from mmgp import offload, safetensors2, profile_type
 try:
     import triton
 except ImportError:
+    triton = None
+    print("Triton not available, will use fallback implementations")
     pass
+
+# Handle flash_attn import gracefully
+try:
+    import flash_attn
+    FLASH_ATTN_AVAILABLE = True
+except ImportError:
+    FLASH_ATTN_AVAILABLE = False
+    print("Flash Attention not available, will use standard attention")
 from pathlib import Path
 from datetime import datetime
 import gradio as gr
@@ -56,11 +66,21 @@ target_mmgp_version = "3.4.8"
 WanGP_version = "5.41"
 prompt_enhancer_image_caption_model, prompt_enhancer_image_caption_processor, prompt_enhancer_llm_model, prompt_enhancer_llm_tokenizer = None, None, None, None
 
+# Check for GPU availability
+CUDA_AVAILABLE = torch.cuda.is_available()
+print(f"CUDA available: {CUDA_AVAILABLE}")
+if not CUDA_AVAILABLE:
+    print("Running on CPU mode - video generation will be significantly slower")
+
 from importlib.metadata import version
-mmgp_version = version("mmgp")
-if mmgp_version != target_mmgp_version:
-    print(f"Incorrect version of mmgp ({mmgp_version}), version {target_mmgp_version} is needed. Please upgrade with the command 'pip install -r requirements.txt'")
-    exit()
+try:
+    mmgp_version = version("mmgp")
+    if mmgp_version != target_mmgp_version:
+        print(f"Warning: mmgp version {mmgp_version} detected, version {target_mmgp_version} is recommended.")
+        print("Some features may not work correctly. Consider upgrading with 'pip install -r requirements.txt'")
+except Exception as e:
+    print(f"Warning: Could not check mmgp version: {e}")
+    print("This may cause issues with some features.")
 lock = threading.Lock()
 current_task_id = None
 task_id = 0