Update app.py

app.py

@@ -225,16 +225,28 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
 
     conversation = []
     prompt_files = []
-    if message["files"]:
-        choice, contents = mode_load(message["files"][-1])
+    # if message["files"]:
+    #     choice, contents = mode_load(message["files"][-1])
+    #     if choice == "image":
+    #         conversation.append({"role": "user", "image": contents, "content": message['text']})
+    #     elif choice == "doc":
+    #         format_msg = contents + "\n\n\n" + "{} files uploaded.\n" + message['text']
+    #         conversation.append({"role": "user", "content": format_msg})
+    # else:
+    #     choice = ""
+    #     conversation.append({"role": "user", "image": "", "content": message['text']})
+
+    if message["file_content"]:
+        choice, contents = mode_load(message["file_content"])
         if choice == "image":
             conversation.append({"role": "user", "image": contents, "content": message['text']})
         elif choice == "doc":
-            format_msg = contents + "\n\n\n" + "{} files uploaded.\n" + message['text']
+            format_msg = contents + "\n\n\n" + "{} files uploaded.\n".format(message['file_name']) + message['text']
             conversation.append({"role": "user", "content": format_msg})
     else:
-        choice = ""
-        conversation.append({"role": "user", "image": "", "content": message['text']})
+        conversation.append({"role": "user", "content": message['text']})
+
+    print(f"Conversation is -\n{conversation}")
 
             
     print(f"Conversation is -\n{conversation}")
@@ -344,59 +356,38 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
 #         return PlainTextResponse(f"Error: {str(e)}")
 
 
+
+
 @app.post("/chat/")
 async def test_endpoint(
     text: str = Form(...), 
     file: UploadFile = File(None)
 ):
-    # Construir el mensaje base solo con el texto
-    message = {"text": text, "files": []}
-
     # Verificar si se ha subido un archivo
     if file:
         # Leer el archivo en memoria
         file_content = BytesIO(await file.read())
         file_name = file.filename
-
-        # Agregar el archivo al mensaje
-        message["files"].append({"file_content": file_content, "file_name": file_name})
-    
-    # Llamar a la función `simple_chat` con el mensaje y recolectar la salida
-    response = "".join([chunk for chunk in simple_chat(message)])
-    
-    return {"response": response}
-
-
-# @app.post("/chat/")
-# async def test_endpoint(
-#     text: str = Form(...), 
-#     file: UploadFile = File(None)
-# ):
-#     # Verificar si se ha subido un archivo
-#     if file:
-#         # Leer el archivo en memoria
-#         file_content = BytesIO(await file.read())
-#         file_name = file.filename
         
-#         # Construir el mensaje con el archivo y el texto
-#         message = {
-#             "text": text,
-#             "file_content": file_content,
-#             "file_name": file_name
-#         }
-#     else:
-#         # Si no se sube archivo, solo se incluye el texto
-#         message = {
-#             "text": text,
-#             "file_content": None,
-#             "file_name": None
-#         }
+        # Construir el mensaje con el archivo y el texto
+        message = {
+            "text": text,
+            "file_content": file_content,
+            "file_name": file_name
+        }
+    else:
+        # Si no se sube archivo, solo se incluye el texto
+        message = {
+            "text": text,
+            "file_content": None,
+            "file_name": None
+        }
     
-#     # Llamar a la función `simple_chat` con el mensaje
-#     print(message)
-#     response = simple_chat(message)
+    # Llamar a la función `simple_chat` con el mensaje
+    print(message)
+    response = simple_chat(message)
     
-#     return response
+    return response
 
 
 with gr.Blocks(css=CSS, theme="soft", fill_height=True) as demo: