Update app.py

app.py

@@ -203,6 +203,23 @@ def transcription(file):
         print(f"Exception: {e}")
         return None
 
+def transcription_content(file):
+    try:
+        deepgram = DeepgramClient(DEEPGRAM_API_KEY)
+        if isinstance(file, str):  # File from folder
+            with open(file, "rb") as f:
+                buffer_data = f.read()
+        else:  # Uploaded file
+            buffer_data = file.file.read()
+
+        payload = {"buffer": buffer_data}
+        options = PrerecordedOptions(model="nova-2", smart_format=True, diarize=True)
+        response = deepgram.listen.prerecorded.v("1").transcribe_file(payload, options)
+        return response..results.channels[0].alternatives[0].transcript
+    except Exception as e:
+        print(f"Exception: {e}")
+        return None
+
 
 
 ALLOWED_EXTENSIONS = {".mp3", ".wav", ".wma"}
@@ -236,7 +253,7 @@ def inference(input_prompt: Prompt, question: Prompt):
     return custom_qa_automation(text=input_prompt.text, custom_question=question.text)
 
 
-@app.post("/api/transcribe-audio", summary="Transcribe audio file", tags=["Transcription"])
+@app.post("/api/transcribe-display", summary="Transcribe audio file", tags=["Transcription"])
 async def transcribe_audio_route(file: UploadFile = File(...)):
     # Check if the file extension is allowed
     file_extension = os.path.splitext(file.filename)[1].lower()
@@ -248,3 +265,16 @@ async def transcribe_audio_route(file: UploadFile = File(...)):
         return {"transcription": transcription_result}
     else:
         return {"error": "Transcription failed"}
+
+@app.post("/api/transcribe-content", summary="Transcribe audio file", tags=["Transcription"])
+async def transcribe_audio_route(file: UploadFile = File(...)):
+    # Check if the file extension is allowed
+    file_extension = os.path.splitext(file.filename)[1].lower()
+    if file_extension not in ALLOWED_EXTENSIONS:
+        raise HTTPException(status_code=400, detail=f"File format not supported. Allowed formats: {', '.join(ALLOWED_EXTENSIONS)}")
+
+    transcription_result = transcription_content(file)
+    if transcription_result:
+        return {"transcription": transcription_result}
+    else:
+        return {"error": "Transcription failed"}