update chat

app.py

@@ -207,12 +207,24 @@ def gradio_reset(chat_state, img_list):
         value="Upload & Start Chat", interactive=True), chat_state, img_list
 
 
+def build_image(image):
+    if image is None:
+        return  None
+    # res = draw_bounding_boxes(image=image, boxes=boxes_to_draw, colors=color_to_draw, width=8)
+    from torchvision.transforms import ToPILImage
+    # res = ToPILImage()(res)
+    _, path = tempfile.mkstemp(suffix='.jpg', dir=TEMP_FILE_DIR)
+    image.save(path)
+
+    return  path
+
 def upload_img(gr_img, text_input, chat_state,chatbot):
     if gr_img is None:
         return None, None, gr.update(interactive=True), chat_state, None
     chat_state = []
     img_list = []
-    chatbot = chatbot + [[(gr_img,), None]]
+    path = build_image(gr_img)
+    chatbot = chatbot + [[(path,), None]]
     llm_message = chat.upload_img(gr_img, chat_state, img_list)
     return gr.update(interactive=False), gr.update(interactive=True, placeholder='Type and press Enter'), gr.update(
         value="Start Chatting", interactive=False), chat_state, img_list