Update app.py

app.py

@@ -3,297 +3,89 @@ import gradio as gr
 import datetime
 import pytz
 import asyncio
-import requests
 
 # Framework 1: LlamaIndex
-from llama_index.core.agent import ReActAgent
+from llama_index.core.agent.workflow import AgentWorkflow
 from llama_index.core.tools import FunctionTool
-from llama_index.llms.huggingface import HuggingFaceInferenceAPI
+from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 
 # Framework 2: smolagents
-from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
 
 # 0. SHARED CONFIG
 HF_TOKEN = os.getenv("HF_TOKEN")
+# 7B is the sweet spot for free serverless inference in 2026
 MODEL_ID = "Qwen/Qwen2.5-7B-Instruct"
 
 # ==========================================
-# WEATHER FUNCTION WITH REAL API
+# PART 1: LLAMAINDEX AGENT
 # ==========================================
-def get_weather(location: str) -> str:
-    """Get weather for a location using OpenWeatherMap API"""
-    try:
-        # You can get a free API key from https://openweathermap.org/api
-        api_key = os.getenv("WEATHER_API_KEY", "")
-        
-        if api_key:
-            url = f"http://api.openweathermap.org/data/2.5/weather?q={location}&appid={api_key}&units=metric"
-            response = requests.get(url, timeout=5)
-            data = response.json()
-            
-            if response.status_code == 200:
-                temp = data['main']['temp']
-                desc = data['weather'][0]['description']
-                humidity = data['main']['humidity']
-                return f"🌤️ Weather in {location}: {desc}, {temp}°C, Humidity: {humidity}%"
-            else:
-                # Fallback to mock data if API fails
-                return f"🌤️ Weather in {location}: Sunny, 22°C (mock data - add real API key for live weather)"
-        else:
-            # Mock weather if no API key
-            weather_data = {
-                "tokyo": "Cloudy, 18°C",
-                "london": "Rainy, 12°C",
-                "new york": "Sunny, 15°C",
-                "paris": "Partly Cloudy, 14°C",
-                "sydney": "Clear, 22°C"
-            }
-            for city, weather in weather_data.items():
-                if city in location.lower():
-                    return f"🌤️ Weather in {location}: {weather} (mock data)"
-            return f"🌤️ Weather in {location}: Sunny, 22°C (mock data - set WEATHER_API_KEY for real data)"
-    except Exception as e:
-        return f"🌤️ Weather in {location}: Sunny, 22°C (error: {str(e)[:50]})"
+li_llm = HuggingFaceInferenceAPI(
+    model_name=MODEL_ID,
+    token=HF_TOKEN,
+    provider="together" 
+)
 
-# ==========================================
-# TIME FUNCTIONS
-# ==========================================
 def get_tokyo_time() -> str:
     """Returns the current time in Tokyo, Japan."""
     tz = pytz.timezone('Asia/Tokyo')
-    now = datetime.datetime.now(tz)
-    return f"🕐 Tokyo Time: {now.strftime('%A, %B %d, %Y %H:%M:%S')}"
+    return f"The current time in Tokyo is {datetime.datetime.now(tz).strftime('%H:%M:%S')}"
 
-def get_world_time(city: str = "all") -> str:
-    """Get current time for various cities."""
-    timezones = {
-        "Tokyo": "Asia/Tokyo",
-        "New York": "America/New_York",
-        "London": "Europe/London",
-        "Paris": "Europe/Paris",
-        "Sydney": "Australia/Sydney",
-        "Dubai": "Asia/Dubai",
-        "Singapore": "Asia/Singapore"
-    }
-    
-    if city.lower() != "all" and city.title() in timezones:
-        tz = pytz.timezone(timezones[city.title()])
-        now = datetime.datetime.now(tz)
-        return f"🕐 {city} Time: {now.strftime('%H:%M:%S')}"
-    
-    result = "🌍 World Clock:\n"
-    for city_name, tz_str in timezones.items():
-        tz = pytz.timezone(tz_str)
-        now = datetime.datetime.now(tz)
-        result += f"• {city_name}: {now.strftime('%H:%M:%S')}\n"
-    return result
+li_tools = [FunctionTool.from_defaults(fn=get_tokyo_time)]
 
