Upload 2 files

tools/translate_phrase_tool.txt

@@ -0,0 +1,39 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+import os
+import requests
+
+class TranslatePhraseTool(Tool):
+    name = "translate_phrase"
+    description = "Translates common travel phrases to a local language."
+    inputs = {
+        'text': {'type': 'string', 'description': 'Text to translate (e.g., "Hello", "Thank you", "Where is the bathroom?")'},
+        'language': {'type': 'string', 'description': 'Target language (e.g., "Spanish", "Japanese", "French")'}
+    }
+    output_type = "string"
+
+    def __init__(self, api_key=None):
+        super().__init__()
+        # You can set an API key for a real translation API
+        self.api_key = api_key or os.environ.get("TRANSLATION_API_KEY")
+        
+        # Common travel phrases in various languages (for demo/fallback purposes)
+        self.phrase_translations = {
+            "hello": {
+                "spanish": {"text": "Hola", "pronunciation": "oh-lah"},
+                "french": {"text": "Bonjour", "pronunciation": "bohn-zhoor"},
+                "italian": {"text": "Ciao", "pronunciation": "chow"},
+                "german": {"text": "Hallo", "pronunciation": "hah-loh"},
+                "japanese": {"text": "こんにちは (Konnichiwa)", "pronunciation": "kohn-nee-chee-wah"},
+                "mandarin": {"text": "你好 (Nǐ hǎo)", "pronunciation": "nee how"},
+                "arabic": {"text": "مرحبا (Marhaba)", "pronunciation": "mar-ha-ba"},
+                "russian": {"text": "Здравствуйте (Zdravstvuyte)", "pronunciation": "zdrah-stvooy-tye"},
+                "portuguese": {"text": "Olá", "pronunciation": "oh-lah"},
+                "thai": {"text": "สวัสดี (Sawatdee)", "pronunciation": "sa-wat-dee"}
+            },
+            "thank you": {
+                "spanish": {"text": "Gracias", "pronunciation": "grah-see-ahs"},
+                "french": {"text": "Merci", "pronunciation": "mair-see"},
+                "italian": {"text": "Grazie", "pronunciation": "graht-see-eh"},
+                "german": {"text": "Danke", "pronunciation": "dahn-kuh"},
+                "japanese": {"text": "ありがとう (Arigatou)", "pronunciation": "ah-

tools/visa_requirements_tool.py

@@ -0,0 +1,185 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+import os
+import requests
+
+class GetVisaRequirementsTool(Tool):
+    name = "get_visa_requirements"
+    description = "Checks visa requirements for traveling to a destination."
+    inputs = {
+        'nationality': {'type': 'string', 'description': "Traveler's passport country (e.g., 'US', 'UK', 'Canada')"},
+        'destination': {'type': 'string', 'description': "Country to visit (e.g., 'Japan', 'France', 'Brazil')"}
+    }
+    output_type = "string"
+
+    def __init__(self, api_key=None):
+        super().__init__()
+        # You can set an API key for a real visa API service
+        self.api_key = api_key or os.environ.get("VISA_API_KEY")
+        
+        # Map of common country names to their normalized forms
+        self.country_mapping = {
+            "us": "united states", "usa": "united states", "united states of america": "united states",
+            "uk": "united kingdom", "britain": "united kingdom", "great britain": "united kingdom",
+            "uae": "united arab emirates", "emirates": "united arab emirates",
+            "china": "china", "prc": "china",
+            "japan": "japan", 
+            "korea": "south korea", "south korea": "south korea", "republic of korea": "south korea",
+            "india": "india",
+            "germany": "germany",
+            "france": "france",
+            "italy": "italy",
+            "spain": "spain",
+            "mexico": "mexico",
+            "brazil": "brazil",
+            "australia": "australia", "aus": "australia",
+            "new zealand": "new zealand", "nz": "new zealand",
+            "canada": "canada", "can": "canada",
+            "russia": "russia", "russian federation": "russia",
+            "south africa": "south africa", "sa": "south africa",
+            "thailand": "thailand",
+            "vietnam": "vietnam",
+            "indonesia": "indonesia",
+            "malaysia": "malaysia",
+            "philippines": "philippines", "ph": "philippines",
+            "singapore": "singapore", "sg": "singapore",
+            "egypt": "egypt",
+            "turkey": "turkey", "türkiye": "turkey"
+        }
+        
+        # Sample visa requirement data for frequently traveled routes
+        self.visa_data = {
+            "united states": {
+                "european union": "No visa required for stays up to 90 days in the Schengen Area",
+                "united kingdom": "No visa required for stays up to 6 months",
+                "japan": "No visa required for stays up to 90 days",
+                "australia": "Electronic Travel Authority (ETA) required",
+                "china": "Visa required, must apply in advance",
+                "india": "e-Visa available, apply online before travel",
+                "brazil": "No visa required for stays up to 90 days",
+                "mexico": "No visa required for stays up to 180 days",
+                "south africa": "No visa required for stays up to 90 days",
+                "thailand": "No visa required for stays up to 30 days",
+                "canada": "No visa required for US citizens"
+            },
+            "united kingdom": {
+                "european union": "No visa required for stays up to 90 days in the Schengen Area",
+                "united states": "ESTA required for entry",
+                "japan": "No visa required for stays up to 90 days",
+                "australia": "eVisitor visa required",
+                "china": "Visa required, must apply in advance",
+                "india": "e-Visa available, apply online before travel",
+                "brazil": "No visa required for stays up to 90 days",
+                "mexico": "No visa required for stays up to 180 days",
+                "south africa": "No visa required for stays up to 90 days",
+                "thailand": "No visa required for stays up to 30 days",
+                "canada": "eTA required for British citizens"
+            },
+            "canada": {
+                "european union": "No visa required for stays up to 90 days in the Schengen Area",
+                "united states": "No visa required for Canadian citizens",
+                "japan": "No visa required for stays up to 90 days",
+                "australia": "eVisitor visa required",
+                "china": "Visa required, must apply in advance",
+                "india": "e-Visa available, apply online before travel",
+                "brazil": "No visa required for stays up to 90 days",
+                "mexico": "No visa required for stays up to 180 days",
+                "south africa": "No visa required for stays up to 90 days",
+                "thailand": "No visa required for stays up to 30 days",
+                "united kingdom": "No visa required for stays up to 6 months"
+            },
+            "japan": {
+                "european union": "No visa required for stays up to 90 days in the Schengen Area",
+                "united states": "ESTA required for entry",
+                "australia": "eVisitor visa required",
+                "china": "Visa required, must apply in advance",
+                "india": "e-Visa available, apply online before travel",
+                "brazil": "Visa required for Japanese citizens",
+                "mexico": "No visa required for stays up to 180 days",
+                "south africa": "No visa required for stays up to 90 days",
+                "thailand": "No visa required for stays up to 30 days",
+                "united kingdom": "No visa required for stays up to 6 months",
+                "canada": "eTA required for Japanese citizens"
+            },
+            "australia": {
+                "european union": "No visa required for stays up to 90 days in the Schengen Area",
+                "united states": "ESTA required for entry",
+                "japan": "No visa required for stays up to 90 days",
+                "china": "Visa required, must apply in advance",
+                "india": "e-Visa available, apply online before travel",
+                "brazil": "Visa required for Australian citizens",
+                "mexico": "No visa required for stays up to 180 days",
+                "south africa": "No visa required for stays up to 90 days",
+                "thailand": "No visa required for stays up to 30 days",
+                "united kingdom": "No visa required for stays up to 6 months",
+                "canada": "eTA required for Australian citizens"
+            }
+        }
+
+    def forward(self, nationality: str, destination: str) -> str:
+        try:
+            # Normalize inputs
+            nationality = nationality.lower().strip()
+            destination = destination.lower().strip()
+            
+            # Apply mappings if available
+            nationality = self.country_mapping.get(nationality, nationality)
+            destination = self.country_mapping.get(destination, destination)
+            
+            # Try to use a real visa API if the API key is available
+            if self.api_key:
+                try:
+                    # This would be the API call in a real implementation
+                    # For now, we'll fall back to stored data
+                    pass
+                except:
+                    # Fall back to stored data if API call fails
+                    return self._check_with_stored_data(nationality, destination)
+            
+            # If no API key is available, use the stored data
+            return self._check_with_stored_data(nationality, destination)
+        
+        except Exception as e:
+            return f"Error retrieving visa information: {str(e)}"
+    
+    def _check_with_stored_data(self, nationality: str, destination: str) -> str:
+        # Skip if same country (generally no visa needed for citizens)
+        if nationality == destination:
+            return f"As a citizen of {nationality.title()}, you generally don't need a visa to visit your own country."
+        
+        # Check if we have data for this nationality
+        if nationality not in self.visa_data:
+            # Try to search web for information if available
+            try:
+                import importlib
+                if importlib.util.find_spec("duckduckgo_search"):
+                    from duckduckgo_search import DDGS
+                    ddgs = DDGS()
+                    results = ddgs.text(f"visa requirements for {nationality} citizens traveling to {destination}")
+                    if results:
+                        return f"Based on web search, for {nationality.title()} citizens traveling to {destination.title()}: {results[0]['body']}\n\n(Note: Always verify visa requirements with the official embassy or consulate before travel.)"
+            except:
+                pass
+                
+            return f"I don't have specific visa information for citizens of {nationality.title()}. Please check with the embassy of {destination.title()} for accurate visa requirements."
+        
+        # Check if we have data for this destination
+        if destination not in self.visa_data[nationality]:
+            # Try to search web for information if available
+            try:
+                import importlib
+                if importlib.util.find_spec("duckduckgo_search"):
+                    from duckduckgo_search import DDGS
+                    ddgs = DDGS()
+                    results = ddgs.text(f"visa requirements for {nationality} citizens traveling to {destination}")
+                    if results:
+                        return f"Based on web search, for {nationality.title()} citizens traveling to {destination.title()}: {results[0]['body']}\n\n(Note: Always verify visa requirements with the official embassy or consulate before travel.)"
+            except:
+                pass
+                
+            return f"I don't have specific visa information for {nationality.title()} citizens traveling to {destination.title()}. Please check with the embassy of {destination.title()} for accurate visa requirements."
+        
+        # Get the visa requirements
+        requirements = self.visa_data[nationality][destination]
+        
+        return f"🛂 Visa requirements for {nationality.title()} citizens traveling to {destination.title()}:\n\n{requirements}\n\n(Note: Visa requirements may change. Always verify with the official embassy or consulate before travel.)"