Update app.py

app.py

@@ -4,47 +4,75 @@ import time
 def bot_response(message, history):
     print("\n=== Bot Response Function ===")
     print(f"Received message: {message}")
-    print(f"Initial history: {history}")
     
     history = history or []
-    messages = []
     
-    # First message
-    print("\nAdding first message pair...")
-    messages.append({"role": "assistant", "content": "Thinking..."})
-    history.extend(messages)
-    print(f"History after first message: {history}")
+    # Regular message without metadata
+    msg = gr.ChatMessage(
+        role="assistant",
+        content="Let me help you analyze that..."
+    )
+    history.append(msg)
+    yield history
+    time.sleep(1)
+    
+    # Tool execution with metadata
+    parent_id = "tool_call_1"
+    tool_msg = gr.ChatMessage(
+        role="assistant",
+        content="```python\nprint('Running analysis...')\n```",
+        metadata={
+            "title": "🛠️ Using Python Interpreter",
+            "id": parent_id,
+            "status": "pending"
+        }
+    )
+    history.append(tool_msg)
     yield history
     time.sleep(1)
     
-    # Second message
-    print("\nAdding second message...")
-    messages = [{"role": "assistant", "content": "This is the part 1 of bot response"}]
-    history.extend(messages)
-    print(f"History after second message: {history}")
+    # Regular message showing progress
+    progress_msg = gr.ChatMessage(
+        role="assistant",
+        content="Processing your request..."
+    )
+    history.append(progress_msg)
     yield history
     time.sleep(1)
     
-    # Third message
-    print("\nAdding third message...")
-    messages = [{"role": "assistant", "content": "And this is part 2"}]
-    history.extend(messages)
-    print(f"History after third message: {history}")
+    # Execution result with metadata
+    execution_msg = gr.ChatMessage(
+        role="assistant",
+        content="Output: Analysis complete",
+        metadata={
+            "title": "📝 Execution Result",
+            "parent_id": parent_id,
+            "status": "pending"
+        }
+    )
+    history.append(execution_msg)
     yield history
     time.sleep(1)
     
-    # Fourth message
-    print("\nAdding fourth message...")
-    messages = [{"role": "assistant", "content": "LAstly, some code:\n```python\nprint('hello world')\n```"}]
-    history.extend(messages)
-    print(f"History after fourth message: {history}")
+    # Update tool messages to done
+    tool_msg.metadata["status"] = "done"
+    execution_msg.metadata["status"] = "done"
+    yield history
+    
+    # Final message without metadata
+    final_msg = gr.ChatMessage(
+        role="assistant",
+        content="Based on the analysis, I can confirm the process completed successfully."
+    )
+    history.append(final_msg)
     yield history
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot(
         height=600,
         group_consecutive_messages=False,
-        type="messages"
+        type="messages",
+        bubble_full_width=False
     )
     msg = gr.Textbox()
     clear = gr.Button("Clear")
@@ -52,13 +80,10 @@ with gr.Blocks() as demo:
     def user(user_message, history):
         print("\n=== User Message Function ===")
         print(f"User message: {user_message}")
-        print(f"Current history: {history}")
         
-        # Only add user message if it's not empty
         if user_message.strip():
-            new_history = history + [{"role": "user", "content": user_message}]
-            print(f"New history after adding user message: {new_history}")
-            return "", new_history
+            history.append(gr.ChatMessage(role="user", content=user_message))
+            return "", history
         return "", history
 
     msg.submit(