Update app.py

app.py

@@ -9,16 +9,15 @@ 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
+# GEMINI (FIXED SAFE PARSING)
 # =====================================================
 
 def call_gemini(prompt, api_key):
 
     if not api_key:
-        return "⚠️ GEMINI_API_KEY not found."
+        return "⚠️ Missing API Key"
 
     url = (
         "https://generativelanguage.googleapis.com/v1beta/models/"
@@ -27,21 +26,25 @@ def call_gemini(prompt, api_key):
     )
 
     payload = {
-        "contents": [
-            {"parts": [{"text": prompt}]}
-        ]
+        "contents": [{"parts": [{"text": prompt}]}]
     }
 
     try:
         r = requests.post(url, json=payload)
         data = r.json()
+
+        # SAFE CHECK
+        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 Error: {e}"
+        return f"Gemini Exception: {e}"
 
 
 # =====================================================
-# COUNTRY NORMALIZATION
+# COUNTRY
 # =====================================================
 
 country_codes = {
@@ -50,22 +53,14 @@ country_codes = {
     "egypt": "EG",
     "مصر": "EG",
     "kenya": "KE",
-    "كينيا": "KE",
     "ethiopia": "ET",
-    "إثيوبيا": "ET",
-    "saudi arabia": "SA",
-    "السعودية": "SA",
     "usa": "US",
-    "united states": "US",
-    "america": "US",
-    "امريكا": "US",
-    "canada": "CA",
-    "كندا": "CA"
+    "america": "US"
 }
 
 
 # =====================================================
-# FILTERS (FIXED)
+# FIXED FILTERS (IMPORTANT)
 # =====================================================
 
 species_filters = {
@@ -79,17 +74,16 @@ species_filters = {
 
 
 # =====================================================
-# GET SPECIES (WORKING VERSION)
+# CLEAN GBIF (STRICT FILTERING)
 # =====================================================
 
-def get_species(country_code, filter_value):
+def get_species(country_code, filter_value, species_type):
 
     url = (
         "https://api.gbif.org/v1/occurrence/search?"
         f"country={country_code}"
         f"&{filter_value}"
         "&limit=200"
-        "&has_coordinate=true"
         "&occurrence_status=PRESENT"
     )
 
@@ -100,28 +94,35 @@ def get_species(country_code, filter_value):
 
         species = []
 
-        bad_words = [
-            "unclassified", "sp.", "cf.", "aff.",
-            "unknown", "incertae", "cf", "sp"
-        ]
+        bad = ["sp.", "cf.", "aff.", "unclassified", "unknown"]
 
         for item in results:
 
             sp = item.get("species")
+            kingdom = item.get("kingdom")
+            cls = item.get("class")
 
             if not sp:
                 continue
 
             sp_low = sp.lower()
 
-            # clean junk names
-            if any(w in sp_low for w in bad_words):
+            if any(b in sp_low for b in bad):
                 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
+
+            if species_type == "Plants" and kingdom != "Plantae":
+                continue
+
+            if species_type != "Plants" and kingdom != "Animalia":
+                continue
+
             if sp not in species:
                 species.append(sp)
 
@@ -138,13 +139,12 @@ def get_species(country_code, filter_value):
 
 def get_species_links(species_name, country):
 
-    links = []
-
     clean = " ".join(species_name.split()[:2])
     q = quote(clean)
     c = quote(country)
 
-    # GBIF
+    links = []
+
     try:
         r = requests.get(
             f"https://api.gbif.org/v1/species/match?name={q}"
@@ -157,20 +157,9 @@ 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})"
-    )
-
-    # Google Scholar
-    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
 
@@ -187,43 +176,34 @@ species_type = st.selectbox(
 )
 
 # =====================================================
-# MAIN BUTTON
+# RUN
 # =====================================================
 
 if st.button("Search Biodiversity"):
 
-    # normalize input
     country_clean = country.strip().lower()
     country_code = country_codes.get(country_clean)
 
-    if not country:
-        st.warning("Please enter a country")
-        st.stop()
-
     if not country_code:
-        st.error(f"Country '{country}' not supported yet")
+        st.error("Country not supported")
         st.stop()
 
     filter_value = species_filters[species_type]
 
-    with st.spinner("Fetching biodiversity data..."):
-        species_list = get_species(country_code, filter_value)
+    with st.spinner("Fetching clean biodiversity data..."):
+        species_list = get_species(country_code, filter_value, species_type)
 
     if not species_list:
-        st.warning("No species found (try another filter or country)")
+        st.warning("No clean species found")
         st.stop()
 
-    # =================================================
-    # DISPLAY
-    # =================================================
-
     st.subheader("🐾 Species Found")
 
     for sp in species_list:
         st.write("•", sp)
 
     # =================================================
-    # AI
+    # AI FIXED
     # =================================================
 
     prompt = f"""
@@ -235,14 +215,12 @@ Group: {species_type}
 Species:
 {chr(10).join(species_list)}
 
-Give ecological and GIS analysis only for real species.
+Only use real species. No 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
@@ -251,10 +229,7 @@ Give ecological and GIS analysis only for real species.
     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("---")