Merge branch 'main' of github.com:Monadical-SAS/reflector into feat-sharing

server/reflector/views/transcripts.py

@@ -1,12 +1,14 @@
 from datetime import datetime, timedelta
 from typing import Annotated, Literal, Optional
 
+import httpx
 import reflector.auth as auth
 from fastapi import (
     APIRouter,
     Depends,
     HTTPException,
     Request,
+    Response,
     WebSocket,
     WebSocketDisconnect,
     status,
@@ -245,6 +247,42 @@ async def transcript_get_audio_mp3(
         transcript_id, user_id=user_id
     )
 
+    if transcript.audio_location == "storage":
+        # proxy S3 file, to prevent issue with CORS
+        url = await transcript.get_audio_url()
+        headers = {}
+
+        copy_headers = ["range", "accept-encoding"]
+        for header in copy_headers:
+            if header in request.headers:
+                headers[header] = request.headers[header]
+
+        async with httpx.AsyncClient() as client:
+            resp = await client.request(request.method, url, headers=headers)
+            return Response(
+                content=resp.content,
+                status_code=resp.status_code,
+                headers=resp.headers,
+            )
+
+    if transcript.audio_location == "storage":
+        # proxy S3 file, to prevent issue with CORS
+        url = await transcript.get_audio_url()
+        headers = {}
+
+        copy_headers = ["range", "accept-encoding"]
+        for header in copy_headers:
+            if header in request.headers:
+                headers[header] = request.headers[header]
+
+        async with httpx.AsyncClient() as client:
+            resp = await client.request(request.method, url, headers=headers)
+            return Response(
+                content=resp.content,
+                status_code=resp.status_code,
+                headers=resp.headers,
+            )
+
     if not transcript.audio_mp3_filename.exists():
         raise HTTPException(status_code=500, detail="Audio not found")
 
@@ -269,8 +307,8 @@ async def transcript_get_audio_waveform(
         transcript_id, user_id=user_id
     )
 
-    if not transcript.audio_mp3_filename.exists():
-        raise HTTPException(status_code=500, detail="Audio not found")
+    if not transcript.audio_waveform_filename.exists():
+        raise HTTPException(status_code=404, detail="Audio not found")
 
     return transcript.audio_waveform