add multithreading

app.py

@@ -14,17 +14,18 @@ from huggingface_hub import login
 from diffusers import FluxPipeline
 from aura_sr import AuraSR
 from deep_translator import GoogleTranslator
-from fastapi import FastAPI, Response, Request
+from fastapi import FastAPI, Response, Request, HTTPException, status
 import uvicorn
 from pydantic import BaseModel
 
 def log(*args, **kwargs):
+    tid = threading.get_ident()
     tz = pytz.timezone(kwargs.pop('timezone', 'Europe/Moscow'))
     time_fmt = kwargs.pop('time_format', '%Y-%m-%d %H:%M:%S')
     separator = kwargs.pop('sep', ' - ')
     moscow_time = datetime.now(tz).strftime(time_fmt)
     message = separator.join(str(arg) for arg in args)
-    print(f"[{moscow_time}]{separator}{message}", **kwargs)
+    print(f"[{moscow_time}]{separator}[tid={tid}]{separator}{message}", **kwargs)
 
 log("Старт сервиса")
 
@@ -190,16 +191,22 @@ class GenerateRequest(BaseModel):
 
 @fastapi_app.post("/api/generate_image")
 async def api_generate_image(rq: GenerateRequest, rest_request: Request):
-    thread_id = threading.get_ident()
     client_ip = rest_request.headers.get('X-Forwarded-For', rest_request.client.host)
-    log(f"tid=[{thread_id}] Получили запрос через рест АПИ.")
-    log(f"tid=[{thread_id}] Client IP: {client_ip}")
+    log("Получили запрос через рест АПИ.")
+    log(f"Client IP: {client_ip}")
+    rq_token = rest_request.headers.get('Authorization')
+    if not rq_token:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Not authorized"
+        )
+    login(token = rq_token)
     image = await generate_image_async(rq.object_name, rq.remove_bg, rq.upscale)
-    log(f"tid=[{thread_id}] Подготовка ответа для рест АПИ.")
+    log(f"Подготовка ответа для рест АПИ.")
     img_byte_arr = io.BytesIO()
     image.save(img_byte_arr, format='WEBP')
     img_byte_arr = img_byte_arr.getvalue()
-    log(f"tid=[{thread_id}] Возвращаем ответ для рест АПИ.")
+    log(f"Возвращаем ответ для рест АПИ.")
     return Response(content=img_byte_arr, media_type="image/webp")