Update app.py

app.py

@@ -1,56 +1,97 @@
 import os
 import gradio as gr
+import datetime
+import pytz
+import math
+import requests
+from deep_translator import GoogleTranslator
+
+# Framework 1: LlamaIndex
 from llama_index.core.agent import ReActAgent
 from llama_index.core.tools import FunctionTool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
-from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool
 
-# 1. SETUP LLM
+# Framework 2: smolagents
+# FIXED IMPORT: HfApiModel is now InferenceClientModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
+
+# 0. SHARED CONFIG
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# --- LLAMAINDEX SETUP ---
+# ==========================================
+# PART 1: LLAMAINDEX AGENT
+# ==========================================
 li_llm = HuggingFaceInferenceAPI(
-    model_name="Qwen/Qwen2.5-7B-Instruct", 
+    model_name="Qwen/Qwen2.5-7B-Instruct",
     token=HF_TOKEN,
     task="conversational"
 )
 
-# Define simple tools for LlamaIndex
-def multiply(a: float, b: float) -> float:
-    """Multiplies two numbers."""
-    return a * b
+def get_tokyo_time() -> str:
+    """Returns the current time in Tokyo."""
+    tz = pytz.timezone('Asia/Tokyo')
+    return f"The current time in Tokyo is {datetime.datetime.now(tz).strftime('%H:%M:%S')}"
 
-li_tools = [FunctionTool.from_defaults(fn=multiply)]
+li_tools = [FunctionTool.from_defaults(fn=get_tokyo_time)]
 
-# Fix: Ensure correct initialization
+# Initializing via constructor to avoid Pydantic __getattr__ issues
 li_agent = ReActAgent.from_tools(
-    tools=li_tools, 
-    llm=li_llm, 
+    tools=li_tools,
+    llm=li_llm,
     verbose=True
 )
 
-# --- SMOLAGENTS SETUP ---
-smol_model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=HF_TOKEN)
-smol_agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=smol_model)
-
-# 2. DEFINE WRAPPER FUNCTIONS FOR GRADIO
 def chat_llama(message, history):
-    # LlamaIndex .chat() keeps its own history if using the same agent object
-    response = li_agent.chat(message)
-    return str(response)
+    try:
+        response = li_agent.chat(message)
+        return str(response)
+    except Exception as e:
+        return f"LlamaIndex Error: {str(e)}"
+
+# ==========================================
+# PART 2: SMOLAGENTS
+# ==========================================
+# Using the corrected model class
+smol_model = InferenceClientModel(
+    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    token=HF_TOKEN
+)
+
+@tool
+def weather_tool(location: str) -> str:
+    """Get the current weather for a location.
+    Args:
+        location: The city name.
+    """
+    return f"The weather in {location} is sunny and 25°C."
+
+smol_agent = CodeAgent(
+    model=smol_model,
+    tools=[weather_tool, DuckDuckGoSearchTool()],
+    additional_authorized_imports=['math', 'requests']
+)
 
 def chat_smol(message, history):
-    # smolagents .run()
-    response = smol_agent.run(message)
-    return str(response)
-
-# 3. COMBINE INTO TABS
-with gr.Blocks() as demo:
-    gr.Markdown("# Dual Agent Interface: LlamaIndex vs Smolagents")
-    with gr.Tab("LlamaIndex (ReAct)"):
-        gr.ChatInterface(chat_llama)
-    with gr.Tab("Smolagents (Code)"):
-        gr.ChatInterface(chat_smol)
+    try:
+        response = smol_agent.run(message)
+        return str(response)
+    except Exception as e:
+        return f"Smolagents Error: {str(e)}"
+
+# ==========================================
+# PART 3: UNIFIED GRADIO UI
+# ==========================================
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🤖 Multi-Framework Agent Space")
+    gr.Markdown("Compare how different agentic frameworks handle your requests.")
+    
+    with gr.Tab("LlamaIndex (ReAct Agent)"):
+        gr.Markdown("This agent uses the classic **Reasoning + Acting** text loop.")
+        gr.ChatInterface(fn=chat_llama)
+        
+    with gr.Tab("smolagents (Code Agent)"):
+        gr.Markdown("This agent solves tasks by writing and executing **Python code**.")
+        gr.ChatInterface(fn=chat_smol)
 
 if __name__ == "__main__":
     demo.launch()