Update main.py

main.py

@@ -19,12 +19,29 @@ ALLOWED_EXTENSIONS = {'wav', 'mp3', 'mp4', 'mpeg', 'mpga', 'm4a', 'webm', 'flac'
 
 # Initialize model variable
 model = None
-MODEL_SIZE = "base"  # Change this to your preferred model size
+MODEL_SIZE = "tiny"  # Use 'tiny' for faster loading on Hugging Face Spaces, change to 'base', 'small', 'medium', or 'large' as needed
+
+# Set cache directory for Hugging Face Spaces
+def setup_cache_directory():
+    """Setup cache directory for Hugging Face Spaces"""
+    # Create a writable cache directory in the current working directory
+    cache_dir = os.path.join(os.getcwd(), ".whisper_cache")
+    os.makedirs(cache_dir, exist_ok=True)
+    
+    # Set environment variables for Whisper cache
+    os.environ['XDG_CACHE_HOME'] = cache_dir
+    os.environ['WHISPER_CACHE'] = cache_dir
+    
+    logger.info(f"Cache directory set to: {cache_dir}")
+    return cache_dir
 
 def load_whisper_model():
     """Load Whisper model with proper error handling"""
     global model
     try:
+        # Setup cache directory first
+        cache_dir = setup_cache_directory()
+        
         # Try multiple import strategies for openai-whisper
         whisper_module = None
         
@@ -67,7 +84,15 @@ def load_whisper_model():
             return False
         
         logger.info(f"Loading Whisper model: {MODEL_SIZE}")
-        model = whisper_module.load_model(MODEL_SIZE)
+        logger.info(f"Using cache directory: {cache_dir}")
+        
+        # Load model with explicit download root
+        try:
+            model = whisper_module.load_model(MODEL_SIZE, download_root=cache_dir)
+        except TypeError:
+            # Fallback if download_root parameter is not supported
+            model = whisper_module.load_model(MODEL_SIZE)
+        
         logger.info("Whisper model loaded successfully")
         return True
         
@@ -82,12 +107,42 @@ def load_whisper_model():
         logger.error("1. pip uninstall whisper")
         logger.error("2. pip install openai-whisper torch torchaudio")
         return False
+    except PermissionError as e:
+        logger.error(f"Permission error: {e}")
+        logger.error("Cannot write to cache directory. This might be a Hugging Face Spaces limitation.")
+        logger.error("Trying alternative cache locations...")
+        
+        # Try alternative cache locations
+        alternative_dirs = [
+            "/tmp/.whisper_cache",
+            os.path.expanduser("~/.whisper_cache"),
+            "./whisper_models"
+        ]
+        
+        for alt_dir in alternative_dirs:
+            try:
+                os.makedirs(alt_dir, exist_ok=True)
+                os.environ['XDG_CACHE_HOME'] = alt_dir
+                os.environ['WHISPER_CACHE'] = alt_dir
+                logger.info(f"Trying alternative cache: {alt_dir}")
+                
+                import whisper
+                model = whisper.load_model(MODEL_SIZE, download_root=alt_dir)
+                logger.info(f"Successfully loaded model with cache: {alt_dir}")
+                return True
+            except Exception as alt_e:
+                logger.warning(f"Alternative cache {alt_dir} failed: {alt_e}")
+                continue
+        
+        logger.error("All cache directory attempts failed")
+        return False
     except Exception as e:
         logger.error(f"Error loading Whisper model: {e}")
         logger.error("This could be due to:")
         logger.error("- Insufficient memory")
         logger.error("- Missing PyTorch/CUDA dependencies")
         logger.error("- Network issues downloading the model")
+        logger.error("- Hugging Face Spaces limitations")
         return False
 
 # Try to load the model at startup