-# ==========================================
-# CALCULATOR FUNCTION
-# ==========================================
-def calculate(expression: str) -> str:
-    """Calculate mathematical expressions."""
+li_agent = AgentWorkflow.from_tools_or_functions(
+    li_tools,
+    llm=li_llm,
+)
+
+async def chat_llama(message, history):
     try:
-        # Safe evaluation with basic math functions
-        import math
-        
-        # Create a safe environment
-        safe_dict = {
-            'abs': abs, 'round': round, 'min': min, 'max': max,
-            'sum': sum, 'len': len, 'int': int, 'float': float,
-            'math': math, 'sqrt': math.sqrt, 'pow': math.pow,
-            'sin': math.sin, 'cos': math.cos, 'tan': math.tan,
-            'pi': math.pi, 'e': math.e
-        }
-        
-        # Remove any dangerous characters
-        expression = ''.join(c for c in expression if c.isalnum() or c in ' +-*/().,')
-        
-        result = eval(expression, {"__builtins__": {}}, safe_dict)
-        return f"🧮 {expression} = {result}"
+        result = await li_agent.run(user_msg=message)
+        return str(result)
     except Exception as e:
-        return f"❌ Calculation error: {str(e)[:100]}"
-
-# ==========================================
-# PART 1: LLAMAINDEX AGENT (FIXED)
-# ==========================================
-try:
-    li_llm = HuggingFaceInferenceAPI(
-        model_name=MODEL_ID,
-        token=HF_TOKEN,
-    )
-    
-    # Create tools for LlamaIndex
-    li_tools = [
-        FunctionTool.from_defaults(fn=get_tokyo_time, name="get_tokyo_time"),
-        FunctionTool.from_defaults(fn=get_weather, name="get_weather"),
-        FunctionTool.from_defaults(fn=get_world_time, name="get_world_time"),
-        FunctionTool.from_defaults(fn=calculate, name="calculate")
-    ]
-    
-    li_agent = ReActAgent.from_tools(
-        li_tools,
-        llm=li_llm,
-        verbose=False
-    )
-    
-    async def chat_llama(message, history):
-        try:
-            response = await li_agent.achat(message)
-            return str(response)
-        except Exception as e:
-            return f"❌ LlamaIndex Error: {str(e)[:200]}"
-            
-except Exception as e:
-    print(f"LlamaIndex setup error: {e}")
-    async def chat_llama(message, history):
-        return f"❌ LlamaIndex not available: {str(e)[:200]}"
+        return f"LlamaIndex Error: {str(e)}"
 
 # ==========================================
 # PART 2: SMOLAGENTS
 # ==========================================
-try:
-    from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
-    
-    @tool
-    def weather_tool(location: str) -> str:
-        """Get the current weather for a location."""
-        return get_weather(location)
-    
-    @tool
-    def time_tool(city: str = "all") -> str:
-        """Get current time for a city or all major cities."""
-        return get_world_time(city)
-    
-    @tool
-    def calculator_tool(expression: str) -> str:
-        """Calculate mathematical expressions."""
-        return calculate(expression)
-    
-    smol_agent = CodeAgent(
-        tools=[weather_tool, time_tool, calculator_tool, DuckDuckGoSearchTool()],
-        model=MODEL_ID,
-        max_steps=5
-    )
-    
-    def chat_smol(message, history):
-        try:
-            response = smol_agent.run(message)
-            return str(response)
-        except Exception as e:
-            return f"❌ Smolagents Error: {str(e)[:200]}"
-            
-except Exception as e:
-    print(f"Smolagents setup error: {e}")
-    def chat_smol(message, history):
-        return f"❌ Smolagents not available: {str(e)[:200]}"
+smol_model = InferenceClientModel(
+    model_id=MODEL_ID,
+    token=HF_TOKEN,
+    provider="together"
+)
+
+@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 currently sunny and 22°C."
+
+smol_agent = CodeAgent(
+    model=smol_model,
+    tools=[weather_tool, DuckDuckGoSearchTool()]
+)
+
+def chat_smol(message, history):
+    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(), title="🤖 AI Agent Studio") as demo:
-    # Header
-    gr.Markdown("# 🤖 AI Agent Studio")
-    gr.Markdown(f"**Model:** {MODEL_ID} | **Provider:** Together AI")
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🤖 Consolidated AI Agent Space")
+    gr.Markdown(f"Currently using **{MODEL_ID}** via Together AI Provider.")
     
