Update UI layout and assets

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -4,4 +4,5 @@
 .vscode
 
 __pycache__/
+image/*.png
 

README.md

@@ -12,4 +12,6 @@ hf_oauth_scopes:
 - inference-api
 ---
 
+Paper: https://arxiv.org/abs/2602.02486
+
 An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).

app.py

@@ -588,43 +588,73 @@ def reset():
 
 with gr.Blocks() as demo:
     gr.Markdown("## RE-TRAC (REcursive TRAjectory Compression) 30B Demo")
-    gr.Markdown(f"Config: `{CONFIG_PATH}`")
-
-    chatbot_kwargs = {"height": 520, "label": "Messages (full stream)"}
-    if CHATBOT_SUPPORTS_GROUPING:
-        chatbot_kwargs["group_consecutive_messages"] = False
-    if CHATBOT_SUPPORTS_MESSAGES:
-        chatbot = gr.Chatbot(type="messages", **chatbot_kwargs)
-    else:
-        chatbot = gr.Chatbot(**chatbot_kwargs)
-    # #region agent log
-    try:
-        attr_type = getattr(chatbot, "type", None)
-        attr_format = getattr(chatbot, "format", None)
-        attr_data_format = getattr(chatbot, "data_format", None)
-        CHATBOT_EXPECTS_MESSAGES = _should_use_messages_format() or (
-            (attr_type == "messages") or (attr_format == "messages") or (attr_data_format == "messages")
-        )
-        _debug_log(
-            "chatbot instance attrs",
-            {
-                "attr_type": attr_type,
-                "attr_format": attr_format,
-                "attr_data_format": attr_data_format,
-                "expects_messages": CHATBOT_EXPECTS_MESSAGES,
-            },
-            "H1",
-        )
-    except Exception:
-        pass
-    # #endregion
-    status = gr.Markdown("Idle.")
-
-    ui_state = gr.State(None)
-
-    query = gr.Textbox(label="Query (only once)", placeholder="Enter your query and press Run")
-    run_btn = gr.Button("Run", variant="primary")
-    reset_btn = gr.Button("Reset")
+    with gr.Row():
+        with gr.Column(scale=6):
+            gr.Markdown(f"Config: `{CONFIG_PATH}`")
+
+            chatbot_kwargs = {"height": 520, "label": "Messages (full stream)"}
+            if CHATBOT_SUPPORTS_GROUPING:
+                chatbot_kwargs["group_consecutive_messages"] = False
+            if CHATBOT_SUPPORTS_MESSAGES:
+                chatbot = gr.Chatbot(type="messages", **chatbot_kwargs)
+            else:
+                chatbot = gr.Chatbot(**chatbot_kwargs)
+            # #region agent log
+            try:
+                attr_type = getattr(chatbot, "type", None)
+                attr_format = getattr(chatbot, "format", None)
+                attr_data_format = getattr(chatbot, "data_format", None)
+                CHATBOT_EXPECTS_MESSAGES = _should_use_messages_format() or (
+                    (attr_type == "messages") or (attr_format == "messages") or (attr_data_format == "messages")
+                )
+                _debug_log(
+                    "chatbot instance attrs",
+                    {
+                        "attr_type": attr_type,
+                        "attr_format": attr_format,
+                        "attr_data_format": attr_data_format,
+                        "expects_messages": CHATBOT_EXPECTS_MESSAGES,
+                    },
+                    "H1",
+                )
+            except Exception:
+                pass
+            # #endregion
+            status = gr.Markdown("Idle.")
+
+            ui_state = gr.State(None)
+
+            query = gr.Textbox(label="Query (only once)", placeholder="Enter your query and press Run")
+            run_btn = gr.Button("Run", variant="primary")
+            reset_btn = gr.Button("Reset")
+        with gr.Column(scale=4):
+            gr.Markdown(
+                """
+## Introduction
+We introduce Re-TRAC, a recursive framework that resolves the inefficiency of isolated trials in agentic search. It not only boosts both commercial and open-source models by 15–20% over ReAct on BrowseComp but also drives SFT-only performance to new heights (30% for 4B, 53% for 30B).
+
+</div>
+<p align="center">
+🤗 <a href="https://github.com/microsoft/InfoAgent/tree/main/retrac">HuggingFace(comming soon)</a> |
+💻 <a href="https://github.com/microsoft/InfoAgent/tree/main/retrac">GitHub</a> | 
+📑 <a href="https://arxiv.org/abs/2602.02486">Paper</a> | 
+🌐 <a href="https://huggingface.co/spaces/JialiangZhu/RE-TRAC">Demo</a>
+</p>
+
+## Method Overview
+"""
+            )
+            gr.Image(
+                value="image/method.png",
+                label="Method Overview",
+                show_label=False,
+            )
+            gr.Markdown("## Main Result")
+            gr.Image(
+                value="image/main_table.png",
+                label="Main Result",
+                show_label=False,
+            )
 
     run_btn.click(
         run_once,