Update app.py

app.py

@@ -282,32 +282,32 @@ def add_message(history, message):
         history = []
         our_chatbot = chat_manager.get_chatbot(args, model_path, tokenizer, model, image_processor, context_len)
         chat_image_num = 0
-        
-    if len(message["files"]) <= 1:
-        for x in message["files"]:
-            history.append(((x,), None))
-            chat_image_num += 1
-        if chat_image_num > 1:
-            history = []
-            chat_manager.reset_chatbot()
-            our_chatbot = chat_manager.get_chatbot(args, model_path, tokenizer, model, image_processor, context_len)
-            chat_image_num = 0
-            for x in message["files"]:
-                history.append(((x,), None))
-                chat_image_num += 1
+    
+    # if len(message["files"]) <= 1:
+    #     for x in message["files"]:
+    #         history.append(((x,), None))
+    #         chat_image_num += 1
+    #     if chat_image_num > 1:
+    #         history = []
+    #         chat_manager.reset_chatbot()
+    #         our_chatbot = chat_manager.get_chatbot(args, model_path, tokenizer, model, image_processor, context_len)
+    #         chat_image_num = 0
+    #         for x in message["files"]:
+    #             history.append(((x,), None))
+    #             chat_image_num += 1
                 
-        if message["text"] is not None:
-            history.append((message["text"], None))
+    #     if message["text"] is not None:
+    #         history.append((message["text"], None))
         
-        print(f"### Chatbot instance ID: {id(our_chatbot)}")
-        return history, gr.MultimodalTextbox(value=None, interactive=False)
-    else:
-        for x in message["files"]:
-            history.append(((x,), None))
-        if message["text"] is not None:
-            history.append((message["text"], None))
-
-        return history, gr.MultimodalTextbox(value=None, interactive=False)
+    #     print(f"### Chatbot instance ID: {id(our_chatbot)}")
+    #     return history, gr.MultimodalTextbox(value=None, interactive=False)
+    # else:
+    for x in message["files"]:
+        history.append(((x,), None))
+    if message["text"] is not None:
+        history.append((message["text"], None))
+    print(f"### Chatbot instance ID: {id(our_chatbot)}")
+    return history, gr.MultimodalTextbox(value=None, interactive=False)
 
 
 @spaces.GPU