Update app.py

app.py

@@ -18,6 +18,218 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+
+@tool
+def get_country_capital(country_name:str)-> str:
+    """A tool that fetches the country's capital 
+    Args:
+        country_name: A string representing the country name
+    """
+    countries = {
+        "Afghanistan" : "Kabul",
+        "Albania" : "Tirana",
+        "Algeria" : "Algiers",
+        "Andorra" : "Andorra la Vella",
+        "Angola" : "Luanda",
+        "Antigua and Barbuda" : "Saint John’s",
+        "Argentina" : "Buenos Aires",
+        "Armenia" : "Yerevan",
+        "Australia" : "Canberra",
+        "Austria" : "Vienna",
+        "Azerbaijan" : "Baku",
+        "The Bahamas" : "Nassau",
+        "Bahrain" : "Manama",
+        "Bangladesh" : "Dhaka",
+        "Barbados" : "Bridgetown",
+        "Belarus" : "Minsk",
+        "Belgium" : "Brussels",
+        "Belize" : "Belmopan",
+        "Benin" : "Porto-Novo",
+        "Bhutan" : "Thimphu",
+        "Bolivia" : "La Paz, Sucre",
+        "Bosnia and Herzegovina" : "Sarajevo",
+        "Botswana" : "Gaborone",
+        "Brazil" : "Brasilia",
+        "Brunei" : "Bandar Seri Begawan",
+        "Bulgaria" : "Sofia",
+        "Burkina Faso" : "Ouagadougou",
+        "Burundi" : "Bujumbura",
+        "Cambodia" : "Phnom Penh",
+        "Cameroon" : "Yaounde",
+        "Canada" : "Ottawa",
+        "Cape Verde" : "Praia",
+        "Central African Republic" : "Bangui",
+        "Chad" : "N’Djamena",
+        "Chile" : "Santiago",
+        "China" : "Beijing",
+        "Colombia" : "Bogota",
+        "Comoros" : "Moroni",
+        "Republic of the Congo": "Brazzaville",
+        "Democratic Republic of the Congo" : "Kinshasa",
+        "Costa Rica" : "San Jose",
+        "Cote d’Ivoire" : "Yamoussoukro",
+        "Croatia" : "Zagreb",
+        "Cuba" : "Havana",
+        "Cyprus" : "Nicosia",
+        "Czech Republic" : "Prague",
+        "Denmark" : "Copenhagen",
+        "Djibouti" : "Djibouti",
+        "Dominica" : "Roseau",
+        "Dominican Republic" : "Santo Domingo",
+        "East Timor" : "Dili",
+        "Ecuador" : "Quito",
+        "Egypt" : "Cairo",
+        "El Salvador" : "San Salvador",
+        "Equatorial Guinea" : "Malabo",
+        "Eritrea" : "Asmara",
+        "Estonia" : "Tallinn",
+        "Ethiopia" : "Addis Ababa",
+        "Fiji" : "Suva",
+        "Finland" : "Helsinki",
+        "France" : "Paris",
+        "Gabon" : "Libreville",
+        "The Gambia" : "Banjul",
+        "Georgia" : "Tbilisi",
+        "Germany" : "Berlin",
+        "Ghana" : "Accra",
+        "Greece" : "Athens",
+        "Grenada" : "Saint George’s",
+        "Guatemala" : "Guatemala City",
+        "Guinea" : "Conakry",
+        "Guinea-Bissau" : "Bissau",
+        "Guyana" : "Georgetown",
+        "Haiti" : "Port-au-Prince",
+        "Honduras" : "Tegucigalpa",
+        "Hungary" : "Budapest",
+        "Iceland" : "Reykjavik",
+        "India" : "New Delhi",
+        "Indonesia" : "Jakarta",
+        "Iran" : "Tehran",
+        "Iraq" : "Baghdad",
+        "Ireland" : "Dublin",
+        "Israel" : "Jerusalem",
+        "Italy" : "Rome",
+        "Jamaica" : "Kingston",
+        "Japan" : "Tokyo",
+        "Jordan" : "Amman",
+        "Kazakhstan" : "Astana",
+        "Kenya" : "Nairobi",
+        "Kiribati" : "Tarawa Atoll",
+        "North Korea" : "Pyongyang",
+        "South Korea" : "Seoul",
+        "Kosovo" : "Pristina",
+        "Kuwait" : "Kuwait City",
+        "Kyrgyzstan" : "Bishkek",
+        "Laos" : "Vientiane",
+        "Latvia" : "Riga",
+        "Lebanon" : "Beirut",
+        "Lesotho" : "Maseru",
+        "Liberia" : "Monrovia",
+        "Libya" : "Tripoli",
+        "Liechtenstein" : "Vaduz",
+        "Lithuania" : "Vilnius",
+        "Luxembourg" : "Luxembourg",
+        "Macedonia" : "Skopje",
+        "Madagascar" : "Antananarivo",
+        "Malawi" : "Lilongwe",
+        "Malaysia" : "Kuala Lumpur",
+        "Maldives" : "Male",
+        "Mali" : "Bamako",
+        "Malta" : "Valletta",
+        "Marshall Islands" : "Majuro",
+        "Mauritania" : "Nouakchott",
+        "Mauritius" : "Port Louis",
+        "Mexico" : "Mexico City",
+        "Federated States of Micronesia" : "Palikir",
+        "Moldova" : "Chisinau",
+        "Monaco" : "Monaco",
+        "Mongolia" : "Ulaanbaatar",
+        "Montenegro" : "Podgorica",
+        "Morocco" : "Rabat",
+        "Mozambique" : "Maputo",
+        "Myanmar" : "Naypyidaw",
+        "Namibia" : "Windhoek",
+        "Nauru" : "Yaren District",
+        "Nepal" : "Kathmandu",
+        "Netherlands" : "Amsterdam",
+        "New Zealand" : "Wellington",
+        "Nicaragua" : "Managua",
+        "Niger" : "Niamey",
+        "Nigeria" : "Abuja",
+        "Norway" : "Oslo",
+        "Oman" : "Muscat",
+        "Pakistan" : "Islamabad",
+        "Palau" : "Melekeok",
+        "Panama" : "Panama City",
+        "Papua New Guinea" : "Port Moresby",
+        "Paraguay" : "Asuncion",
+        "Peru" : "Lima",
+        "Philippines" : "Manila",
+        "Poland" : "Warsaw",
+        "Portugal" : "Lisbon",
+        "Qatar" : "Doha",
+        "Romania" : "Bucharest",
+        "Russia" : "Moscow",
+        "Rwanda" : "Kigali",
+        "Saint Kitts and Nevis" : "Basseterre",
+        "Saint Lucia" : "Castries",
+        "Saint Vincent and the Grenadines" : "Kingstown",
+        "Samoa" : "Apia",
+        "San Marino" : "San Marino",
+        "Sao Tome and Principe" : "Sao Tome",
+        "Saudi Arabia" : "Riyadh",
+        "Senegal" : "Dakar",
+        "Serbia" : "Belgrade",
+        "Seychelles" : "Victoria",
+        "Sierra Leone" : "Freetown",
+        "Singapore" : "Singapore",
+        "Slovakia" : "Bratislava",
+        "Slovenia" : "Ljubljana",
+        "Solomon Islands" : "Honiara",
+        "Somalia" : "Mogadishu",
+        "South Africa" : "Pretoria, Cape Town, Bloemfontein",
+        "South Sudan" : "Juba",
+        "Spain" : "Madrid",
+        "Sri Lanka" : "Colombo, Sri Jayewardenepura Kotte",
+        "Sudan" : "Khartoum",
+        "Suriname" : "Paramaribo",
+        "Swaziland" : "Mbabane",
+        "Sweden" : "Stockholm",
+        "Switzerland" : "Bern",
+        "Syria" : "Damascus",
+        "Taiwan" : "Taipei",
+        "Tajikistan" : "Dushanbe",
+        "Tanzania" : "Dodoma",
+        "Thailand" : "Bangkok",
+        "Togo" : "Lome",
+        "Tonga" : "Nuku’alofa",
+        "Trinidad and Tobago" : "Port-of-Spain",
+        "Tunisia" : "Tunis",
+        "Turkey" : "Ankara",
+        "Turkmenistan" : "Ashgabat",
+        "Tuvalu" : "Funafuti",
+        "Uganda" : "Kampala",
+        "Ukraine" : "Kyiv",
+        "United Arab Emirates" : "Abu Dhabi",
+        "United Kingdom" : "London",
+        "United States of America" : "Washington D.C.",
+        "Uruguay" : "Montevideo",
+        "Uzbekistan" : "Tashkent",
+        "Vanuatu" : "Port-Vila",
+        "Vatican City" : "Vatican City",
+        "Venezuela" : "Caracas",
+        "Vietnam" : "Hanoi",
+        "Yemen" : "Sanaa",
+        "Zambia" : "Lusaka",
+        "Zimbabwe" : "Harare"
+    }
+    
+    capital = countries.get(country_name)
+    if not capital:
+        return f"Cannot find the capital of {country_name} Try to enter the name correct"
+    return f"The Capital of {country_name} is {capital}"
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.