Update main.py

main.py

@@ -1,10 +1,10 @@
+from flask_cors import CORS
 from flask import Flask, request, jsonify
 import tempfile
 import os
 from werkzeug.utils import secure_filename
 import logging
 from datetime import datetime
-from flask_cors import CORS
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -25,23 +25,69 @@ def load_whisper_model():
     """Load Whisper model with proper error handling"""
     global model
     try:
-        # Try importing openai-whisper first
-        import whisper
+        # Try multiple import strategies for openai-whisper
+        whisper_module = None
+        
+        # Strategy 1: Direct import (most common)
+        try:
+            import whisper as whisper_module
+        except ImportError:
+            pass
+        
+        # Strategy 2: Try importing as openai_whisper
+        if whisper_module is None:
+            try:
+                import openai_whisper as whisper_module
+            except ImportError:
+                pass
+        
+        # Strategy 3: Try importing with explicit path
+        if whisper_module is None:
+            try:
+                import sys
+                import importlib.util
+                # This is a fallback - usually not needed
+                import whisper as whisper_module
+            except ImportError:
+                pass
+        
+        if whisper_module is None:
+            logger.error("OpenAI Whisper not installed. Install with: pip install openai-whisper")
+            return False
+        
+        # Check if the module has the load_model function
+        if not hasattr(whisper_module, 'load_model'):
+            logger.error("Whisper module found but missing 'load_model' function")
+            logger.error("This suggests you have the wrong 'whisper' package installed")
+            logger.error("Solution:")
+            logger.error("1. pip uninstall whisper")
+            logger.error("2. pip uninstall openai-whisper (if exists)")
+            logger.error("3. pip install openai-whisper")
+            logger.error("4. pip install torch torchaudio")
+            return False
+        
         logger.info(f"Loading Whisper model: {MODEL_SIZE}")
-        model = whisper.load_model(MODEL_SIZE)
+        model = whisper_module.load_model(MODEL_SIZE)
         logger.info("Whisper model loaded successfully")
         return True
-    except ImportError:
-        logger.error("OpenAI Whisper not installed. Install with: pip install openai-whisper")
+        
+    except ImportError as e:
+        logger.error(f"Import error: {e}")
+        logger.error("OpenAI Whisper not installed. Install with: pip install openai-whisper torch torchaudio")
         return False
     except AttributeError as e:
         logger.error(f"Whisper import error: {e}")
         logger.error("Make sure you have the correct whisper package installed:")
-        logger.error("pip uninstall whisper")
-        logger.error("pip install openai-whisper")
+        logger.error("Solution:")
+        logger.error("1. pip uninstall whisper")
+        logger.error("2. pip install openai-whisper torch torchaudio")
         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")
         return False
 
 # Try to load the model at startup