Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import urllib.parse
 from fastapi import FastAPI, HTTPException, Request, status
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import FileResponse, StreamingResponse
+from fastapi.responses import FileResponse
 from pydantic import BaseModel
 
 app = FastAPI()
@@ -102,79 +102,37 @@ async def receive_upload(filename: str, request: Request):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-# 🎬 3. DOWNLOAD E STREAMING OTIMIZADO DE VÍDEO (Com suporte a Range Requests)
+# 🎬 3. DOWNLOAD E STREAMING NATIVO DE ALTA PERFORMANCE (Usa FileResponse nativo do FastAPI)
+# O FileResponse lida automaticamente com Range Requests de forma assíncrona extremamente veloz.
 @app.get("/download")
-def download_file(name: str, request: Request):
+def download_file(name: str):
     try:
         filepath = os.path.join(STORAGE_DIR, name)
         if not os.path.exists(filepath):
             raise HTTPException(status_code=404, detail="Arquivo não encontrado")
         
-        file_size = os.path.getsize(filepath)
-        range_header = request.headers.get("range")
-        
-        # Se o navegador pedir apenas partes do arquivo (comum em players de vídeo/áudio)
-        if range_header:
-            try:
-                # Exemplo de range: "bytes=0-1024"
-                byte_ranges = range_header.replace("bytes=", "").split("-")
-                start = int(byte_ranges[0])
-                end = int(byte_ranges[1]) if byte_ranges[1] else file_size - 1
-            except ValueError:
-                raise HTTPException(status_code=status.HTTP_416_REQUESTED_RANGE_NOT_SATISFIABLE, detail="Intervalo de bytes inválido")
-            
-            if start >= file_size or end >= file_size:
-                raise HTTPException(status_code=status.HTTP_416_REQUESTED_RANGE_NOT_SATISFIABLE, detail="Intervalo fora do limite do arquivo")
-            
-            chunk_size = end - start + 1
-            
-            def file_generator(path, offset, size):
-                with open(path, "rb") as f:
-                    f.seek(offset)
-                    remaining = size
-                    while remaining > 0:
-                        chunk_to_read = min(1024 * 1024, remaining)  # Lendo em partes de 1MB
-                        data = f.read(chunk_to_read)
-                        if not data:
-                            break
-                        remaining -= len(data)
-                        yield data
-
-            headers = {
-                "Content-Range": f"bytes {start}-{end}/{file_size}",
-                "Accept-Ranges": "bytes",
-                "Content-Length": str(chunk_size),
-                "Access-Control-Expose-Headers": "Content-Range, Content-Length, Accept-Ranges",
-                "Cache-Control": "public, max-age=31536000, immutable"
-            }
-            
-            # Descobre o MIME Type do arquivo
-            lower_name = name.lower()
-            media_type = "application/octet-stream"
-            if lower_name.endswith(".mp4"):
-                media_type = "video/mp4"
-            elif lower_name.endswith(".webm"):
-                media_type = "video/webm"
-            elif lower_name.endswith(".mkv"):
-                media_type = "video/x-matroska"
-            elif lower_name.endswith(".mp3"):
-                media_type = "audio/mpeg"
-
-            return StreamingResponse(
-                file_generator(filepath, start, chunk_size),
-                status_code=206,
-                headers=headers,
-                media_type=media_type
-            )
-            
-        # Se for um download comum (sem range request)
-        return FileResponse(
+        # Descobre o MIME Type do arquivo para que o navegador saiba como reproduzir
+        lower_name = name.lower()
+        media_type = "application/octet-stream"
+        if lower_name.endswith(".mp4"):
+            media_type = "video/mp4"
+        elif lower_name.endswith(".webm"):
+            media_type = "video/webm"
+        elif lower_name.endswith(".mkv"):
+            media_type = "video/x-matroska"
+        elif lower_name.endswith(".mp3"):
+            media_type = "audio/mpeg"
+
+        # FileResponse do FastAPI cuida de Range Requests automaticamente e de forma performática
+        response = FileResponse(
             filepath, 
-            media_type="application/octet-stream", 
+            media_type=media_type, 
             filename=name
         )
-    except HTTPException as h_err:
-        raise h_err
+        response.headers["Accept-Ranges"] = "bytes"
+        response.headers["Cache-Control"] = "public, max-age=31536000, immutable"
+        return response
+        
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))