add location tool

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.mmdb filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,4 +1,6 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+import geoip2.database
+import requests
 import datetime
 import pytz
 import yaml
@@ -6,6 +8,21 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+@tool
+def show_internet_location()-> str:
+    """A tool that shows user's current location based on IP address.
+    """
+    try:
+        response = requests.get("https://api64.ipify.org?format=json")
+        public_ip = response.json()["ip"]
+
+        reader = geoip2.database.Reader('geo_cities/GeoLite2-City.mmdb')
+        response = reader.city(public_ip)
+
+        internet_location = f"Based on your IP address {public_ip}, you are in {response.country.name}, {response.city.name}"
+        return internet_location
+    except Exception as e:
+        return f"Error fetching the response for query internet location: {str(e)}"
 
 @tool
 def search(query: str)-> str:
@@ -57,7 +74,7 @@ with open("prompts.yaml", 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,search], ## add your tools here (don't remove final answer)
+    tools=[final_answer,search,show_internet_location], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

prompts.yaml

@@ -37,6 +37,16 @@
   result = 5 + 3 + 1294.678
   final_answer(result)
   ```<end_code>
+  
+  ---
+  Task: "What is my current location?"
+
+  Thought: I will use python code to retrieve user's current location based on IP address and then return the final answer using the `final_answer` tool
+  Code:
+  ```py
+  result = show_internet_location()
+  final_answer(result)
+  ```<end_code>
 
   ---
   Task:

requirements.txt

@@ -3,3 +3,4 @@ smolagents
 requests
 duckduckgo_search
 pandas
+geoip2