Update app.py

app.py

@@ -12,71 +12,94 @@ from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 # Framework 2: smolagents
 from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
 
-# 0. CONFIG
+# 0. SHARED CONFIG
 HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL_ID = "Qwen/Qwen2.5-7B-Instruct"
 
 # ==========================================
-# AGENTS SETUP
+# PART 1: LLAMAINDEX AGENT
 # ==========================================
-
-# LlamaIndex 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_li(location: str) -> str:
-    """Get current weather for a location. Args: location: city name."""
-    return f"LlamaIndex reports: The weather in {location} is 22°C and sunny."
+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_agent = AgentWorkflow.from_tools_or_functions(
-    [FunctionTool.from_defaults(fn=get_tokyo_time), FunctionTool.from_defaults(fn=get_weather_li)],
-    llm=li_llm,
-)
+li_tools = [
+    FunctionTool.from_defaults(fn=get_tokyo_time),
+    FunctionTool.from_defaults(fn=get_weather) # Added weather tool here
+]
+
+li_agent = AgentWorkflow.from_tools_or_functions(li_tools, llm=li_llm)
 
-# smolagents Setup
+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: SMOLAGENTS
+# ==========================================
 smol_model = InferenceClientModel(model_id=MODEL_ID, token=HF_TOKEN, provider="together")
+
 @tool
 def weather_tool(location: str) -> str:
-    """Get current weather. Args: location: city name."""
-    return f"smolagents reports: The weather in {location} is 22°C and sunny."
+    """
+    Get the current weather for a location.
 
-smol_agent = CodeAgent(model=smol_model, tools=[weather_tool, DuckDuckGoSearchTool()])
+    Args:
+        location: The city and country, e.g., 'London, UK'.
+    """
+    return f"smolagents Tool: The weather in {location} is currently sunny and 22°C."
 
-# ==========================================
-# CHAT FUNCTIONS
-# ==========================================
-async def chat_llama(message, history):
-    result = await li_agent.run(user_msg=message)
-    return str(result)
+smol_agent = CodeAgent(
+    model=smol_model,
+    tools=[weather_tool, DuckDuckGoSearchTool()],
+    add_base_tools=True
+)
 
 def chat_smol(message, history):
-    return str(smol_agent.run(message))
+    try:
+        response = smol_agent.run(message)
+        return str(response)
+    except Exception as e:
+        return f"❌ Smolagents Error: {str(e)}"
 
 # ==========================================
-# IMPROVED UI
+# PART 3: GRADIO 6.0 UI
 # ==========================================
-with gr.Blocks(theme=gr.themes.Soft(), css=".agent-header {text-align:center} .sidebar-info {font-size: 0.9em}") as demo:
-    with gr.Row(elem_classes="agent-header"):
-        gr.Markdown(f"# 🤖 Agentic Framework Workspace\n**Backend:** {MODEL_ID} via Together AI")
-
-    with gr.Row():
-        with gr.Column(scale=1):
-            with gr.Group():
-                gr.Markdown("### 🛠️ Available Tools")
-                gr.Markdown("**LlamaIndex:**\n- `tokyo_time`\n- `get_weather`", elem_classes="sidebar-info")
-                gr.Markdown("**smolagents:**\n- `weather_tool`\n- `DuckDuckGo Search`\n- `Python Interpreter`", elem_classes="sidebar-info")
-            gr.Info("LlamaIndex uses Event-Driven Workflows, while smolagents generates and executes Python code directly.")
-
-        with gr.Column(scale=4):
-            with gr.Tabs():
-                with gr.Tab("🏗️ LlamaIndex (Workflow)"):
-                    gr.ChatInterface(fn=chat_llama, examples=["How is the weather in Tokyo?", "What time is it in Tokyo?"])
-                
-                with gr.Tab("💻 smolagents (CodeAgent)"):
-                    gr.ChatInterface(fn=chat_smol, examples=["Search for 2026 AI trends", "What is the weather in London?"])
+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.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()
+    # Gradio 6.0: Move theme/css to launch()
+    demo.launch(
+        theme=gr.themes.Soft(), 
+        css="#main-title { text-align: center; margin-bottom: 20px; }"
+    )