Update app.py

app.py

@@ -8,10 +8,10 @@ import urllib3
 import pickle
 from functools import lru_cache
 from dotenv import load_dotenv
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import JSONResponse
 import time
-from tqdm import tqdm # Importando tqdm
+from tqdm import tqdm
 
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
@@ -137,9 +137,13 @@ async def process_message(message):
 async def api_generate_multimodel(request: Request):
     try:
         data = await request.json()
-        message = data["message"]
+        message = data.get("message")
+        if not message:
+            raise HTTPException(status_code=400, detail="Mensaje faltante")
         response = await process_message(message)
         return JSONResponse({"response": response})
+    except HTTPException as e:
+        raise e
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)