Update app.py

app.py

@@ -235,25 +235,7 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
     else:
         choice = ""
         conversation.append({"role": "user", "image": "", "content": message['text']})
-    # else:
-    #     for prompt, answer in history:
-    #             if answer is None:
-    #                 prompt_files.append(prompt[0])
-    #                 conversation.extend([{"role": "user", "content": ""}, {"role": "assistant", "content": ""}])
-    #             else:
-    #                 conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
-    #         if len(prompt_files) > 0:
-    #             choice, contents = mode_load(prompt_files[-1])
-    #         else:
-    #             choice = ""
-    #             conversation.append({"role": "user", "image": "", "content": message['text']})
-
-
-    #         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})
+
             
     print(f"Conversation is -\n{conversation}")
 
@@ -362,37 +344,59 @@ 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
-        
-        # 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
-        }
+
+        # 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
     print(message)
     response = simple_chat(message)
     
-    return response
+    return 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
+#         }
+    
+#     # Llamar a la función `simple_chat` con el mensaje
+#     print(message)
+#     response = simple_chat(message)
+    
+#     return response
 
 
 with gr.Blocks(css=CSS, theme="soft", fill_height=True) as demo: