Update app.py

app.py

@@ -342,19 +342,23 @@ def simple_chat(message: dict, file_obj: io.BytesIO = None, temperature: float =
 
 
 @app.post("/chat/")
-async def test_endpoint(message: dict):
-    if "text" not in message:
+async def test_endpoint(text: str, file_url: str = None):
+    if not text:
         raise HTTPException(status_code=400, detail="Missing 'text' in request body")
 
-    if "file" in message and message["file"]:
-        file_url = message["file"]
+    if file_url:
         file_obj = download_file(file_url)
         if file_obj is None:
             raise HTTPException(status_code=400, detail="Failed to download file from URL")
     else:
         file_obj = None
 
-    response = simple_chat(message, file_obj)
+    print("Mensaje: ")
+    print(text)
+    print("File: ")
+    print(file_obj)
+    
+    response = simple_chat(text, file_obj)
     return response
 
 def download_file(url: str) -> io.BytesIO: