Update app.py

app.py

@@ -262,14 +262,14 @@ def add_message(history, message):
         if message["text"] is not None:
             history.append((message["text"], None))
         
-        return [], gr.MultimodalTextbox(value=None, interactive=False)
+        return history, gr.MultimodalTextbox(value=None, interactive=False)
     
 
 @spaces.GPU
 def bot(history, temperature, top_p, max_output_tokens):
-    if not history:
-        gr.Warning("Only one image can be uploaded in a conversation. Please reduce the number of images and start a new conversation.")
-        return history 
+    # if not history:
+    #     gr.Warning("Only one image can be uploaded in a conversation. Please reduce the number of images and start a new conversation.")
+    #     return history 
     print("### turn start history",history)
     print("### turn start conv",our_chatbot.conversation)
     text = history[-1][0]
@@ -277,9 +277,12 @@ def bot(history, temperature, top_p, max_output_tokens):
     text_this_term = ""
     # import pdb;pdb.set_trace()
     num_new_images = 0
-    
+    previous_image = False
     for i, message in enumerate(history[:-1]):
         if type(message[0]) is tuple:
+            if previous_image:
+                gr.Warning("Only one image can be uploaded in a conversation. Please reduce the number of images and start a new conversation.")
+                return history 
             # print("### message[0]",message[0])
             # if len(message[0])>1:
             #     gr.Warning("Only one image can be uploaded in a conversation. Please reduce the number of images and start a new conversation.")
@@ -294,8 +297,10 @@ def bot(history, temperature, top_p, max_output_tokens):
                 num_new_images += 1
             else:
                 raise ValueError("Invalid image file")
+            previous_image = True
         else:
             num_new_images = 0
+            previous_image = False
 
     # for message in history[-i-1:]:
     #     images_this_term.append(message[0][0])