Update app.py

app.py

@@ -1,91 +1,105 @@
 import os
-import gradio as gr
 import datetime
 import pytz
-import asyncio
-
-# Framework 1: LlamaIndex
-from llama_index.core.agent.workflow import AgentWorkflow
+import math
+import requests
+import gradio as gr
+from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
+from deep_translator import GoogleTranslator
+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
-from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
-
-# 0. SHARED CONFIG
-HF_TOKEN = os.getenv("HF_TOKEN")
-
-# ==========================================
-# PART 1: LLAMAINDEX AGENT
-# ==========================================
-li_llm = HuggingFaceInferenceAPI(
-    model_name="Qwen/Qwen2.5-7B-Instruct",
-    token=HF_TOKEN,
-)
-
-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')}"
+# ============ 1. SMOLAGENTS TOOLS ============
 
-li_tools = [FunctionTool.from_defaults(fn=get_tokyo_time)]
-
-# Using positional argument for tools list to avoid TypeError
-li_agent = AgentWorkflow.from_tools_or_functions(
-    li_tools,
-    llm=li_llm,
-)
-
-async def chat_llama(message, history):
+@tool
+def time_tool(timezone: str = "UTC") -> str:
+    """
+    Get current time in a specific timezone.
+    Args:
+        timezone: The timezone to check (e.g., 'Asia/Tokyo').
+    """
     try:
-        # AgentWorkflow requires async execution
-        result = await li_agent.run(user_msg=message)
-        return str(result)
+        tz = pytz.timezone(timezone)
+        now = datetime.datetime.now(tz)
+        return f"⏰ Time in {timezone}: {now.strftime('%Y-%m-%d %H:%M:%S')}"
     except Exception as e:
-        return f"LlamaIndex Error: {str(e)}"
+        return f"Error: {str(e)}"
 
-# ==========================================
-# PART 2: SMOLAGENTS
-# ==========================================
-smol_model = InferenceClientModel(
-    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    token=HF_TOKEN
-)
+@tool
+def translator_tool(text: str, target_language: str) -> str:
+    """
+    Translates text into a specified language.
+    Args:
+        text: The text or phrase to translate.
+        target_language: The destination language (e.g., 'french', 'german').
+    """
+    try:
+        translation = GoogleTranslator(source='auto', target=target_language).translate(text)
+        return f"Translated to {target_language.title()}: {translation}"
+    except Exception as e:
+        return f"Translation error: {str(e)}"
 
 @tool
 def weather_tool(location: str) -> str:
-    """Get the current weather for a location.
+    """
+    Get the current weather for any location worldwide.
     Args:
-        location: The city name.
+        location: The name of the city or place (e.g., 'London' or 'Tokyo').
     """
-    return f"The weather in {location} is currently sunny and 22°C."
+    try:
+        geo_url = f"https://geocoding-api.open-meteo.com/v1/search?name={location}&count=1&language=en&format=json"
+        geo_res = requests.get(geo_url).json()
+        if not geo_res.get('results'): return "Location not found."
+        data = geo_res['results'][0]
+        weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={data['latitude']}&longitude={data['longitude']}&current=temperature_2m"
+        w_res = requests.get(weather_url).json()
+        return f"🌤️ {data['name']}: {w_res['current']['temperature_2m']}°C"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# ============ AGENT CONFIGURATION ============
+
+HF_TOKEN = os.getenv("HF_TOKEN", "")
+smol_model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=HF_TOKEN)
 
-# Tool initialization happens here; ensure ddgs is in requirements.txt
 smol_agent = CodeAgent(
     model=smol_model,
-    tools=[weather_tool, DuckDuckGoSearchTool()]
+    tools=[DuckDuckGoSearchTool(), time_tool, translator_tool, weather_tool],
+    max_steps=5
 )
 
+# LlamaIndex Setup
+llama_llm = HuggingFaceInferenceAPI(
+    model_name="Qwen/Qwen2.5-7B-Instruct", 
+    token=HF_TOKEN,
+    task="conversational"
+)
+
+llama_agent = ReActAgent.from_tools(
+    [
+        FunctionTool.from_defaults(fn=lambda: f"Tokyo: {datetime.datetime.now(pytz.timezone('Asia/Tokyo'))}", name="get_tokyo_time"),
+        FunctionTool.from_defaults(fn=lambda a, b: a * b, name="multiply")
+    ],
+    llm=llama_llm,
+    verbose=True
+)
+
+# ============ GRADIO UI ============
+
 def chat_smol(message, history):
-    try:
-        # CodeAgent .run() is synchronous
-        response = smol_agent.run(message)
-        return str(response)
-    except Exception as e:
-        return f"Smolagents Error: {str(e)}"
+    return str(smol_agent.run(message))
+
+def chat_llama(message, history):
+    response = llama_agent.chat(message)
+    return str(response)
 
-# ==========================================
-# PART 3: UNIFIED GRADIO UI
-# ==========================================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🤖 Dual-Agent Testing Space")
-    gr.Markdown("Comparing **Event-Driven Workflows** (LlamaIndex) vs **Code Execution** (smolagents).")
-    
-    with gr.Tab("LlamaIndex (Workflow)"):
-        gr.ChatInterface(fn=chat_llama)
-        
-    with gr.Tab("smolagents (CodeAgent)"):
-        gr.ChatInterface(fn=chat_smol)
+    gr.Markdown("# 🤖 Multi-Framework AI Agent Space")
+    with gr.Tab("Smolagents (Alfred)"):
+        gr.ChatInterface(chat_smol)
+    with gr.Tab("LlamaIndex (ReAct)"):
+        gr.ChatInterface(chat_llama)
 
 if __name__ == "__main__":
     demo.launch()