Update app.py

app.py

@@ -36,11 +36,12 @@ smol_agent = CodeAgent(model=smol_model, tools=[DuckDuckGoSearchTool()])
 # UI LOGIC
 # ==========================================
 async def chat_llama(message, history):
+    # History in Gradio 6 is already a list of dicts
     result = await li_agent.run(user_msg=message)
     return str(result)
 
 def chat_smol(message, history):
-    # Yield a 'Thinking' status using ChatMessage metadata
+    # Using the new ChatMessage format for metadata thoughts
     yield [
         ChatMessage(role="user", content=message),
         ChatMessage(
@@ -62,48 +63,39 @@ def chat_smol(message, history):
 # GRADIO 6.0 ENHANCED UI
 # ==========================================
 with gr.Blocks(fill_height=True) as demo:
-    # Sidebar replaces global config area
     with gr.Sidebar():
         gr.Markdown("# 🛠️ Workspace Settings")
-        gr.Markdown("Adjust your agent behaviors and view tool status below.")
+        gr.Separator() # Correct replacement for gr.Divider()
         
-        gr.HTML("<hr>") # This replaces gr.Divider()
+        gr.Markdown("### 🤖 Framework Info")
+        gr.Info("LlamaIndex: Workflow-based.\nsmolagents: Code-based.")
         
-        gr.Markdown("### 🤖 Active Frameworks")
-        gr.CheckboxGroup(["LlamaIndex", "smolagents"], label="Enabled Agents", value=["LlamaIndex", "smolagents"])
-        
-        gr.HTML("<hr>")
-        
-        gr.Info("LlamaIndex is optimized for business logic. smolagents is built for raw tool execution.")
+        gr.Separator()
+        gr.Markdown("Status: **Online**")
 
-    # Main Chat View
     with gr.Column():
-        gr.Markdown(f"# 🤖 Multi-Agent Hub\nPowered by `{MODEL_ID}`")
+        gr.Markdown(f"# 🤖 Multi-Agent Hub")
         
         with gr.Tabs():
             with gr.Tab("🏗️ LlamaIndex (Workflow)"):
+                # REMOVED type="messages" as it is now default
                 gr.ChatInterface(
                     fn=chat_llama,
-                    type="messages",
                     examples=["How is the weather in Tokyo?"]
                 )
 
             with gr.Tab("💻 smolagents (CodeAgent)"):
-                # Custom implementation for thought metadata
+                # Manual Chatbot to allow for 'Thinking' metadata
                 chatbot = gr.Chatbot(
                     label="Code Agent", 
-                    type="messages", 
-                    show_label=False,
+                    type="messages", # Explicitly allowed here for the component
                     avatar_images=(None, "https://raw.githubusercontent.com/huggingface/smolagents/main/docs/source/imgs/smolagents_logo.png")
                 )
                 msg = gr.Textbox(placeholder="Ask me to search or run code...", container=False)
                 
+                # Setup event handling
                 msg.submit(chat_smol, [msg, chatbot], [chatbot])
                 msg.submit(lambda: "", None, [msg])
 
 if __name__ == "__main__":
-    # Custom CSS for the Title
-    demo.launch(
-        theme=gr.themes.Soft(),
-        css=".gradio-container { max-width: 1200px !important; }"
-    )
+    demo.launch(theme=gr.themes.Soft())