Update app.py

app.py

@@ -9,6 +9,7 @@ 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
@@ -19,10 +20,16 @@ def call_gemini(prompt, api_key):
     if not api_key:
         return "⚠️ GEMINI_API_KEY not found."
 
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-3.1-flash-lite:generateContent?key={api_key}"
+    url = (
+        "https://generativelanguage.googleapis.com/v1beta/models/"
+        "gemini-3.1-flash-lite:generateContent?key="
+        f"{api_key}"
+    )
 
     payload = {
-        "contents": [{"parts": [{"text": prompt}]}]
+        "contents": [
+            {"parts": [{"text": prompt}]}
+        ]
     }
 
     try:
@@ -34,57 +41,44 @@ def call_gemini(prompt, api_key):
 
 
 # =====================================================
-# COUNTRY (fixed)
+# COUNTRY CLEAN INPUT
 # =====================================================
 
 country_codes = {
-    "sudan": "SD",
-    "السودان": "SD",
-    "egypt": "EG",
-    "مصر": "EG",
-    "kenya": "KE",
-    "كينيا": "KE",
-    "ethiopia": "ET",
-    "إثيوبيا": "ET",
-    "saudi arabia": "SA",
-    "السعودية": "SA",
-    "usa": "US",
-    "united states": "US",
-    "america": "US",
-    "امريكا": "US",
-    "canada": "CA",
-    "كندا": "CA"
+    "sudan": "SD", "السودان": "SD",
+    "egypt": "EG", "مصر": "EG",
+    "kenya": "KE", "كينيا": "KE",
+    "ethiopia": "ET", "إثيوبيا": "ET",
+    "saudi arabia": "SA", "السعودية": "SA",
+    "usa": "US", "united states": "US", "america": "US", "امريكا": "US",
+    "canada": "CA", "كندا": "CA"
 }
 
-
 # =====================================================
-# TAXON FILTERS (FIXED - IMPORTANT)
+# TAXON FILTERS (FIXED APPROACH)
 # =====================================================
 
 species_filters = {
-    "Birds": "taxon_key=212",
-    "Plants": "taxon_key=6",
-    "Mammals": "taxon_key=359",
-    "Reptiles": "taxon_key=358",
-    "Fish": "taxon_key=204",
-    "Insects": "taxon_key=216"
+    "Birds": "kingdom=Animalia&class=Aves",
+    "Plants": "kingdom=Plantae",
+    "Mammals": "kingdom=Animalia&class=Mammalia",
+    "Reptiles": "kingdom=Animalia&class=Reptilia",
+    "Fish": "kingdom=Animalia&class=Actinopterygii",
+    "Insects": "kingdom=Animalia&class=Insecta"
 }
 
-
 # =====================================================
-# GET SPECIES (CLEAN + FIXED GBIF LOGIC)
+# GET CLEAN SPECIES (IMPORTANT FIX)
 # =====================================================
 
 def get_species(country_code, filter_value):
 
     url = (
-        "https://api.gbif.org/v1/occurrence/search?"
+        "https://api.gbif.org/v1/species/search?"
         f"country={country_code}"
         f"&{filter_value}"
-        "&limit=200"
-        "&has_coordinate=true"
-        "&has_geospatial_issue=false"
-        "&occurrence_status=PRESENT"
+        "&rank=SPECIES"
+        "&limit=50"
     )
 
     try:
@@ -96,29 +90,32 @@ def get_species(country_code, filter_value):
 
         bad_words = [
             "unclassified", "sp.", "cf.", "aff.",
-            "unknown", "incertae", "plant", "bird"
+            "unknown", "incertae"
         ]
 
         for item in results:
 
-            sp = item.get("species")
+            sp = item.get("scientificName")
             kingdom = item.get("kingdom")
+            cls = item.get("class")
 
-            # FILTER 1: must exist
             if not sp:
                 continue
 
-            # FILTER 2: remove junk names
             sp_low = sp.lower()
+
             if any(w in sp_low for w in bad_words):
                 continue
 
-            # FILTER 3: must be real species name
             if len(sp.split()) < 2:
                 continue
 
-            # FILTER 4: strict bird check
-            if species_type == "Birds" and kingdom != "Animalia":
+            # strict filtering
+            if species_type == "Birds" and cls != "Aves":
+                continue
+            if species_type == "Plants" and kingdom != "Plantae":
+                continue
+            if species_type != "Plants" and kingdom != "Animalia":
                 continue
 
             if sp not in species:
@@ -132,7 +129,7 @@ def get_species(country_code, filter_value):
 
 
 # =====================================================
-# LINKS (kept clean)
+# LINKS
 # =====================================================
 
 def get_species_links(species_name, country):
@@ -143,7 +140,6 @@ def get_species_links(species_name, country):
     q = quote(clean)
     c = quote(country)
 
-    # GBIF
     try:
         r = requests.get(
             f"https://api.gbif.org/v1/species/match?name={q}"
@@ -156,17 +152,14 @@ def get_species_links(species_name, country):
     except:
         pass
 
-    # Wikipedia
     links.append(
         f"- [Wikipedia](https://en.wikipedia.org/wiki/{clean.replace(' ','_')})"
     )
 
-    # iNaturalist
     links.append(
         f"- [iNaturalist](https://www.inaturalist.org/search?q={q})"
     )
 
-    # Scholar
     links.append(
         f"- [Google Scholar](https://scholar.google.com/scholar?q={q}+{c})"
     )
@@ -186,12 +179,12 @@ species_type = st.selectbox(
 )
 
 # =====================================================
-# MAIN
+# SEARCH
 # =====================================================
 
 if st.button("Search Biodiversity"):
 
-    # normalize country
+    # clean country input
     country_clean = country.strip().lower()
     country_code = country_codes.get(country_clean)
 
@@ -205,32 +198,36 @@ if st.button("Search Biodiversity"):
 
     filter_value = species_filters[species_type]
 
-    with st.spinner("Fetching GBIF data..."):
+    with st.spinner("Fetching clean GBIF species..."):
         species_list = get_species(country_code, filter_value)
 
     if not species_list:
         st.warning("No clean species found")
         st.stop()
 
+    # =================================================
+    # DISPLAY
+    # =================================================
+
     st.subheader("🐾 Species Found")
 
     for sp in species_list:
         st.write("•", sp)
 
     # =================================================
-    # AI
+    # GEMINI
     # =================================================
 
     prompt = f"""
 You are a biodiversity scientist.
 
 Country: {country}
-Type: {species_type}
+Group: {species_type}
 
-Species:
+Species list:
 {chr(10).join(species_list)}
 
-Only analyze real species.
+Only analyze real species. No hallucination.
 """
 
     st.subheader("🤖 AI Analysis")
@@ -246,7 +243,10 @@ Only analyze real species.
     st.subheader("🔗 Links")
 
     for sp in species_list:
-        st.markdown(f"### {sp}")
+
+        st.markdown(f"### 🔬 {sp}")
+
         for link in get_species_links(sp, country):
             st.markdown(link)
+
         st.markdown("---")