Improved UI. Attempt to fix issue where one heatmap pane freezes when the other one has completed

app.py

@@ -200,6 +200,10 @@ def process_search_tta(
     sent_no:  set[str] = set()
     last_tta = None
     last_no  = None
+    # Track previous status strings so we can emit updates when only the
+    # status (Running…/Done.) changes even if no new frame was produced.
+    prev_status_tta = "Running…"
+    prev_status_no  = "Running…"
 
     try:
         while thread_tta.is_alive() or thread_no.is_alive():
@@ -221,10 +225,14 @@ def process_search_tta(
                     last_no = fp
                     updated = True
 
-            if updated:
-                status_tta = "Running…" if thread_tta.is_alive() else "Done."
-                status_no  = "Running…" if thread_no.is_alive()  else "Done."
+            status_tta = "Running…" if thread_tta.is_alive() else "Done."
+            status_no  = "Running…" if thread_no.is_alive()  else "Done."
+
+            # Emit update if we have a new frame OR if the status label has changed
+            if updated or status_tta != prev_status_tta or status_no != prev_status_no:
                 yield gr.update(interactive=False), last_tta, last_no, gr.update(value=status_tta), gr.update(value=status_no)
+                prev_status_tta = status_tta
+                prev_status_no  = status_no
 
             time.sleep(POLL_INTERVAL)
     finally:
@@ -244,7 +252,9 @@ with gr.Blocks(title="Search-TTA (Simplified)", theme=gr.themes.Base()) as demo:
     gr.Markdown(
         """
         # Search-TTA: A Multimodal Test-Time Adaptation Framework for Visual Search in the Wild Demo
-        <a href="https://search-tta.github.io">Project Website</a> <br>
+        Click on any of the examples below and run the demo.  <br>
+        Note that the model initialization, RL planner, and TTA updates are not fully optimized on GPU for this huggingface demo, and hence may experience some lag during execution. We will improve this in the future.  <br>
+        <a href="https://search-tta.github.io">Project Website</a> 
         """
     )
     # gr.Markdown(

app_multimodal_inference.py

@@ -184,7 +184,8 @@ with gr.Blocks(title="Search-TTA", theme=gr.themes.Base()) as demo:
     gr.Markdown(
         """
         # Search-TTA: A Multimodal Test-Time Adaptation Framework for Visual Search in the Wild Demo
-        <a href="https://search-tta.github.io">Project Website</a> <br>
+        Click on any of the examples below and run the demo.  <br>
+        <a href="https://search-tta.github.io">Project Website</a> 
         """
     )