Update app.py

app.py

@@ -278,7 +278,8 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
 
         if "file" in message and message["file"]:
             file_contents = message["file"]
-            choice, contents = mode_load(file_contents)
+            # Aquí debes asegurarte de que `mode_load` pueda manejar `file_contents` como bytes
+            choice, contents = mode_load(io.BytesIO(file_contents))
             if choice == "image":
                 conversation.append({"role": "user", "image": contents, "content": message["text"]})
             elif choice == "doc":
@@ -311,21 +312,27 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
 @app.post("/chat/")
 async def test_endpoint(
     text: str = Form(...),
-    file: UploadFile = File(None)  # 'None' makes the file optional
+    file: Optional[UploadFile] = File(None)
 ):
     if not text:
         raise HTTPException(status_code=400, detail="Missing 'text' in request body")
 
+    # Lee el archivo si está presente
+    file_contents = None
     if file:
-        # Process the file if it is provided
         file_contents = await file.read()
-        # Do something with file_contents if needed
-        print("File received")
-    else:
-        print("No file received")
-    
-    response = simple_chat(text)
-    return {"response": response}
+
+    # Construye el diccionario para `simple_chat`
+    message = {
+        "text": text,
+        "file": file_contents
+    }
+
+    print(message)
+
+    # Llama a `simple_chat` con el diccionario
+    response = simple_chat(message)
+    return response
 
 
 with gr.Blocks(css=CSS, theme="soft", fill_height=True) as demo: