Update app.py

app.py

@@ -2,23 +2,21 @@ import os
 import gradio as gr
 import datetime
 import pytz
-import math
 import requests
-from deep_translator import GoogleTranslator
 
-# Framework 1: LlamaIndex (Core components for Agent Construction)
-from llama_index.core.agent import AgentRunner, ReActAgentWorker
+# Framework 1: LlamaIndex - Using the most direct, stable imports
+from llama_index.core.agent import ReActAgent
 from llama_index.core.tools import FunctionTool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 
-# Framework 2: smolagents (Updated Model Class)
+# Framework 2: smolagents
 from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
 
 # 0. SHARED CONFIG
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 # ==========================================
-# PART 1: LLAMAINDEX AGENT (Fixed Initialization)
+# PART 1: LLAMAINDEX AGENT
 # ==========================================
 li_llm = HuggingFaceInferenceAPI(
     model_name="Qwen/Qwen2.5-7B-Instruct",
@@ -27,20 +25,18 @@ li_llm = HuggingFaceInferenceAPI(
 )
 
 def get_tokyo_time() -> str:
-    """Returns the current time in Tokyo."""
+    """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')}"
+    return f"The current time is {datetime.datetime.now(tz).strftime('%H:%M:%S')}"
 
 li_tools = [FunctionTool.from_defaults(fn=get_tokyo_time)]
 
-# FIX: Using ReActAgentWorker + AgentRunner instead of ReActAgent.from_tools
-# This bypasses the Pydantic AttributeError issue
-worker = ReActAgentWorker.from_tools(
-    tools=li_tools, 
-    llm=li_llm, 
+# Using the class-based initialization which is now corrected in latest 0.10+ versions
+li_agent = ReActAgent.from_tools(
+    tools=li_tools,
+    llm=li_llm,
     verbose=True
 )
-li_agent = AgentRunner(worker)
 
 def chat_llama(message, history):
     try:
@@ -63,12 +59,11 @@ def weather_tool(location: str) -> str:
     Args:
         location: The city name.
     """
-    return f"The weather in {location} is sunny and 25°C."
+    return f"The weather in {location} is currently sunny and 22°C."
 
 smol_agent = CodeAgent(
     model=smol_model,
-    tools=[weather_tool, DuckDuckGoSearchTool()],
-    additional_authorized_imports=['math', 'requests', 'pytz', 'datetime']
+    tools=[weather_tool, DuckDuckGoSearchTool()]
 )
 
 def chat_smol(message, history):
@@ -82,13 +77,12 @@ def chat_smol(message, history):
 # PART 3: UNIFIED GRADIO UI
 # ==========================================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🤖 Multi-Framework Agent Space")
-    gr.Markdown("Compare the two leading agent frameworks side-by-side.")
+    gr.Markdown("# 🤖 Multi-Framework AI Space")
     
-    with gr.Tab("LlamaIndex (ReAct Agent)"):
+    with gr.Tab("LlamaIndex (ReAct)"):
         gr.ChatInterface(fn=chat_llama)
         
-    with gr.Tab("smolagents (Code Agent)"):
+    with gr.Tab("smolagents (Code)"):
         gr.ChatInterface(fn=chat_smol)
 
 if __name__ == "__main__":