Fix: Update Whisper package and add error handling

Dockerfile

@@ -2,6 +2,11 @@ FROM python:3.10-slim
 
 WORKDIR /code
 
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    && rm -rf /var/lib/apt/lists/*
+
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 

app.py

@@ -1,16 +1,41 @@
-from fastapi import FastAPI, UploadFile, File, Form
+from fastapi import FastAPI, UploadFile, File, Form, HTTPException
 from fastapi.responses import JSONResponse
-import whisper
+import openai.whisper as whisper
 import tempfile
 import os
 from transformers import pipeline
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI(title="TranscriptoCast AI (Demo)")
 
 # Load models once at startup
-whisper_model = whisper.load_model("base")
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-translator = pipeline("translation", model="facebook/mbart-large-50-many-to-many-mmt")
+try:
+    logger.info("Loading Whisper model...")
+    whisper_model = whisper.load_model("base")
+    logger.info("Whisper model loaded successfully")
+except Exception as e:
+    logger.error(f"Error loading Whisper model: {str(e)}")
+    raise HTTPException(status_code=500, detail="Failed to load Whisper model")
+
+try:
+    logger.info("Loading summarization model...")
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    logger.info("Summarization model loaded successfully")
+except Exception as e:
+    logger.error(f"Error loading summarization model: {str(e)}")
+    raise HTTPException(status_code=500, detail="Failed to load summarization model")
+
+try:
+    logger.info("Loading translation model...")
+    translator = pipeline("translation", model="facebook/mbart-large-50-many-to-many-mmt")
+    logger.info("Translation model loaded successfully")
+except Exception as e:
+    logger.error(f"Error loading translation model: {str(e)}")
+    raise HTTPException(status_code=500, detail="Failed to load translation model")
 
 @app.post("/transcribe")
 async def transcribe(file: UploadFile = File(...)):

requirements.txt

@@ -1,6 +1,6 @@
 fastapi
 uvicorn
-whisper
+openai-whisper
 torch
 transformers
 pydantic