Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import requests
 import os
-from urllib.parse import quote
 
 # =====================================================
 # APP
@@ -12,78 +11,29 @@ st.title("🌍 Biodiversity GIS Explorer")
 st.write("Explore biodiversity species using GBIF + AI analysis")
 
 # =====================================================
-# GEMINI SAFE
-# =====================================================
-
-def call_gemini(prompt, api_key):
-
-    if not api_key:
-        return "⚠️ Missing API Key"
-
-    url = (
-        "https://generativelanguage.googleapis.com/v1beta/models/"
-        "gemini-3.1-flash-lite:generateContent?key="
-        f"{api_key}"
-    )
-
-    payload = {
-        "contents": [{"parts": [{"text": prompt}]}]
-    }
-
-    try:
-        r = requests.post(url, json=payload)
-        data = r.json()
-
-        if "candidates" not in data:
-            return f"Gemini Error: {data}"
-
-        return data["candidates"][0]["content"]["parts"][0]["text"]
-
-    except Exception as e:
-        return f"Gemini Exception: {e}"
-
-
-# =====================================================
-# COUNTRY
+# COUNTRY MAP
 # =====================================================
 
 country_codes = {
     "sudan": "SD",
     "السودان": "SD",
     "egypt": "EG",
-    "مصر": "EG",
     "kenya": "KE",
-    "ethiopia": "ET",
-    "usa": "US",
-    "america": "US"
+    "ethiopia": "ET"
 }
 
 # =====================================================
-# FILTERS
+# GBIF CLEAN FETCH (NO MIXING)
 # =====================================================
 
-species_filters = {
-    "Birds": "class=Aves",
-    "Plants": "kingdom=Plantae",
-    "Mammals": "class=Mammalia",
-    "Reptiles": "class=Reptilia",
-    "Fish": "class=Actinopterygii",
-    "Insects": "class=Insecta"
-}
-
-# =====================================================
-# CLEAN GBIF (STRONG FILTER ENGINE)
-# =====================================================
-
-def get_species(country_code, filter_value, species_type):
+def get_species_grouped(country_code):
 
     url = (
         "https://api.gbif.org/v1/occurrence/search?"
         f"country={country_code}"
-        f"&{filter_value}"
         "&limit=300"
-        "&occurrence_status=PRESENT"
         "&has_coordinate=true"
+        "&occurrence_status=PRESENT"
     )
 
     try:
@@ -91,111 +41,66 @@ def get_species(country_code, filter_value, species_type):
         data = r.json()
         results = data.get("results", [])
 
-        species = []
+        grouped = {
+            "Birds": [],
+            "Reptiles": [],
+            "Mammals": [],
+            "Insects": [],
+            "Amphibians": [],
+            "Other": []
+        }
 
-        bad_words = [
-            "sp.", "cf.", "aff.", "unclassified",
-            "unknown", "incertae", "sp", "cf"
-        ]
+        bad_words = ["sp.", "cf.", "aff.", "unknown", "unclassified"]
 
         for item in results:
 
             sp = item.get("species")
-            kingdom = item.get("kingdom")
             cls = item.get("class")
-            status = item.get("establishmentMeans")
-            tax_status = item.get("taxonomicStatus")
+            kingdom = item.get("kingdom")
 
             if not sp:
                 continue
 
-            sp_low = sp.lower()
-
-            # remove junk names
-            if any(b in sp_low for b in bad_words):
+            if any(b in sp.lower() for b in bad_words):
                 continue
 
-            # must be proper scientific name
             if len(sp.split()) < 2:
                 continue
 
-            # =================================================
-            # STRONG FILTER ENGINE
-            # =================================================
-
-            if tax_status and tax_status != "ACCEPTED":
+            # avoid duplicates
+            if any(sp in v for v in grouped.values()):
                 continue
 
-            # BIRDS ONLY CLEAN
-            if species_type == "Birds":
-
-                if cls != "Aves":
-                    continue
+            # =========================
+            # CLASSIFICATION FIX
+            # =========================
 
