Update app.py

app.py

@@ -9,9 +9,10 @@ from urllib.parse import quote
 
 st.set_page_config(page_title="Biodiversity GIS Explorer", layout="wide")
 st.title("🌍 Biodiversity GIS Explorer")
+st.write("Explore biodiversity species using GBIF + AI analysis")
 
 # =====================================================
-# GEMINI (FIXED SAFE PARSING)
+# GEMINI (SAFE)
 # =====================================================
 
 def call_gemini(prompt, api_key):
@@ -33,7 +34,6 @@ def call_gemini(prompt, api_key):
         r = requests.post(url, json=payload)
         data = r.json()
 
-        # SAFE CHECK
         if "candidates" not in data:
             return f"Gemini Error: {data}"
 
@@ -55,12 +55,12 @@ country_codes = {
     "kenya": "KE",
     "ethiopia": "ET",
     "usa": "US",
-    "america": "US"
+    "america": "US",
+    "canada": "CA"
 }
 
-
 # =====================================================
-# FIXED FILTERS (IMPORTANT)
+# FILTERS
 # =====================================================
 
 species_filters = {
@@ -72,9 +72,8 @@ species_filters = {
     "Insects": "class=Insecta"
 }
 
-
 # =====================================================
-# CLEAN GBIF (STRICT FILTERING)
+# GET SPECIES (STRICT + CLEAN + NATIVE FILTER)
 # =====================================================
 
 def get_species(country_code, filter_value, species_type):
@@ -84,6 +83,7 @@ def get_species(country_code, filter_value, species_type):
         f"country={country_code}"
         f"&{filter_value}"
         "&limit=200"
+        "&has_coordinate=true"
         "&occurrence_status=PRESENT"
     )
 
@@ -94,34 +94,47 @@ def get_species(country_code, filter_value, species_type):
 
         species = []
 
-        bad = ["sp.", "cf.", "aff.", "unclassified", "unknown"]
+        bad_words = [
+            "unclassified", "sp.", "cf.", "aff.",
+            "unknown", "incertae", "sp", "cf"
+        ]
 
         for item in results:
 
             sp = item.get("species")
             kingdom = item.get("kingdom")
             cls = item.get("class")
+            status = item.get("establishmentMeans")
 
             if not sp:
                 continue
 
             sp_low = sp.lower()
 
-            if any(b in sp_low for b in bad):
+            # remove junk names
+            if any(b in sp_low for b in bad_words):
                 continue
 
+            # must be binomial name
             if len(sp.split()) < 2:
                 continue
 
-            # STRICT FILTER (FIX FOR YOUR PROBLEM)
-            if species_type == "Birds" and cls != "Aves":
-                continue
+            # STRICT TAXON FILTER
+            if species_type == "Birds":
+                if cls != "Aves":
+                    continue
 
-            if species_type == "Plants" and kingdom != "Plantae":
-                continue
+                # remove invasive / introduced species
+                if status in ["INTRODUCED", "ASSISTED COLONIZATION"]:
+                    continue
 
-            if species_type != "Plants" and kingdom != "Animalia":
-                continue
+            elif species_type == "Plants":
+                if kingdom != "Plantae":
+                    continue
+
+            else:
+                if kingdom != "Animalia":
+                    continue
 
             if sp not in species:
                 species.append(sp)
@@ -157,9 +170,17 @@ def get_species_links(species_name, country):
     except:
         pass
 
-    links.append(f"- [Wikipedia](https://en.wikipedia.org/wiki/{clean.replace(' ','_')})")
-    links.append(f"- [iNaturalist](https://www.inaturalist.org/search?q={q})")
-    links.append(f"- [Google Scholar](https://scholar.google.com/scholar?q={q}+{c})")
+    links.append(
+        f"- [Wikipedia](https://en.wikipedia.org/wiki/{clean.replace(' ','_')})"
+    )
+
+    links.append(
+        f"- [iNaturalist](https://www.inaturalist.org/search?q={q})"
+    )
+
+    links.append(
+        f"- [Google Scholar](https://scholar.google.com/scholar?q={q}+{c})"
+    )
 
     return links
 
@@ -184,6 +205,10 @@ if st.button("Search Biodiversity"):
     country_clean = country.strip().lower()
     country_code = country_codes.get(country_clean)
 
+    if not country:
+        st.warning("Enter country")
+        st.stop()
+
     if not country_code:
         st.error("Country not supported")
         st.stop()
@@ -203,7 +228,7 @@ if st.button("Search Biodiversity"):
         st.write("•", sp)
 
     # =================================================
-    # AI FIXED
+    # AI
     # =================================================
 
     prompt = f"""
@@ -215,12 +240,14 @@ Group: {species_type}
 Species:
 {chr(10).join(species_list)}
 
-Only use real species. No hallucination.
+Analyze only real native biodiversity. Avoid hallucination.
 """
 
     st.subheader("🤖 AI Analysis")
 
-    st.write(call_gemini(prompt, os.getenv("GEMINI_API_KEY")))
+    st.write(
+        call_gemini(prompt, os.getenv("GEMINI_API_KEY"))
+    )
 
     # =================================================
     # LINKS
@@ -229,7 +256,10 @@ Only use real species. No hallucination.
     st.subheader("🔗 Links")
 
     for sp in species_list:
+
         st.markdown(f"### 🔬 {sp}")
+
         for link in get_species_links(sp, country):
             st.markdown(link)
+
         st.markdown("---")