step 6

app.py

@@ -38,24 +38,24 @@ def encode_image(image):
     inputs = processor(images=[image], return_tensors="pt", padding=True, truncation=True)
     return inputs
 
-def bot(history, message, image=None):
-    if image is not None:
-        image = preprocess_image(image)
-    top_3_products = initial_query(image, message)
+def send_message(chatbot, txt, btn):
+    if btn is not None:
+        image = preprocess_image(btn)
+    else:
+        image = None
+    top_3_products = initial_query(image, txt)
     output_html = generate_output_html(top_3_products)
-    history = history + [(message, output_html)]
-    return history
-
-def send_message(chatbot, txt):
-    bot(chatbot, txt)
+    chatbot.append_message("You", txt)
+    chatbot.append_message("AI", output_html)
 
 chatbot = gr.Chatbot([]).style(height=750)
 txt = gr.Textbox(placeholder="Enter text and press enter, or upload an image", show_label=False)
-btn = gr.UploadButton("📁", file_types=["image", "video", "audio"])
-send_btn = gr.Button("Send", on_click=send_message, args=[chatbot, txt])
+btn = gr.UploadButton("📸", file_types=["image", "video", "audio"])
+send_btn = gr.Button("Send", action=send_message)
 
 with gr.Blocks() as demo:
     chatbot
     gr.Row(txt, btn, send_btn)
 
 demo.launch()
+