Update app.py

app.py

@@ -26,8 +26,10 @@ cur_dir = os.path.dirname(os.path.abspath(__file__))
 def ovis_chat(message, history):
     try :
         image_input = Image.open(message["files"][0]).convert("RGB")
+        new_image = True
     except : 
         image_input = None
+        new_image = False
     # preprocess inputs
     conversations = []
     response = ""
@@ -47,10 +49,9 @@ def ovis_chat(message, history):
         elif isinstance(msg[0],tuple): 
             # case history pair is an image and user did not pass a new image
             # we override the none with the history image
-            if image_input is None : 
-                # if someone uploads a file and not an image this will break
+            if new_image is False : 
+                # always aim for the latest image in the history
                 image_input = Image.open(msg[0][0]).convert("RGB")
-    print("conversations : ",conversations)
 
     text_input = text_input.replace(image_placeholder, '')
     conversations.append({
@@ -138,9 +139,6 @@ latex_delimiters_set = [{
         "display": True
     }]
 
-#     send_click_event = send_btn.click(submit_chat, [chatbot, text_input], [chatbot, text_input]).then(ovis_chat,[chatbot, image_input],chatbot)
-#     submit_event = text_input.submit(submit_chat, [chatbot, text_input], [chatbot, text_input]).then(ovis_chat,[chatbot, image_input],chatbot)
-#     clear_btn.click(clear_chat, outputs=[chatbot, image_input, text_input])
 
 demo = gr.ChatInterface(fn=ovis_chat, textbox=gr.MultimodalTextbox(),multimodal=True)
 demo.launch(debug=True)