Update app.py

app.py

@@ -219,21 +219,31 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
         
     conversation = []
 
-    # Procesar el mensaje
-    if message.file:
-        file_contents = message.file.file.read()
-        # Aquí deberías procesar el archivo como corresponda, por ejemplo:
-        # choice, contents = mode_load(file_contents)
-        # Por ahora solo agregaremos un marcador de posición
-        choice = "doc"
-        contents = "Contenido del archivo"
+    # if message.file:
+    #     file_contents = message.file.file.read()
+    #     # Aquí deberías procesar el archivo como corresponda, por ejemplo:
+    #     # choice, contents = mode_load(file_contents)
+    #     # Por ahora solo agregaremos un marcador de posición
+    #     choice = "doc"
+    #     contents = "Contenido del archivo"
+    #     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:
+    #     conversation.append({"role": "user", "content": message.text})
+
+    if "file" in message and message["file"]:
+        file_path = message["file"]
+        choice, contents = mode_load(file_path)
         if choice == "image":
-            conversation.append({"role": "user", "image": contents, "content": message.text})
+            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" + message["text"]
             conversation.append({"role": "user", "content": format_msg})
     else:
-        conversation.append({"role": "user", "content": message.text})
+        conversation.append({"role": "user", "content": message["text"]})
 
     # Preparar entrada para el modelo
     input_ids = tokenizer.apply_chat_template(conversation, tokenize=True, add_generation_prompt=True,