Update app.py

app.py

@@ -14,18 +14,22 @@ class LogRequestMiddleware(BaseHTTPMiddleware):
         # Captura o início da requisição
         start_time = time.time()
 
+        # Verificar o IP do cliente no cabeçalho X-Forwarded-For
+        client_ip = request.headers.get("X-Forwarded-For")
+        if not client_ip:
+            client_ip = request.client.host  # Se não houver X-Forwarded-For, usa o IP direto
+
         # Realiza a requisição
         response = await call_next(request)
 
         # Dados da requisição
-        ip = request.client.host  # IP de origem
         method = request.method  # Método HTTP
         path = request.url.path  # Caminho da URL
         status_code = response.status_code  # Código de status da resposta
         process_time = time.time() - start_time  # Tempo de processamento
 
         # Log detalhado
-        logger.info(f"Request: {method} {path} from {ip} - Status: {status_code} - Process Time: {process_time:.4f} seconds")
+        logger.info(f"Request: {method} {path} from {client_ip} - Status: {status_code} - Process Time: {process_time:.4f} seconds")
 
         return response