Update app.py

app.py

@@ -134,32 +134,36 @@ def predict(_chatbot, task_history) -> tuple:
         query = [{'image': chat_query}]
     else:
         query = [{'text': _parse_text(chat_query)}]
-    
     inputs = tokenizer.from_list_format(query)
     tokenized_inputs = tokenizer(inputs, return_tensors='pt')
     tokenized_inputs = tokenized_inputs.to(model.device)
     pred = model.generate(**tokenized_inputs)
     response = tokenizer.decode(pred.cpu()[0], skip_special_tokens=False)
-
     if 'image' in query[0]:
         print("Model response:", response)
-
     if 'image' in query[0]:
         image = tokenizer.draw_bbox_on_latest_picture(response)
         if image is not None:
             image_path = save_image(image, uploaded_file_dir)
             _chatbot[-1] = (chat_query, (image_path,))
-            task_history[-1] = (chat_query, image_path)
+            if task_history:
+                task_history[-1] = (chat_query, image_path)
+            else:
+                task_history.append((chat_query, image_path))
         else:
-            _chatbot[-1] = (chat_query, response)  # Make sure this is a tuple of two elements
-            task_history[-1] = (chat_query, response)
+            _chatbot[-1] = (chat_query, response)
+            if task_history:
+                task_history[-1] = (chat_query, response)
+            else:
+                task_history.append((chat_query, response))
     else:
         _chatbot[-1] = (chat_query, response)
-        task_history[-1] = (chat_query, response)
-
+        if task_history:
+            task_history[-1] = (chat_query, response)
+        else:
+            task_history.append((chat_query, response))
     return _chatbot, task_history
 
-
 def save_uploaded_image(image_file, upload_dir):
     if image is None:
         return None