Update app.py

app.py

@@ -3,103 +3,107 @@ import gradio as gr
 import datetime
 import pytz
 import asyncio
+from gradio import ChatMessage
 
-# Framework 1: LlamaIndex
+# Frameworks
 from llama_index.core.agent.workflow import AgentWorkflow
 from llama_index.core.tools import FunctionTool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
-
-# Framework 2: smolagents
 from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
 
-# 0. SHARED CONFIG
+# 0. CONFIG
 HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL_ID = "Qwen/Qwen2.5-7B-Instruct"
 
 # ==========================================
-# PART 1: LLAMAINDEX AGENT
+# AGENT SETUP
 # ==========================================
 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')}"
-
 def get_weather(location: str) -> str:
-    """
-    Get the current weather for a specific location.
-    
-    Args:
-        location: The city name to check weather for.
-    """
-    return f"LlamaIndex Tool: The weather in {location} is currently sunny and 22°C."
-
-li_tools = [
-    FunctionTool.from_defaults(fn=get_tokyo_time),
-    FunctionTool.from_defaults(fn=get_weather) # Added weather tool here
-]
+    """Get the current weather. Args: location: city name."""
+    return f"The weather in {location} is currently sunny and 22°C."
 
-li_agent = AgentWorkflow.from_tools_or_functions(li_tools, llm=li_llm)
+li_agent = AgentWorkflow.from_tools_or_functions(
+    [FunctionTool.from_defaults(fn=get_weather)],
+    llm=li_llm
+)
 
-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)}"
+smol_model = InferenceClientModel(model_id=MODEL_ID, token=HF_TOKEN, provider="together")
+smol_agent = CodeAgent(model=smol_model, tools=[DuckDuckGoSearchTool()])
 
 # ==========================================
-# PART 2: SMOLAGENTS
+# UI LOGIC
 # ==========================================
-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"smolagents Tool: The weather in {location} is currently sunny and 22°C."
-
-smol_agent = CodeAgent(
-    model=smol_model,
-    tools=[weather_tool, DuckDuckGoSearchTool()],
-    add_base_tools=True
-)
+async def chat_llama(message, history):
+    result = await li_agent.run(user_msg=message)
+    return str(result)
 
 def chat_smol(message, history):
+    # Yield a 'Thinking' status using ChatMessage metadata
+    yield [
+        ChatMessage(role="user", content=message),
+        ChatMessage(
+            role="assistant", 
+            content="Agent is searching and executing Python code...",
+            metadata={"title": "🔍 Step 1: Thinking", "status": "pending"}
+        )
+    ]
     try:
         response = smol_agent.run(message)
-        return str(response)
+        yield [
+            ChatMessage(role="user", content=message),
+            ChatMessage(role="assistant", content=str(response))
+        ]
     except Exception as e:
-        return f"❌ Smolagents Error: {str(e)}"
+        yield [ChatMessage(role="assistant", content=f"Error: {str(e)}")]
 
 # ==========================================
-# PART 3: GRADIO 6.0 UI
+# GRADIO 6.0 ENHANCED UI
 # ==========================================
-with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 Consolidated AI Agent Space", elem_id="main-title")
-    gr.Markdown(f"Currently using **{MODEL_ID}** via Together AI Provider.")
-    
-    with gr.Tabs():
-        with gr.Tab("🏗️ LlamaIndex (Workflow)"):
-            gr.ChatInterface(
-                fn=chat_llama,
-                examples=["What's the weather in Tokyo?", "What time is it in Japan?"]
-            )
+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.HTML("<hr>") # This replaces gr.Divider()
+        
+        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.")
+
+    # Main Chat View
+    with gr.Column():
+        gr.Markdown(f"# 🤖 Multi-Agent Hub\nPowered by `{MODEL_ID}`")
         
-        with gr.Tab("💻 smolagents (CodeAgent)"):
-            gr.ChatInterface(
-                fn=chat_smol,
-                examples=["Search for the latest AI news", "How is the weather in Paris?"]
-            )
+        with gr.Tabs():
+            with gr.Tab("🏗️ LlamaIndex (Workflow)"):
+                gr.ChatInterface(
+                    fn=chat_llama,
+                    type="messages",
+                    examples=["How is the weather in Tokyo?"]
+                )
+
+            with gr.Tab("💻 smolagents (CodeAgent)"):
+                # Custom implementation for thought metadata
+                chatbot = gr.Chatbot(
+                    label="Code Agent", 
+                    type="messages", 
+                    show_label=False,
+                    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)
+                
+                msg.submit(chat_smol, [msg, chatbot], [chatbot])
+                msg.submit(lambda: "", None, [msg])
 
 if __name__ == "__main__":
-    # Gradio 6.0: Move theme/css to launch()
+    # Custom CSS for the Title
     demo.launch(
-        theme=gr.themes.Soft(), 
-        css="#main-title { text-align: center; margin-bottom: 20px; }"
+        theme=gr.themes.Soft(),
+        css=".gradio-container { max-width: 1200px !important; }"
     )