Update app.py

app.py

@@ -8,6 +8,8 @@ import torch.nn.functional as F
 import faiss
 import cv2
 import gradio as gr
+print("Gradio version:", gr.__version__)
+
 from huggingface_hub import snapshot_download
 
 # ----------------------------
@@ -327,15 +329,15 @@ def chat_and_render(user_text, display_history, llm_history, view_key, tau_slide
         return display_history, llm_history, load_gt_np_for_view(view_key), None, view_key, tau_slider, ""
         
 
-    #display_history = display_history + [{"role":"user","content": user_text}]
+    display_history = display_history + [{"role":"user","content": user_text}]
     llm_history     = llm_history     + [{"role":"user","content": user_text}]
 
     try:
         obj = route_with_groq(llm_history[-6:])
     except Exception as e:
         msg = f"[Router error] {e}"
-        #display_history += [{"role":"assistant","content": msg}]
-        display_history = display_history + [(user_text, assistant_msg)]
+        display_history += [{"role":"assistant","content": msg}]
+        #display_history = display_history + [(user_text, assistant_msg)]
         llm_history += [{"role":"assistant","content": msg}]
         return display_history, llm_history, load_gt_np_for_view(view_key), None, view_key, tau_slider, ""
 
@@ -355,8 +357,8 @@ def chat_and_render(user_text, display_history, llm_history, view_key, tau_slide
 
     if label is None:
         assistant_msg = reply or "Which object should I look for?"
-        #display_history += [{"role":"assistant","content": assistant_msg}]
-        display_history = display_history + [(user_text, assistant_msg)]
+        display_history += [{"role":"assistant","content": assistant_msg}]
+        #display_history = display_history + [(user_text, assistant_msg)]
         llm_history += [{"role":"assistant","content": assistant_msg}]
         return display_history, llm_history, load_gt_np_for_view(view_key), None, view_key, tau_used, ""
 
@@ -364,14 +366,14 @@ def chat_and_render(user_text, display_history, llm_history, view_key, tau_slide
         gt  = load_gt_np_for_view(view_key)
         sel = sess.render_selected_rgb_np(label=label, tau=tau_used, frame=view_key, split="train")
         assistant_msg = reply or f"Showing {label} (tau={tau_used:.2f})."
-        #display_history += [{"role":"assistant","content": assistant_msg}]
-        display_history = display_history + [(user_text, assistant_msg)]
+        display_history += [{"role":"assistant","content": assistant_msg}]
+        #display_history = display_history + [(user_text, assistant_msg)]
         llm_history += [{"role":"assistant","content": assistant_msg}]
         return display_history, llm_history, gt, sel, view_key, tau_used, ""
     except Exception as e:
         assistant_msg = f"[Render error] {e}"
-        #display_history += [{"role":"assistant","content": assistant_msg}]
-        display_history = display_history + [(user_text, assistant_msg)]
+        display_history += [{"role":"assistant","content": assistant_msg}]
+        #display_history = display_history + [(user_text, assistant_msg)]
         llm_history += [{"role":"assistant","content": assistant_msg}]
         return display_history, llm_history, load_gt_np_for_view(view_key), None, view_key, tau_used, ""
 
@@ -388,8 +390,8 @@ with gr.Blocks() as demo:
         img_gt  = gr.Image(type="numpy", label="Original")
         img_sel = gr.Image(type="numpy", label="Selected")
 
-    #chatbot = gr.Chatbot(type="messages", label="Chat")
-    chatbot = gr.Chatbot(label="Chat")
+    chatbot = gr.Chatbot(type="messages", label="Chat")
+    #chatbot = gr.Chatbot(label="Chat")
     msg = gr.Textbox(placeholder="Ask: Where is the apple?", label="Message")
     send = gr.Button("Send")