Update app.py

app.py

@@ -3,103 +3,132 @@ import gradio as gr
 import datetime
 import pytz
 import asyncio
-from gradio import ChatMessage
+import requests
 
-# 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. CONFIG
+# 0. SHARED CONFIG
 HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL_ID = "Qwen/Qwen2.5-7B-Instruct"
 
+def get_coordinates(location: str):
+    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 None, None
+    res = geo_res["results"][0]
+    return res["latitude"], res["longitude"]
+
 # ==========================================
-# AGENT SETUP
+# PART 1: LLAMAINDEX AGENT
 # ==========================================
 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(location: str) -> str:
-    """Get the current weather. Args: location: city name."""
-    return f"The weather in {location} is currently sunny and 22°C."
+    """
+    Get the real-time weather for a specific location.
+    Args:
+        location: The city name to check weather for.
+    """
+    lat, lon = get_coordinates(location)
+    if lat is None:
+        return f"Could not find coordinates for {location}."
+    
+    weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}&current=temperature_2m,relative_humidity_2m,weather_code"
+    res = requests.get(weather_url).json()
+    
+    current = res.get("current", {})
+    temp = current.get("temperature_2m")
+    hum = current.get("relative_humidity_2m")
+    
+    return f"LlamaIndex: In {location}, it is currently {temp}°C with {hum}% humidity."
 
-li_agent = AgentWorkflow.from_tools_or_functions(
-    [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)}"
 
 # ==========================================
-# UI LOGIC
+# PART 2: SMOLAGENTS
 # ==========================================
-async def chat_llama(message, history):
-    result = await li_agent.run(user_msg=message)
-    return str(result)
+smol_model = InferenceClientModel(model_id=MODEL_ID, token=HF_TOKEN, provider="together")
+
+@tool
+def weather_tool(location: str) -> str:
+    """
+    Get the current real-time weather for a location.
+    Args:
+        location: The city and country, e.g., 'London, UK'.
+    """
+    lat, lon = get_coordinates(location)
+    if lat is None:
+        return f"Sorry, I couldn't find the location: {location}"
+    
+    weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}&current=temperature_2m,wind_speed_10m"
+    res = requests.get(weather_url).json()
+    
+    current = res.get("current", {})
+    temp = current.get("temperature_2m")
+    wind = current.get("wind_speed_10m")
+    
+    return f"smolagents: The current temperature in {location} is {temp}°C with a wind speed of {wind} km/h."
+
+smol_agent = CodeAgent(
+    model=smol_model,
+    tools=[weather_tool, DuckDuckGoSearchTool()],
+    add_base_tools=True
+)
 
 def chat_smol(message, history):
-    # Yield a 'Thinking' status using ChatMessage metadata
-    yield [
-        ChatMessage(role="user", content=message),
-        ChatMessage(
-            role="assistant", 
-            content="Agent is searching and executing Python code...",
-            metadata={"title": "🔍 Step 1: Thinking", "status": "pending"}
-        )
-    ]
     try:
         response = smol_agent.run(message)
-        yield [
-            ChatMessage(role="user", content=message),
-            ChatMessage(role="assistant", content=str(response))
-        ]
+        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 STABLE UI
+# PART 3: GRADIO 6.0 UI
 # ==========================================
-with gr.Blocks(fill_height=True) as demo:
-    # Sidebar
-    with gr.Sidebar():
-        gr.Markdown("# ⚙️ Control Panel")
-        gr.HTML("<hr style='border: 0.5px solid #e5e7eb;'>") # Manual divider
-        
-        gr.Markdown("### 🤖 Framework Status")
-        gr.Markdown("- **LlamaIndex**: Ready")
-        gr.Markdown("- **smolagents**: Ready")
-        
-        gr.HTML("<hr style='border: 0.5px solid #e5e7eb;'>")
-        gr.Info("Use LlamaIndex for structured logic and smolagents for web searches.")
-
-    # Main Content
-    with gr.Column():
-        gr.Markdown(f"# 🤖 Multi-Agent Hub")
+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.Tabs():
-            # LlamaIndex Tab
-            with gr.Tab("🏗️ LlamaIndex (Workflow)"):
-                gr.ChatInterface(
-                    fn=chat_llama,
-                    examples=["How is the weather in Tokyo?"]
-                )
-
-            # smolagents Tab
-            with gr.Tab("💻 smolagents (CodeAgent)"):
-                chatbot = gr.Chatbot(
-                    label="Code Agent", 
-                    type="messages",
-                    avatar_images=(None, "https://raw.githubusercontent.com/huggingface/smolagents/main/docs/source/imgs/smolagents_logo.png")
-                )
-                with gr.Row():
-                    msg = gr.Textbox(placeholder="Ask me to search or run code...", show_label=False, scale=4)
-                    submit = gr.Button("Send", variant="primary", scale=1)
-                
-                # Logic for custom chatbot behavior
-                submit.click(chat_smol, [msg, chatbot], [chatbot])
-                submit.click(lambda: "", None, [msg])
+        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; }"
+    )