Update app.py

app.py

@@ -1,43 +1,30 @@
 from flask import Flask, request, jsonify
-
 import whisper
-import os
-import tempfile
 import io
-import torchaudio
 
 app = Flask(__name__)
 
-# Initialize Whisper model
-whisper_model = whisper.load_model("small")  # Renamed variable
-
-
-
+# Load Whisper model once for efficiency
+model = whisper.load_model("small")
 
 @app.route('/transcribe', methods=['POST'])
 def transcribe():
     try:
-        # Read raw bytes from the request
+        # Read raw bytes from request
         audio_bytes = request.data  
         if not audio_bytes:
             return jsonify({"error": "No audio data provided"}), 400
 
-        # Convert bytes to a file-like object
-        audio_file = io.BytesIO(audio_bytes)
+        # Save bytes to a temporary file (required for Whisper)
+        temp_audio_path = "temp_audio.wav"
+        with open(temp_audio_path, "wb") as f:
+            f.write(audio_bytes)
 
-        # Load audio as a waveform using torchaudio
-        waveform, sample_rate = torchaudio.load(audio_file)
-
-        # Whisper expects a NumPy array, so we convert it
-        audio_numpy = waveform.squeeze().numpy()
-
-        # Transcribe the audio
-        result = model.transcribe(audio_numpy)
+        # Transcribe using Whisper
+        result = model.transcribe(temp_audio_path)
 
         return jsonify({"text": result["text"]})
 
     except Exception as e:
         print("Error:", str(e))  # Log error for debugging
-        return jsonify({"error": "Internal Server Error", "details": str(e)}), 500
-
-  
+        return jsonify({"error": "Internal Server Error", "details": str(e)}), 500