-    # Quick Tools Section
-    with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown("### 🛠️ Quick Tools")
-            
-            # Weather Tool
-            with gr.Group():
-                weather_city = gr.Textbox(
-                    label="Check Weather",
-                    placeholder="Enter city name...",
-                    scale=4
-                )
-                weather_btn = gr.Button("🌤️ Get Weather", scale=1)
-                weather_output = gr.Textbox(label="Result", interactive=False)
-                
-                def get_weather_quick(city):
-                    return get_weather(city) if city else "Please enter a city name."
-                
-                weather_btn.click(
-                    get_weather_quick,
-                    inputs=weather_city,
-                    outputs=weather_output
-                )
-            
-            # Calculator Tool
-            with gr.Group():
-                calc_expr = gr.Textbox(
-                    label="Calculator",
-                    placeholder="Enter math expression...",
-                    scale=4
-                )
-                calc_btn = gr.Button("🧮 Calculate", scale=1)
-                calc_output = gr.Textbox(label="Result", interactive=False)
-                
-                calc_btn.click(
-                    calculate,
-                    inputs=calc_expr,
-                    outputs=calc_output
-                )
-            
-            # Time Tool
-            with gr.Group():
-                time_btn = gr.Button("🌍 Show World Clock")
-                time_output = gr.Textbox(label="World Times", interactive=False)
-                
-                time_btn.click(
-                    lambda: get_world_time("all"),
-                    outputs=time_output
-                )
-    
-    # Chat Agents Section
-    with gr.Row():
-        with gr.Tabs():
-            with gr.TabItem("🦙 LlamaIndex Agent"):
-                llama_chat = gr.ChatInterface(
-                    fn=chat_llama,
-                    title="LlamaIndex ReAct Agent",
-                    description="Structured agent with reasoning capabilities",
-                    examples=[
-                        "What's the weather in Tokyo?",
-                        "What time is it in New York?",
-                        "Calculate 25 * 4 + 18",
-                        "Tell me the time in London and Sydney"
-                    ]
-                )
-            
-            with gr.TabItem("🔬 Smolagents"):
-                smol_chat = gr.ChatInterface(
-                    fn=chat_smol,
-                    title="Smolagents CodeAgent",
-                    description="Code-based agent with web search",
-                    examples=[
-                        "Search for AI news",
-                        "What's the weather like today?",
-                        "Calculate 15% of 200",
-                        "Tell me about machine learning"
-                    ]
-                )
-    
-    # Footer
-    gr.Markdown("---")
-    gr.Markdown("""
-    ### 💡 Try asking:
-    - "What's the weather in London and what time is it there?"
-    - "Calculate the area of a circle with radius 5"
-    - "Search for latest technology news"
-    - "Compare time between Tokyo and New York"
-    """)
-    
-    # API Key Notice
-    if not os.getenv("WEATHER_API_KEY"):
-        gr.Markdown("""
-        ⚠️ **Note:** Using mock weather data. For real weather, add `WEATHER_API_KEY` to your Hugging Face Space secrets.
-        """)
+    with gr.Tab("LlamaIndex (Workflow)"):
+        gr.ChatInterface(fn=chat_llama)
+        
+    with gr.Tab("smolagents (CodeAgent)"):
+        gr.ChatInterface(fn=chat_smol)
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+    demo.launch()