Update app.py

app.py

@@ -56,29 +56,30 @@ def call_gemini(prompt, api_key):
 
 country_codes = {
 
-    "Sudan": "SD",
+    "sudan": "SD",
     "السودان": "SD",
 
-    "Egypt": "EG",
+    "egypt": "EG",
     "مصر": "EG",
 
-    "Kenya": "KE",
+    "kenya": "KE",
     "كينيا": "KE",
 
-    "Ethiopia": "ET",
+    "ethiopia": "ET",
     "إثيوبيا": "ET",
 
-    "Saudi Arabia": "SA",
+    "saudi arabia": "SA",
     "السعودية": "SA",
 
-    "India": "IN",
+    "india": "IN",
     "الهند": "IN",
 
-    "USA": "US",
-    "United States": "US",
+    "usa": "US",
+    "united states": "US",
+    "america": "US",
     "امريكا": "US",
 
-    "Canada": "CA",
+    "canada": "CA",
     "كندا": "CA"
 }
 
@@ -188,7 +189,10 @@ def get_species_links(species_name, country):
     # GBIF
     # -------------------------------------------------
 
-    gbif_api = f"https://api.gbif.org/v1/species/match?name={encoded_species}"
+    gbif_api = (
+        f"https://api.gbif.org/v1/species/match?"
+        f"name={encoded_species}"
+    )
 
     try:
 
@@ -196,7 +200,10 @@ def get_species_links(species_name, country):
 
         if "usageKey" in r:
 
-            gbif_link = f"https://www.gbif.org/species/{r['usageKey']}"
+            gbif_link = (
+                f"https://www.gbif.org/species/"
+                f"{r['usageKey']}"
+            )
 
             links.append(
                 f"- [GBIF Species Page]({gbif_link})"
@@ -211,7 +218,9 @@ def get_species_links(species_name, country):
 
     wiki_name = clean_name.replace(" ", "_")
 
-    wiki_url = f"https://en.wikipedia.org/wiki/{wiki_name}"
+    wiki_url = (
+        f"https://en.wikipedia.org/wiki/{wiki_name}"
+    )
 
     links.append(
         f"- [Wikipedia Article]({wiki_url})"
@@ -247,7 +256,10 @@ def get_species_links(species_name, country):
     # Encyclopedia of Life
     # -------------------------------------------------
 
-    eol = f"https://eol.org/search?q={encoded_species}"
+    eol = (
+        f"https://eol.org/search?q="
+        f"{encoded_species}"
+    )
 
     links.append(
         f"- [Encyclopedia of Life]({eol})"
@@ -306,7 +318,7 @@ def get_species_links(species_name, country):
     )
 
     # -------------------------------------------------
-    # Conservation Information
+    # Conservation
     # -------------------------------------------------
 
     conservation = (
@@ -347,21 +359,33 @@ species_type = st.selectbox(
 
 if st.button("Search Biodiversity"):
 
+    # -------------------------------------------------
+    # CLEAN COUNTRY INPUT
+    # -------------------------------------------------
+
+    country_clean = country.strip().lower()
+
+    country_code = country_codes.get(
+        country_clean
+    )
+
     # -------------------------------------------------
     # VALIDATION
     # -------------------------------------------------
 
     if not country:
 
-        st.warning("Please enter a country.")
+        st.warning(
+            "Please enter a country."
+        )
 
         st.stop()
 
-    country_code = country_codes.get(country)
-
     if not country_code:
 
-        st.error("Country not supported yet.")
+        st.error(
+            f"Country '{country}' not supported yet."
+        )
 
         st.stop()