From b249bbb5b50059120ee43dcd12ce10eb911edd7c Mon Sep 17 00:00:00 2001
From: Debanjum Singh Solanky <debanjum@gmail.com>
Date: Sun, 26 Nov 2023 14:19:46 -0800
Subject: [PATCH] Limit max audio file size allowed for transcription on API
 endpoint

---
 src/khoj/routers/api.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/khoj/routers/api.py b/src/khoj/routers/api.py
index bc55ce48..3fd2285d 100644
--- a/src/khoj/routers/api.py
+++ b/src/khoj/routers/api.py
@@ -595,6 +595,11 @@ async def transcribe(request: Request, common: CommonQueryParams, file: UploadFi
     audio_filename = f"{user.uuid}-{str(uuid.uuid4())}.webm"
     user_message: str = None
 
+    # If the file is too large, return an unprocessable entity error
+    if file.size > 10 * 1024 * 1024:
+        logger.warning(f"Audio file too large to transcribe. Audio file size: {file.size}. Exceeds 10Mb limit.")
+        return Response(content="Audio size larger than 10Mb limit", status_code=422)
+
     # Transcribe the audio from the request
     try:
         # Store the audio from the request in a temporary file
@@ -627,7 +632,7 @@ async def transcribe(request: Request, common: CommonQueryParams, file: UploadFi
     update_telemetry_state(
         request=request,
         telemetry_type="api",
-        api="speech_to_text",
+        api="transcribe",
         **common.__dict__,
     )