Update app.py

app.py

@@ -66,15 +66,26 @@ def add_text(state, imagebox, textbox, image_process_mode):
         state = conv_templates[conv_mode].copy()
 
     if imagebox is not None:
-        image = Image.open(imagebox).convert('RGB')
+        try:
+            image = Image.open(imagebox).convert('RGB')
+        except Exception as e:
+            print(f"Failed to load image: {e}")
+            yield (state, state.to_gradio_chatbot(), "⚠️ Could not load example image.", None) + (enable_btn,) * 5
+            return
+
         textbox = DEFAULT_IMAGE_TOKEN + "\nWhat material is this item and how to dispose of it?"
         textbox = (textbox, image, image_process_mode)
 
+    else:
+        yield (state, state.to_gradio_chatbot(), "⚠️ Please upload or select an image first.", None) + (enable_btn,) * 5
+        return
+
     state.append_message(state.roles[0], textbox)
     state.append_message(state.roles[1], None)
 
     yield (state, state.to_gradio_chatbot(), "", None) + (disable_btn, disable_btn, disable_btn, enable_btn, enable_btn)
 
+
 def delete_text(state, image_process_mode):
     state.messages[-1][-1] = None
     prev_human_msg = state.messages[-2]