Update main.py

main.py

@@ -1,8 +1,9 @@
 from flask import Flask, request, jsonify
 from datetime import datetime
 import whisper
-import os
 import time
+import pytz
+import os
 
 
 app = Flask(__name__)
@@ -16,6 +17,19 @@ print("Loading Whisper model...", flush=True)
 model = whisper.load_model("tiny")
 print("Whisper model loaded.", flush=True)
 
+# Get time of request
+def get_time():
+    nairobi_timezone = pytz.timezone('Africa/Nairobi')
+    current_time_nairobi = datetime.now(nairobi_timezone)
+    
+    curr_day = current_time_nairobi.strftime('%A')
+    curr_date = current_time_nairobi.strftime('%Y-%m-%d')
+    curr_time = current_time_nairobi.strftime('%H:%M:%S')
+    
+    full_date = f"{curr_day} | {curr_date} | {curr_time}"
+    return full_date, curr_time
+
+    
 def transcribe(audio_path):
     print(f"Transcribing audio from: {audio_path}", flush=True)
 
@@ -45,7 +59,7 @@ def transcribe(audio_path):
 @app.route('/transcribe', methods=['POST'])
 def transcribe_audio():
     # Record the time when the request was received
-    request_received_time = datetime.now()
+    request_received_time = get_time()[0]
     print(f"Received request at /transcribe at {request_received_time}", flush=True)
 
     if 'audio' not in request.files:
@@ -81,8 +95,8 @@ def transcribe_audio():
     print(f"Audio file removed from: {audio_path}", flush=True)
     
     # Record the time when the response is being sent
-    response_sent_time = datetime.now()
-    
+    response_sent_time = get_time()[0]
+
     # Return the transcription, detected language, and timing information
     print(f"Transcription: {transcription}, Language: {language}", flush=True)
     return jsonify({