Add automatic model weights download from HuggingFace

- Automatically download model.pth from FF2416/AC-Foley repo
- Create weights directory if not exists
- Better error handling for missing weights
- Improved status messages

app.py

@@ -117,6 +117,30 @@ class AudioFoleyModel:
             except Exception as e:
                 log.warning(f"Could not download model components: {e}")
             
+            # Try to download main model weights from HuggingFace
+            if not hasattr(self.model, 'model_path') or not self.model.model_path or not Path(self.model.model_path).exists():
+                try:
+                    from huggingface_hub import hf_hub_download
+                    log.info("Downloading main model weights from HuggingFace...")
+                    
+                    # Create weights directory
+                    weights_dir = Path("weights")
+                    weights_dir.mkdir(exist_ok=True)
+                    
+                    # Download model.pth from HuggingFace
+                    model_file = hf_hub_download(
+                        repo_id="FF2416/AC-Foley",
+                        filename="model.pth",
+                        local_dir=str(weights_dir),
+                        local_dir_use_symlinks=False
+                    )
+                    self.model.model_path = Path(model_file)
+                    log.info(f"✅ Downloaded model weights to {model_file}")
+                    
+                except Exception as e:
+                    log.warning(f"Could not download main model weights: {e}")
+                    log.info("Will proceed with available components only")
+            
             # Set custom model path if provided
             if model_path and os.path.exists(model_path):
                 self.model.model_path = Path(model_path)
@@ -136,7 +160,7 @@ class AudioFoleyModel:
                     return f"❌ Failed to load model weights: {e}"
             else:
                 log.warning('⚠️ No model weights found, using default initialization')
-                return "⚠️ Model loaded but no weights found. Download model.pth from HuggingFace."
+                return "⚠️ Model components loaded, but main weights not available. Some features may be limited."
             
             # Initialize flow matching
             self.fm = FlowMatching(min_sigma=0, inference_mode='euler', num_steps=25)