Fix missing dependencies

- Add torchdiffeq and other missing packages to requirements.txt
- Update dependency versions to match hf_AC requirements
- Add better error handling and dependency checking in app.py
- Include nitrous-ema, hydra_colorlog, and python-dotenv

app.py

@@ -36,8 +36,36 @@ USAGE_TIPS = """
 4. **CFG强度**: 数值越高越贴合提示词，但可能降低质量
 """
 
+# Check and install missing dependencies
+def check_dependencies():
+    """Check if all required packages are available"""
+    missing_packages = []
+    required_packages = [
+        'torch', 'torchaudio', 'numpy', 'scipy', 'librosa', 
+        'torchdiffeq', 'einops', 'hydra', 'tensordict', 'av'
+    ]
+    
+    for package in required_packages:
+        try:
+            if package == 'hydra':
+                __import__('hydra')
+            elif package == 'av':
+                __import__('av')
+            else:
+                __import__(package)
+        except ImportError:
+            missing_packages.append(package)
+    
+    return missing_packages
+
 # Import hf_AC modules with error handling
 try:
+    # First check basic dependencies
+    missing_deps = check_dependencies()
+    if missing_deps:
+        print(f"Warning: Missing dependencies: {missing_deps}")
+        print("Some dependencies may be installing in the background...")
+    
     from hf_AC.mmaudio.eval_utils import (ModelConfig, all_model_cfg, generate, load_video, 
                                           setup_eval_logging)
     from hf_AC.mmaudio.model.flow_matching import FlowMatching
@@ -48,8 +76,11 @@ try:
     setup_eval_logging()
     log = logging.getLogger()
     HF_AC_AVAILABLE = True
+    print("✅ hf_AC modules loaded successfully!")
+    
 except ImportError as e:
     print(f"Warning: hf_AC modules not available: {e}")
+    print("This may be due to missing dependencies. Please wait for installation to complete.")
     log = logging.getLogger()
     HF_AC_AVAILABLE = False
 

requirements.txt

@@ -2,18 +2,25 @@ torch>=2.0.0
 torchvision
 torchaudio
 gradio>=4.0.0
-huggingface_hub>=0.20.0
-numpy>=1.21.0
-Pillow>=9.0.0
-opencv-python>=4.6.0
+huggingface_hub>=0.26.0
+numpy>=1.21.0,<2.1
+Pillow>=9.5.0
+opencv-python>=4.8.0
 scipy>=1.7.0
-tqdm>=4.60.0
+tqdm>=4.66.1
 einops>=0.6.0
 requests
 librosa>=0.8.1
-av>=10.0.0
-timm>=0.9.0
-open_clip_torch>=2.20.0
-tensordict>=0.3.0
-hydra-core>=1.3.0
+av>=14.0.1
+timm>=1.0.12
+open_clip_torch>=2.29.0
+tensordict>=0.6.1
+hydra-core>=1.3.2
 colorlog
+torchdiffeq>=0.2.5
+cython
+gitpython>=3.1.0
+tensorboard>=2.11.0
+nitrous-ema
+hydra_colorlog
+python-dotenv