-                if status in ["INTRODUCED", "ASSISTED COLONIZATION", "MANAGED"]:
-                    continue
+            if kingdom != "Animalia":
+                continue
 
-                # remove urban-only birds (optional but important)
-                urban = ["Columba livia", "Passer domesticus", "Cairina moschata"]
-                if sp in urban:
-                    continue
+            if cls == "Aves":
+                grouped["Birds"].append(sp)
 
-            # REPTILES ONLY CLEAN
-            elif species_type == "Reptiles":
+            elif cls == "Reptilia":
+                grouped["Reptiles"].append(sp)
 
-                if cls != "Reptilia":
-                    continue
+            elif cls == "Mammalia":
+                grouped["Mammals"].append(sp)
 
-                if kingdom != "Animalia":
-                    continue
+            elif cls == "Insecta":
+                grouped["Insects"].append(sp)
 
-            # PLANTS
-            elif species_type == "Plants":
-                if kingdom != "Plantae":
-                    continue
+            elif cls == "Amphibia":
+                grouped["Amphibians"].append(sp)
 
-            # OTHER ANIMALS
             else:
-                if kingdom != "Animalia":
-                    continue
-
-            if sp not in species:
-                species.append(sp)
+                grouped["Other"].append(sp)
 
-        return species[:10]
+        return grouped
 
     except Exception as e:
         st.error(f"GBIF Error: {e}")
-        return []
-
-
-# =====================================================
-# LINKS
-# =====================================================
-
-def get_species_links(species_name, country):
-
-    clean = " ".join(species_name.split()[:2])
-    q = quote(clean)
-    c = quote(country)
-
-    links = []
-
-    try:
-        r = requests.get(
-            f"https://api.gbif.org/v1/species/match?name={q}"
-        ).json()
-
-        if "usageKey" in r:
-            links.append(f"- [GBIF](https://www.gbif.org/species/{r['usageKey']})")
-    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})")
-
-    return links
+        return None
 
 
 # =====================================================
@@ -204,11 +109,6 @@ def get_species_links(species_name, country):
 
 country = st.text_input("Enter Country")
 
-species_type = st.selectbox(
-    "Select Species Type",
-    ["Birds", "Plants", "Mammals", "Reptiles", "Fish", "Insects"]
-)
-
 # =====================================================
 # RUN
 # =====================================================
@@ -221,53 +121,32 @@ if st.button("Search Biodiversity"):
         st.error("Country not supported")
         st.stop()
 
-    filter_value = species_filters[species_type]
+    with st.spinner("Fetching biodiversity data..."):
+        grouped = get_species_grouped(country_code)
 
-    with st.spinner("Fetching clean biodiversity data..."):
-        species_list = get_species(country_code, filter_value, species_type)
-
-    if not species_list:
-        st.warning("No clean species found")
+    if not grouped:
+        st.warning("No data found")
         st.stop()
 
-    st.subheader("🐾 Species Found")
-
-    for sp in species_list:
-        st.write("•", sp)
-
     # =================================================
-    # AI ANALYSIS
+    # DISPLAY BY GROUP
     # =================================================
 
-    prompt = f"""
-You are a biodiversity scientist.
+    for group_name, species_list in grouped.items():
 
-Country: {country}
-Group: {species_type}
+        if not species_list:
+            continue
 
-Species:
-{chr(10).join(species_list)}
+        st.subheader(f"🐾 {group_name}")
 
-Only use verified scientific knowledge. No hallucination.
-"""
-
-    st.subheader("🤖 AI Analysis")
-
-    st.write(
-        call_gemini(prompt, os.getenv("GEMINI_API_KEY"))
-    )
+        for sp in species_list[:10]:
+            st.write("•", sp)
 
     # =================================================
-    # LINKS
+    # SUMMARY
     # =================================================
 
-    st.subheader("🔗 Links")
-
-    for sp in species_list:
-
-        st.markdown(f"### 🔬 {sp}")
-
-        for link in get_species_links(sp, country):
-            st.markdown(link)
+    st.subheader("📊 Summary")
 
-        st.markdown("---")
+    for k, v in grouped.items():
+        st.write(f"{k}: {len(v)} species")