Update main.py

main.py

@@ -5,7 +5,6 @@ import time
 import pytz
 import os
 
-
 app = Flask(__name__)
 
 @app.route("/")
@@ -29,7 +28,15 @@ def get_time():
     full_date = f"{curr_day} | {curr_date} | {curr_time}"
     return full_date, curr_time
 
-    
+# Convert file size from bytes to KB or MB
+def convert_size(bytes):
+    if bytes < 1024:
+        return f"{bytes} bytes"
+    elif bytes < 1024**2:
+        return f"{bytes / 1024:.2f} KB"
+    else:
+        return f"{bytes / 1024**2:.2f} MB"
+
 def transcribe(audio_path):
     print(f"Transcribing audio from: {audio_path}\n", flush=True)
 
@@ -67,14 +74,15 @@ def transcribe_audio():
         return jsonify({"error": "No audio file provided"}), 400
     
     audio_file = request.files['audio']
-    audio_file_size = len(audio_file.read())  # Calculate the size of the file in bytes
+    audio_file_size_bytes = len(audio_file.read())  # Calculate the size of the file in bytes
     audio_file.seek(0)  # Reset the file pointer after reading
+    audio_file_size = convert_size(audio_file_size_bytes)  # Convert file size to KB or MB
     
     # Save the uploaded audio file
     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)\n", flush=True)
+    print(f"Audio file saved to: {audio_path} (Size: {audio_file_size})\n", flush=True)
     
     # Record the time before starting transcription
     transcription_start_time = time.time()
@@ -88,7 +96,7 @@ def transcribe_audio():
     
     # Calculate the time taken for transcription
     transcription_end_time = time.time()
-    transcription_duration = transcription_end_time - transcription_start_time
+    transcription_duration = round(transcription_end_time - transcription_start_time, 2)
     
     # Clean up the saved file
     os.remove(audio_path)
@@ -103,9 +111,9 @@ def transcribe_audio():
         "transcription": transcription,
         "language": language,
         "request_received_time": request_received_time,
-        "transcription_duration": transcription_duration,
+        "transcription_duration_seconds": transcription_duration,
         "response_sent_time": response_sent_time,
-        "audio_file_size_bytes": audio_file_size
+        "audio_file_size": audio_file_size
     }), 200
 
 @app.route('/healthcheck', methods=['GET'])