Update app.py

app.py

@@ -3,12 +3,13 @@ import gradio as gr
 import datetime
 import pytz
 import asyncio
-from gradio import ChatMessage
 
-# Frameworks
+# Framework 1: LlamaIndex
 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
@@ -16,7 +17,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL_ID = "Qwen/Qwen2.5-7B-Instruct"
 
 # ==========================================
-# AGENT SETUP
+# PART 1: LLAMAINDEX AGENT
 # ==========================================
 li_llm = HuggingFaceInferenceAPI(model_name=MODEL_ID, token=HF_TOKEN, provider="together")
 
@@ -26,91 +27,79 @@ def get_tokyo_time() -> str:
     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."""
-    return f"The weather in {location} is currently sunny and 22°C."
+    """
+    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_agent = AgentWorkflow.from_tools_or_functions(
-    [FunctionTool.from_defaults(fn=get_tokyo_time), FunctionTool.from_defaults(fn=get_weather)],
-    llm=li_llm
-)
+li_tools = [
+    FunctionTool.from_defaults(fn=get_tokyo_time),
+    FunctionTool.from_defaults(fn=get_weather) # Added weather tool here
+]
 
-smol_model = InferenceClientModel(model_id=MODEL_ID, token=HF_TOKEN, provider="together")
-smol_agent = CodeAgent(model=smol_model, tools=[DuckDuckGoSearchTool()])
+li_agent = AgentWorkflow.from_tools_or_functions(li_tools, 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)}"
 
 # ==========================================
-# ENHANCED CHAT LOGIC
+# PART 2: SMOLAGENTS
 # ==========================================
-async def chat_llama(message, history, system_prompt):
-    # LlamaIndex doesn't natively stream thoughts as easily as smolagents in this setup,
-    # so we just return the final result.
-    result = await li_agent.run(user_msg=f"{system_prompt}\n\nUser: {message}")
-    return str(result)
-
-def chat_smol(message, history, system_prompt):
-    # 1. Yield a 'Thinking' state first
-    yield [
-        ChatMessage(role="user", content=message),
-        ChatMessage(
-            role="assistant", 
-            content="Writing Python code to analyze your request...",
-            metadata={"title": "🧠 CodeAgent Thinking", "status": "pending"}
-        )
-    ]
-    
+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
+)
+
+def chat_smol(message, history):
     try:
-        # 2. Run the agent logic
-        response = smol_agent.run(f"{system_prompt}\n\nUser: {message}")
-        
-        # 3. Final Answer
-        yield [
-            ChatMessage(role="user", content=message),
-            ChatMessage(role="assistant", content=str(response))
-        ]
+        response = smol_agent.run(message)
+        return str(response)
     except Exception as e:
-        yield [ChatMessage(role="assistant", content=f"❌ Error: {str(e)}")]
+        return f"❌ Smolagents Error: {str(e)}"
 
 # ==========================================
-# GRADIO 6.0 UI DESIGN
+# PART 3: GRADIO 6.0 UI
 # ==========================================
-with gr.Blocks(fill_height=True, title="Agent Workspace") as demo:
+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.")
     
-    # Sidebar for Global Control
-    with gr.Sidebar():
-        gr.Markdown("# ⚙️ Configuration")
-        system_input = gr.Textbox(
-            label="System Instructions",
-            placeholder="e.g. You are a helpful assistant that speaks like a pirate.",
-            lines=4
-        )
-        gr.Markdown("---")
-        gr.Markdown("### 🛠️ Active Tools")
-        gr.Markdown("- 🕒 Tokyo Time\n- ☀️ Weather (LlamaIndex)\n- 🔍 DuckDuckGo (smolagents)")
-        gr.Divider()
-        gr.Info("LlamaIndex focuses on 'Workflows', while smolagents focuses on 'Code Generation'.")
-
-    # Main Area
-    with gr.Column():
-        gr.Markdown(f"# 🤖 AI Agent Hub\n**Model:** `{MODEL_ID}`")
+    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.Tabs():
-            # TAB 1: LLAMAINDEX
-            with gr.Tab("🏗️ LlamaIndex Workflow"):
-                gr.ChatInterface(
-                    fn=chat_llama,
-                    additional_inputs=[system_input],
-                    examples=["What time is it in Tokyo?", "Weather in London"],
-                    type="messages" # Gradio 6.0 style
-                )
-
-            # TAB 2: SMOLAGENTS
-            with gr.Tab("💻 smolagents (CodeAgent)"):
-                # Manual Chatbot for 'Thinking' metadata support
-                smol_chatbot = gr.Chatbot(label="smol-code-agent", type="messages", avatar_images=(None, "https://raw.githubusercontent.com/huggingface/smolagents/main/docs/source/imgs/smolagents_logo.png"))
-                smol_msg = gr.Textbox(placeholder="Ask me to search or calculate something...", label="Input")
-                
-                # Handling events manually for the 'Thinking' yield
-                smol_msg.submit(chat_smol, [smol_msg, smol_chatbot, system_input], [smol_chatbot])
-                smol_msg.submit(lambda: "", None, [smol_msg]) # Clear textbox
+        with gr.Tab("💻 smolagents (CodeAgent)"):
+            gr.ChatInterface(
+                fn=chat_smol,
+                examples=["Search for the latest AI news", "How is the weather in Paris?"]
+            )
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft())
+    # Gradio 6.0: Move theme/css to launch()
+    demo.launch(
+        theme=gr.themes.Soft(), 
+        css="#main-title { text-align: center; margin-bottom: 20px; }"
+    )