Update app.py

app.py

@@ -8,10 +8,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")
+st.write("Search biodiversity by country and species type")
 
 # =====================================================
-# COUNTRY
+# COUNTRY MAP
 # =====================================================
 
 country_codes = {
@@ -19,11 +19,25 @@ country_codes = {
     "السودان": "SD",
     "egypt": "EG",
     "kenya": "KE",
-    "ethiopia": "ET"
+    "ethiopia": "ET",
+    "usa": "US"
 }
 
 # =====================================================
-# LINKS FUNCTION
+# TYPE FILTER
+# =====================================================
+
+species_filters = {
+    "Birds": "class=Aves",
+    "Reptiles": "class=Reptilia",
+    "Mammals": "class=Mammalia",
+    "Insects": "class=Insecta",
+    "Amphibians": "class=Amphibia",
+    "Fish": "class=Actinopterygii"
+}
+
+# =====================================================
+# LINKS
 # =====================================================
 
 def get_links(species, country):
@@ -31,26 +45,24 @@ def get_links(species, country):
     q = quote(species)
     c = quote(country)
 
-    links = [
+    return [
         f"🔗 [GBIF](https://www.gbif.org/species/match?name={q})",
         f"📘 [Wikipedia](https://en.wikipedia.org/wiki/{species.replace(' ','_')})",
         f"📸 [iNaturalist](https://www.inaturalist.org/search?q={q})",
         f"📚 [Google Scholar](https://scholar.google.com/scholar?q={q}+{c})"
     ]
 
-    return links
-
-
 # =====================================================
-# GBIF GROUPED DATA
+# FETCH FUNCTION (FILTERED)
 # =====================================================
 
-def get_species_grouped(country_code):
+def get_species(country_code, filter_value):
 
     url = (
         "https://api.gbif.org/v1/occurrence/search?"
         f"country={country_code}"
-        "&limit=300"
+        f"&{filter_value}"
+        "&limit=50"
         "&has_coordinate=true"
         "&occurrence_status=PRESENT"
     )
@@ -60,14 +72,7 @@ def get_species_grouped(country_code):
         data = r.json()
         results = data.get("results", [])
 
-        grouped = {
-            "Birds": [],
-            "Reptiles": [],
-            "Mammals": [],
-            "Insects": [],
-            "Amphibians": [],
-            "Other": []
-        }
+        species = []
 
         bad_words = ["sp.", "cf.", "aff.", "unknown", "unclassified"]
 
@@ -89,34 +94,28 @@ def get_species_grouped(country_code):
             if kingdom != "Animalia":
                 continue
 
-            # =========================
-            # CLASSIFY
-            # =========================
-
-            if cls == "Aves":
-                grouped["Birds"].append(sp)
-
-            elif cls == "Reptilia":
-                grouped["Reptiles"].append(sp)
-
-            elif cls == "Mammalia":
-                grouped["Mammals"].append(sp)
-
-            elif cls == "Insecta":
-                grouped["Insects"].append(sp)
-
-            elif cls == "Amphibia":
-                grouped["Amphibians"].append(sp)
+            # strict match with user selection
+            if filter_value == "class=Aves" and cls != "Aves":
+                continue
+            if filter_value == "class=Reptilia" and cls != "Reptilia":
+                continue
+            if filter_value == "class=Mammalia" and cls != "Mammalia":
+                continue
+            if filter_value == "class=Insecta" and cls != "Insecta":
+                continue
+            if filter_value == "class=Amphibia" and cls != "Amphibia":
+                continue
+            if filter_value == "class=Actinopterygii" and cls != "Actinopterygii":
+                continue
 
-            else:
-                grouped["Other"].append(sp)
+            if sp not in species:
+                species.append(sp)
 
-        return grouped
+        return species[:10]
 
     except Exception as e:
         st.error(f"GBIF Error: {e}")
-        return None
-
+        return []
 
 # =====================================================
 # UI
@@ -124,6 +123,15 @@ def get_species_grouped(country_code):
 
 country = st.text_input("Enter Country")
 
+species_type = st.selectbox(
+    "Select Species Type",
+    ["Birds", "Reptiles", "Mammals", "Insects", "Amphibians", "Fish"]
+)
+
+# =====================================================
+# RUN
+# =====================================================
+
 if st.button("Search Biodiversity"):
 
     country_code = country_codes.get(country.strip().lower())
@@ -132,39 +140,21 @@ if st.button("Search Biodiversity"):
         st.error("Country not supported")
         st.stop()
 
-    grouped = get_species_grouped(country_code)
+    filter_value = species_filters[species_type]
 
-    if not grouped:
-        st.warning("No data found")
-        st.stop()
+    species_list = get_species(country_code, filter_value)
 
-    # =================================================
-    # DISPLAY
-    # =================================================
-
-    for group, species_list in grouped.items():
-
-        if not species_list:
-            continue
-
-        st.subheader(f"🐾 {group}")
-
-        for sp in species_list[:10]:
-
-            st.markdown(f"### 🔬 {sp}")
-
-            links = get_links(sp, country)
+    if not species_list:
+        st.warning("No species found for this selection")
+        st.stop()
 
-            for l in links:
-                st.markdown(l)
+    st.subheader(f"🐾 {species_type} in {country}")
 
-            st.markdown("---")
+    for sp in species_list:
 
-    # =================================================
-    # SUMMARY
-    # =================================================
+        st.markdown(f"### 🔬 {sp}")
 
-    st.subheader("📊 Summary")
+        for link in get_links(sp, country):
+            st.markdown(link)
 
-    for k, v in grouped.items():
-        st.write(f"{k}: {len(v)} species")
+        st.markdown("---")