Update app.py

app.py

@@ -89,9 +89,10 @@ def upload_imgorvideo(gr_video, gr_img, text_input,chatbot,audio_flag):
         chat_state = conv_llava_llama_2.copy()
     if gr_img is None and gr_video is None:
         return None, None, None, gr.update(interactive=True), chat_state, None
-    elif gr_img is not None and gr_video is None:
-        print(gr_img)
-        chatbot = [((gr_img,), None)]
+    elif gr_video is not None:
+        print(gr_video)
+        chatbot = [((gr_video,), None)]
+        chat_state = default_conversation.copy()
         chat_state = Conversation(
             system= "You are able to understand the visual content that the user provides."
            "Follow the instructions carefully and explain your answers in detail.",
@@ -102,12 +103,11 @@ def upload_imgorvideo(gr_video, gr_img, text_input,chatbot,audio_flag):
             sep="###",
         )
         img_list = []
-        llm_message = chat.upload_img(gr_img, chat_state, img_list)
+        llm_message = chat.upload_video(gr_video, chat_state, img_list)
         return gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=True, placeholder='Type and press Enter'), gr.update(value="Start Chatting", interactive=False), chat_state, img_list,chatbot
-    elif gr_video is not None and gr_img is None:
-        print(gr_video)
-        chatbot = [((gr_video,), None)]
-        chat_state = default_conversation.copy()
+    elif gr_img is not None:
+        print(gr_img)
+        chatbot = [((gr_img,), None)]
         chat_state = Conversation(
             system= "You are able to understand the visual content that the user provides."
            "Follow the instructions carefully and explain your answers in detail.",
@@ -118,7 +118,7 @@ def upload_imgorvideo(gr_video, gr_img, text_input,chatbot,audio_flag):
             sep="###",
         )
         img_list = []
-        llm_message = chat.upload_video(gr_video, chat_state, img_list)
+        llm_message = chat.upload_img(gr_img, chat_state, img_list)
         return gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=True, placeholder='Type and press Enter'), gr.update(value="Start Chatting", interactive=False), chat_state, img_list,chatbot
     else:
         # img_list = []