Refactor respond function to use structured history format and enhance chatbot UI properties

src/main.py

@@ -26,12 +26,12 @@ def _build_clarification_block(questions: list[str], answers: str) -> str:
 
 
 async def respond(
-    message: str, history: list[tuple[str | None, str | None]], state: dict
+    message: str, history: list[dict], state: dict
 ):
     if not state:
         state = {"stage": "awaiting_query", "query": "", "questions": []}
 
-    history.append((message, None))
+    history.append({"role": "user", "content": message})
 
     if state["stage"] == "awaiting_query":
         state["query"] = message
@@ -42,17 +42,17 @@ async def respond(
             state["stage"] = "awaiting_answers"
             q_text = _format_clarifications(questions)
             history.append(
-                (
-                    None,
-                    f"Please answer the following questions, one per line:\n{q_text}",
-                )
+                {
+                    "role": "assistant",
+                    "content": f"Please answer the following questions, one per line:\n{q_text}",
+                }
             )
             yield history, state
         else:
             state["stage"] = "running"
-            history.append((None, ""))
+            history.append({"role": "assistant", "content": ""})
             async for chunk in manager.run(message, ""):
-                history[-1] = (None, (history[-1][1] or "") + chunk)
+                history[-1]["content"] = (history[-1]["content"] or "") + chunk
                 yield history, state
             state["stage"] = "awaiting_query"
             yield history, state
@@ -60,27 +60,33 @@ async def respond(
         answers_block = _build_clarification_block(state["questions"], message)
         manager = ResearchManager()
         state["stage"] = "running"
-        history.append((None, ""))
+        history.append({"role": "assistant", "content": ""})
         async for chunk in manager.run(state["query"], answers_block):
-            history[-1] = (None, (history[-1][1] or "") + chunk)
+            history[-1]["content"] = (history[-1]["content"] or "") + chunk
             yield history, state
         state["stage"] = "awaiting_query"
         yield history, state
     else:
-        history.append((None, "Please wait for the current task to finish."))
+        history.append({"role": "assistant", "content": "Please wait for the current task to finish."})
         yield history, state
 
 
 def reset():
-    return [(None, WELCOME_MESSAGE)], {
+    return ([{"role": "assistant", "content": WELCOME_MESSAGE}], {
         "stage": "awaiting_query",
         "query": "",
         "questions": [],
-    }
+    })
 
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue="yellow")) as ui:
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(
+        label="Deep Research",
+        height=500,
+        resizable=True,
+        show_copy_button=True,
+        type="messages",
+    )
     state = gr.State({})
     msg = gr.Textbox(placeholder="Type your message and press Enter")