Update app.py

app.py

@@ -257,23 +257,27 @@ def bot(history, temperature, top_p, max_output_tokens):
     num_new_images = 0
     for i, message in enumerate(history[:-1]):
         if type(message[0]) is tuple:
-            images_this_term.append(message[0][0])
-            if is_valid_video_filename(message[0][0]):
-                # 不接受视频
-                raise ValueError("Video is not supported")
-                num_new_images += our_chatbot.num_frames
-            elif is_valid_image_filename(message[0][0]):
-                print("#### Load image from local file",message[0][0])
-                num_new_images += 1
+            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.")
+                return history 
             else:
-                raise ValueError("Invalid image file")
+                images_this_term.append(message[0][0])
+                if is_valid_video_filename(message[0][0]):
+                    # 不接受视频
+                    raise ValueError("Video is not supported")
+                    num_new_images += our_chatbot.num_frames
+                elif is_valid_image_filename(message[0][0]):
+                    print("#### Load image from local file",message[0][0])
+                    num_new_images += 1
+                else:
+                    raise ValueError("Invalid image file")
         else:
             num_new_images = 0
 
     # for message in history[-i-1:]:
     #     images_this_term.append(message[0][0])
 
-    assert len(images_this_term) > 0, "must have an image"
+    # assert len(images_this_term) > 0, "must have an image"
     # image_files = (args.image_file).split(',')
     # image = [load_image(f) for f in images_this_term if f]
 
@@ -329,9 +333,6 @@ def bot(history, temperature, top_p, max_output_tokens):
     our_chatbot.conversation.append_message(our_chatbot.conversation.roles[1], None)
     prompt = our_chatbot.conversation.get_prompt()
 
-    if len(images_this_term) == 0:
-        gr.Warning("You should upload an image. Please upload the image and try again.")
-        return history
     
     if len(images_this_term) > 1:
         gr.Warning("Only one image can be uploaded in a conversation. Please reduce the number of images and start a new conversation.")