Update app.py

app.py

@@ -308,27 +308,39 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
             low_cpu_mem_usage=True,
             trust_remote_code=True
         )
+        
+        tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 
         conversation = []
 
-        if "files" in message and message["files"]:
-            file_content = message["files"][-1]  # Assuming the last file is the one to be processed
-            choice, contents = mode_load(file_content, message["file_name"])
+        # Acceder al contenido del archivo y al nombre del archivo
+        if "file_content" in message and message["file_content"]:
+            file_content = message["file_content"]
+            file_name = message["file_name"]
+            
+            # Guardar el archivo en un archivo temporal
+            with open(file_name, "wb") as f:
+                f.write(file_content.read())
+            
+            # Llamar a `mode_load` con el nombre del archivo
+            choice, contents = mode_load(file_name)
+            
             if choice == "image":
                 conversation.append({"role": "user", "image": contents, "content": message['text']})
             elif choice == "doc":
-                format_msg = contents + "\n\n\n" + "{} files uploaded.\n".format(len(message["files"])) + message['text']
+                format_msg = contents + "\n\n\n" + "{} files uploaded.\n".format(1) + message['text']
                 conversation.append({"role": "user", "content": format_msg})
         else:
-                contents = None
-                conversation.append({"role": "user", "content": message['text']})
+            # Manejar caso donde no se sube archivo
+            conversation.append({"role": "user", "content": message['text']})
 
-        input_ids = tokenizer.apply_chat_template(conversation, tokenize=True, add_generation_prompt=True, return_tensors="pt", return_dict=True).to(model.device)
-        
-        print("   ")
-        print("Conv: ")
+        print("--------------")
+        print("  ")
         print(conversation)
-        print("   ")
+        print("  ")
+        print("--------------")
+
+        input_ids = tokenizer.apply_chat_template(conversation, tokenize=True, add_generation_prompt=True, return_tensors="pt", return_dict=True).to(model.device)
 
         generate_kwargs = dict(
             max_length=max_length,