Update app.py

app.py

@@ -149,15 +149,17 @@ def predict(_chatbot, task_history) -> tuple:
         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)
         else:
             _chatbot[-1] = (chat_query, response)  # Make sure this is a tuple of two elements
+            task_history[-1] = (chat_query, response)
     else:
         _chatbot[-1] = (chat_query, response)
-
-    task_history[-1] = _chatbot[-1]
+        task_history[-1] = (chat_query, response)
 
     return _chatbot, task_history
 
+
 def save_uploaded_image(image_file, upload_dir):
     if image is None:
         return None
@@ -185,14 +187,20 @@ def regenerate(_chatbot, task_history) -> list:
 
 def add_text(history, task_history, text) -> tuple:
     if not text.strip():
-        return history, task_history, ""
+        return history, task_history, chatbot
+    if not any(isinstance(item[0], tuple) for item in history):
+        prompt = "Please upload and submit an image to get started."
+        history.append((prompt, None))
+        task_history.append((prompt, None))
+        chatbot.append(prompt)
+        return history, task_history, chatbot
     task_text = text
     if len(text) >= 2 and text[-1] in PUNCTUATION and text[-2] not in PUNCTUATION:
         task_text = text[:-1]
     history_item = (_parse_text(task_text), None)
     history.append(history_item)
     task_history.append(history_item)
-    return history, task_history, ""
+    return history, task_history, chatbot
 
 def add_file(history, task_history, file):
     if file is None: