add basic auth

app.py

@@ -1,12 +1,36 @@
-from fastapi import FastAPI
-from fastapi.responses import JSONResponse, RedirectResponse
-import gradio as gr
-import threading
+import os
+from dotenv import load_dotenv
 import time
+import base64
 import requests
+import threading
+import gradio as gr
+from fastapi.responses import JSONResponse, RedirectResponse
+from fastapi import FastAPI, Request, Response, status
+
+# Charger les variables d'environnement
+load_dotenv()
 
 app = FastAPI()
 
+USERNAME = os.getenv("USERNAME")
+PASSWORD = os.getenv("PASSWORD")
+
+@app.middleware("http")
+async def basic_auth(request: Request, call_next):
+    auth = request.headers.get("Authorization")
+    expected = "Basic " + base64.b64encode(f"{USERNAME}:{PASSWORD}".encode()).decode()
+
+    if auth != expected:
+        return Response(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            headers={"WWW-Authenticate": "Basic"},
+            content="Unauthorized",
+        )
+    
+    return await call_next(request)
+
+
 # Liste d'URLs à pinger
 url_list = ["https://example.com"]