dummy agent

.env.example

@@ -0,0 +1,2 @@
+SPACE_ID=
+HF_TOKEN=

.gitignore

@@ -1,2 +1,3 @@
 __pycache__
-__pycache__/*
+__pycache__/*
+.env

agent.py

@@ -1,3 +1,9 @@
+import os
+from PIL import Image
+from smolagents import CodeAgent, HfApiModel, InferenceClientModel
+
+import tools.tools as tls
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
@@ -5,7 +11,18 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
+        model = HfApiModel(model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud", provider="together", use_auth_token=True)
+
+        agent = CodeAgent(
+            tools=[tls.search_tool, tls.calculate_cargo_travel_time],
+            model=InferenceClientModel(),
+            additional_authorized_imports=["pandas"],
+            max_steps=20,
+        )
+
+        fixed_answer = agent.run(question)
+
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
+        # fixed_answer = "This is a default answer."
         print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        return str(fixed_answer)

app.py

@@ -3,6 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from huggingface_hub import login
+from dotenv import load_dotenv
 
 import agent
 
@@ -160,7 +162,7 @@ with gr.Blocks() as demo:
     #     """
     # )
 
-    gr.LoginButton()
+    # gr.LoginButton()
 
     gr.ChatInterface(test_init_agent_for_chat, type="messages")
 
@@ -177,6 +179,14 @@ with gr.Blocks() as demo:
 
 
 if __name__ == "__main__":
+    load_dotenv()
+    hf_token = os.getenv("HF_TOKEN")
+    if hf_token:
+        login(hf_token)
+    else:
+        print("ℹ️  HF_TOKEN environment variable not found (running locally?).")
+
+
     print("\n" + "-"*30 + " App Starting " + "-"*30)
     # Check for SPACE_HOST and SPACE_ID at startup for information
     space_host_startup = os.getenv("SPACE_HOST")

requirements.txt

@@ -1,2 +1,11 @@
 gradio
-requests
+requests
+smolagents[litellm]
+matplotlib
+geopandas
+shapely
+kaleido
+smolagents
+typing
+duckduckgo-search
+huggingface_hub

tools/tools.py

@@ -0,0 +1,63 @@
+import math
+from typing import Optional, Tuple
+
+from smolagents import tool, DuckDuckGoSearchTool, VisitWebpageTool
+
+
+search_tool = DuckDuckGoSearchTool()
+
+@tool
+def calculate_cargo_travel_time(
+    origin_coords: Tuple[float, float],
+    destination_coords: Tuple[float, float],
+    cruising_speed_kmh: Optional[float] = 750.0,  # Average speed for cargo planes
+) -> float:
+    """
+    Calculate the travel time for a cargo plane between two points on Earth using great-circle distance.
+
+    Args:
+        origin_coords: Tuple of (latitude, longitude) for the starting point
+        destination_coords: Tuple of (latitude, longitude) for the destination
+        cruising_speed_kmh: Optional cruising speed in km/h (defaults to 750 km/h for typical cargo planes)
+
+    Returns:
+        float: The estimated travel time in hours
+
+    Example:
+        >>> # Chicago (41.8781° N, 87.6298° W) to Sydney (33.8688° S, 151.2093° E)
+        >>> result = calculate_cargo_travel_time((41.8781, -87.6298), (-33.8688, 151.2093))
+    """
+
+    def to_radians(degrees: float) -> float:
+        return degrees * (math.pi / 180)
+
+    # Extract coordinates
+    lat1, lon1 = map(to_radians, origin_coords)
+    lat2, lon2 = map(to_radians, destination_coords)
+
+    # Earth's radius in kilometers
+    EARTH_RADIUS_KM = 6371.0
+
+    # Calculate great-circle distance using the haversine formula
+    dlon = lon2 - lon1
+    dlat = lat2 - lat1
+
+    a = (
+        math.sin(dlat / 2) ** 2
+        + math.cos(lat1) * math.cos(lat2) * math.sin(dlon / 2) ** 2
+    )
+    c = 2 * math.asin(math.sqrt(a))
+    distance = EARTH_RADIUS_KM * c
+
+    # Add 10% to account for non-direct routes and air traffic controls
+    actual_distance = distance * 1.1
+
+    # Calculate flight time
+    # Add 1 hour for takeoff and landing procedures
+    flight_time = (actual_distance / cruising_speed_kmh) + 1.0
+
+    # Format the results
+    return round(flight_time, 2)
+
+
+print(calculate_cargo_travel_time((41.8781, -87.6298), (-33.8688, 151.2093)))