Update app.py

app.py

@@ -3,7 +3,6 @@ import gradio as gr
 import datetime
 import pytz
 import asyncio
-from gradio import ChatMessage
 
 # Framework 1: LlamaIndex
 from llama_index.core.agent.workflow import AgentWorkflow
@@ -17,70 +16,96 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientMod
 HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL_ID = "Qwen/Qwen2.5-7B-Instruct"
 
+custom_css = """
+.agent-header { text-align: center; margin-bottom: 20px; }
+"""
+
 # ==========================================
-# PART 1: AGENT LOGIC
+# PART 1: LLAMAINDEX AGENT
 # ==========================================
 li_llm = HuggingFaceInferenceAPI(model_name=MODEL_ID, token=HF_TOKEN, provider="together")
+
+def get_tokyo_time() -> str:
+    """Returns the current time in Tokyo, Japan."""
+    tz = pytz.timezone('Asia/Tokyo')
+    return f"The current time in Tokyo is {datetime.datetime.now(tz).strftime('%H:%M:%S')}"
+
 li_agent = AgentWorkflow.from_tools_or_functions(
-    [FunctionTool.from_defaults(fn=lambda: f"Current time: {datetime.datetime.now(pytz.timezone('Asia/Tokyo'))}")],
+    [FunctionTool.from_defaults(fn=get_tokyo_time)],
     llm=li_llm,
 )
 
-smol_model = InferenceClientModel(model_id=MODEL_ID, token=HF_TOKEN, provider="together")
-smol_agent = CodeAgent(model=smol_model, tools=[DuckDuckGoSearchTool()])
+async def chat_llama(message, history):
+    try:
+        result = await li_agent.run(user_msg=message)
+        return str(result)
+    except Exception as e:
+        return f"❌ LlamaIndex Error: {str(e)}"
 
 # ==========================================
-# PART 2: UI WRAPPERS (Gradio 6.0 Style)
+# PART 2: SMOLAGENTS
 # ==========================================
-async def chat_llama(message, history):
-    # Workflow-based logic
-    result = await li_agent.run(user_msg=message)
-    return str(result)
+smol_model = InferenceClientModel(model_id=MODEL_ID, token=HF_TOKEN, provider="together")
+
+@tool
+def weather_tool(location: str) -> str:
+    """
+    Get the current weather for a location.
+
+    Args:
+        location: The city and country, e.g., 'London, UK'.
+    """
+    return f"The weather in {location} is currently sunny and 22°C."
+
+smol_agent = CodeAgent(
+    model=smol_model,
+    tools=[weather_tool, DuckDuckGoSearchTool()],
+    add_base_tools=True
+)
 
 def chat_smol(message, history):
-    # We yield a 'Thought' block first
-    yield ChatMessage(
-        role="assistant", 
-        content="I am searching the web and writing Python code to verify the details...",
-        metadata={"title": "🧠 Thinking Process", "status": "pending"}
-    )
-    
     try:
-        # Run agent
         response = smol_agent.run(message)
-        # Yield the final result (this replaces the previous yield in the UI)
-        yield ChatMessage(role="assistant", content=str(response))
+        return str(response)
     except Exception as e:
-        yield ChatMessage(role="assistant", content=f"❌ Error: {str(e)}")
+        return f"❌ Smolagents Error: {str(e)}"
 
 # ==========================================
-# PART 3: THE UI LAYOUT
+# PART 3: REVISED GRADIO UI (v6.0 Compatible)
 # ==========================================
-with gr.Blocks(fill_height=True) as demo:
-    with gr.Row():
-        gr.Markdown(f"# 🤖 Multi-Agent Hub\nRunning `{MODEL_ID}` via Together AI", elem_id="header")
+with gr.Blocks() as demo:
+    
+    gr.Markdown(f"""
+    # 🤖 Multi-Framework Agent Playground
+    **Model:** `{MODEL_ID}` | **Provider:** `Together AI`
+    """, elem_classes="agent-header")
 
     with gr.Tabs():
-        with gr.Tab("🏗️ LlamaIndex (Workflow)"):
-            gr.ChatInterface(
-                fn=chat_llama,
-                examples=["What time is it in Tokyo?"],
-                # 'container' removed to fix your error
-            )
-            gr.Info("Best for structured, multi-step business logic.")
-
-        with gr.Tab("💻 smolagents (Code)"):
-            gr.ChatInterface(
-                fn=chat_smol,
-                examples=["Search for the latest stock price of Nvidia."],
-                # Customizing the chatbot component to look cleaner
-                chatbot=gr.Chatbot(label="Code Execution Agent", show_label=False)
-            )
-            gr.Warning("Best for dynamic problem solving and web searching.")
-
-# Launch with theme and CSS
+        # TAB 1: LLAMAINDEX
+        with gr.Tab("🏗️ LlamaIndex Workflow"):
+            with gr.Row():
+                with gr.Column(scale=4):
+                    # Removed 'type' argument for compatibility
+                    gr.ChatInterface(
+                        fn=chat_llama,
+                        examples=["What time is it in Tokyo?"]
+                    )
+                with gr.Column(scale=1):
+                    gr.Markdown("### Agent Specs")
+                    gr.Info("LlamaIndex handles complex multi-step logic via event-driven workflows.")
+
+        # TAB 2: SMOLAGENTS
+        with gr.Tab("💻 smolagents CodeAgent"):
+            with gr.Row():
+                with gr.Column(scale=4):
+                    gr.ChatInterface(
+                        fn=chat_smol,
+                        examples=["What is the weather in Paris?"]
+                    )
+                with gr.Column(scale=1):
+                    gr.Markdown("### Agent Specs")
+                    gr.Warning("smolagents writes and executes Python code locally.")
+
 if __name__ == "__main__":
-    demo.launch(
-        theme=gr.themes.Soft(),
-        css="#header { text-align: center; padding: 20px; }"
-    )
+    # Moved theme and css here for Gradio 6.0
+    demo.launch(theme=gr.themes.Soft(), css=custom_css)