Update main.py

main.py

@@ -13,9 +13,9 @@ def hello():
     return "Semabox, listens to you!"
     
 # Load the Whisper model
-print("Loading Whisper model...", flush=True)
+print("Loading Whisper model...\n", flush=True)
 model = whisper.load_model("tiny")
-print("Whisper model loaded.", flush=True)
+print("Whisper model loaded.\n", flush=True)
 
 # Get time of request
 def get_time():
@@ -31,36 +31,36 @@ def get_time():
 
     
 def transcribe(audio_path):
-    print(f"Transcribing audio from: {audio_path}", flush=True)
+    print(f"Transcribing audio from: {audio_path}\n", flush=True)
 
     # Load audio and pad/trim it to fit 30 seconds
-    print("Loading and processing audio...", flush=True)
+    print("Loading and processing audio...\n", flush=True)
     audio = whisper.load_audio(audio_path)
     audio = whisper.pad_or_trim(audio)
 
     # Make log-Mel spectrogram and move to the same device as the model
-    print("Creating log-Mel spectrogram...", flush=True)
+    print("Creating log-Mel spectrogram...\n", flush=True)
     mel = whisper.log_mel_spectrogram(audio).to(model.device)
 
     # Detect the spoken language
     print("Detecting language...", flush=True)
     _, probs = model.detect_language(mel)
     language = max(probs, key=probs.get)
-    print(f"Detected language: {language}", flush=True)
+    print(f"Detected language: {language}\n", flush=True)
 
     # Decode the audio
     print("Decoding audio...", flush=True)
     options = whisper.DecodingOptions(fp16=False)
     result = whisper.decode(model, mel, options)
     
-    print("Transcription complete.", flush=True)
+    print("Transcription complete.\n", flush=True)
     return result.text, language
 
 @app.route('/transcribe', methods=['POST'])
 def transcribe_audio():
     # Record the time when the request was received
     request_received_time, _ = get_time()
-    print(f"Received request at /transcribe at {request_received_time}", flush=True)
+    print(f"Received request at /transcribe at {request_received_time}\n", flush=True)
 
     if 'audio' not in request.files:
         print("Error: No audio file provided", flush=True)
@@ -74,7 +74,7 @@ def transcribe_audio():
     audio_path = os.path.join("temp_audio", audio_file.filename)
     os.makedirs("temp_audio", exist_ok=True)
     audio_file.save(audio_path)
-    print(f"Audio file saved to: {audio_path} (Size: {audio_file_size} bytes)", flush=True)
+    print(f"Audio file saved to: {audio_path} (Size: {audio_file_size} bytes)\n", flush=True)
     
     # Record the time before starting transcription
     transcription_start_time = time.time()
@@ -92,13 +92,13 @@ def transcribe_audio():
     
     # Clean up the saved file
     os.remove(audio_path)
-    print(f"Audio file removed from: {audio_path}", flush=True)
+    print(f"Audio file removed from: {audio_path}\n", flush=True)
     
     # Record the time when the response is being sent
     response_sent_time, _ = get_time()
 
     # Return the transcription, detected language, and timing information
-    print(f"Transcription: {transcription}, Language: {language}", flush=True)
+    print(f"Transcription: {transcription}, Language: {language}\n", flush=True)
     return jsonify({
         "transcription": transcription,
         "language": language,
@@ -110,9 +110,9 @@ def transcribe_audio():
 
 @app.route('/healthcheck', methods=['GET'])
 def healthcheck():
-    print("Received request at /healthcheck", flush=True)
+    print("Received request at /healthcheck\n", flush=True)
     return jsonify({"status": "API is running"}), 200
 
 if __name__ == '__main__':
-    print("Starting Flask app...", flush=True)
+    print("Starting Flask app...\n", flush=True)
     app.run(host="0.0.0.0", port=5000)