Update app.py

app.py

@@ -1,12 +1,17 @@
 import streamlit as st
 import requests
 import os
+from urllib.parse import quote
 
-# =========================================
+# =====================================================
 # GEMINI API
-# =========================================
+# =====================================================
 
 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}"
 
     payload = {
@@ -17,108 +22,372 @@ def call_gemini(prompt, api_key):
         ]
     }
 
-    response = requests.post(url, json=payload)
-    result = response.json()
-
     try:
+
+        response = requests.post(url, json=payload)
+
+        result = response.json()
+
         return result["candidates"][0]["content"]["parts"][0]["text"]
+
+    except Exception as e:
+
+        return f"Gemini Error: {e}"
+
+
+# =====================================================
+# COUNTRY CODES
+# =====================================================
+
+country_codes = {
+    "Sudan": "SD",
+    "Egypt": "EG",
+    "Kenya": "KE",
+    "Ethiopia": "ET",
+    "Saudi Arabia": "SA",
+    "India": "IN",
+    "USA": "US",
+    "Canada": "CA"
+}
+
+
+# =====================================================
+# SPECIES FILTERS
+# =====================================================
+
+species_filters = {
+    "Birds": "class=Aves",
+    "Plants": "kingdom=Plantae",
+    "Mammals": "class=Mammalia",
+    "Reptiles": "class=Reptilia",
+    "Fish": "class=Actinopterygii",
+    "Insects": "class=Insecta"
+}
+
+
+# =====================================================
+# GET SPECIES FROM GBIF
+# =====================================================
+
+def get_species(country_code, filter_value):
+
+    url = f"https://api.gbif.org/v1/occurrence/search?country={country_code}&{filter_value}&limit=10"
+
+    try:
+
+        response = requests.get(url)
+
+        data = response.json()
+
+        results = data.get("results", [])
+
+        species = []
+
+        for item in results:
+
+            sp = item.get("species")
+
+            if sp and sp not in species:
+                species.append(sp)
+
+        return species
+
     except:
-        return str(result)
 
+        return []
+
+
+# =====================================================
+# GET LINKS
+# =====================================================
 
-# =========================================
-# SCIENTIFIC LINKS
-# =========================================
+def get_species_links(species_name, country):
 
-def get_species_links(species_name):
     links = []
-    
-    # تنظيف الاسم: أخذ أول كلمتين فقط لتجنب الأخطاء
-    clean_name = " ".join(species_name.split()[:2]).replace(",", "").replace("(", "").replace(")", "")
 
-    # GBIF - تعديل صيغة الرابط ليكون قابل للضغط
-    gbif_url = f"https://api.gbif.org/v1/species/match?name={clean_name}"
+    clean_name = " ".join(species_name.split()[:2])
+
+    encoded_species = quote(clean_name)
+
+    encoded_country = quote(country)
+
+    # -------------------------------------------------
+    # GBIF
+    # -------------------------------------------------
+
+    gbif_api = f"https://api.gbif.org/v1/species/match?name={encoded_species}"
+
     try:
-        r = requests.get(gbif_url).json()
+
+        r = requests.get(gbif_api).json()
+
         if "usageKey" in r:
-            full_url = f"https://www.gbif.org/species/{r['usageKey']}"
-            links.append(f"* [GBIF Portal]({full_url})")
+
+            gbif_link = f"https://www.gbif.org/species/{r['usageKey']}"
+
+            links.append(
+                f"- [GBIF Species Page]({gbif_link})"
+            )
+
     except:
         pass
 
-    # Wikipedia - تعديل صيغة الرابط ليكون قابل للضغط
-    wiki_name = clean_name.replace(' ', '_')
+    # -------------------------------------------------
+    # Wikipedia
+    # -------------------------------------------------
+
+    wiki_name = clean_name.replace(" ", "_")
+
     wiki_url = f"https://en.wikipedia.org/wiki/{wiki_name}"
-    links.append(f"* [Wikipedia Article]({wiki_url})")
 
-    # Google Scholar - تعديل صيغة الرابط ليكون قابل للضغط
-    scholar_url = f"https://scholar.google.com/scholar?q={clean_name}"
-    links.append(f"* [Google Scholar Search]({scholar_url})")
+    links.append(
+        f"- [Wikipedia Article]({wiki_url})"
+    )
+
+    # -------------------------------------------------
+    # Google Scholar
+    # -------------------------------------------------
+
+    scholar_url = f"https://scholar.google.com/scholar?q={encoded_species}+{encoded_country}"
+
+    links.append(
+        f"- [Google Scholar ({country})]({scholar_url})"
+    )
+
+    # -------------------------------------------------
+    # iNaturalist
+    # -------------------------------------------------
+
+    inaturalist = f"https://www.inaturalist.org/search?q={encoded_species}"
+
+    links.append(
+        f"- [iNaturalist Observations]({inaturalist})"
+    )
+
+    # -------------------------------------------------
+    # Encyclopedia of Life
+    # -------------------------------------------------
+
+    eol = f"https://eol.org/search?q={encoded_species}"
+
+    links.append(
+        f"- [Encyclopedia of Life]({eol})"
+    )
+
+    # -------------------------------------------------
+    # Biodiversity Heritage Library
+    # -------------------------------------------------
+
+    bhl = f"https://www.biodiversitylibrary.org/search?searchTerm={encoded_species}"
+
+    links.append(
+        f"- [Biodiversity Heritage Library]({bhl})"
+    )
+
+    # -------------------------------------------------
+    # Google Search
+    # -------------------------------------------------
+
+    google_search = f"https://www.google.com/search?q={encoded_species}+in+{encoded_country}"
+
+    links.append(
+        f"- [Google Search ({country})]({google_search})"
+    )
+
+    # -------------------------------------------------
+    # Images Search
+    # -------------------------------------------------
+
+    image_search = f"https://www.google.com/search?tbm=isch&q={encoded_species}+{encoded_country}"
+
+    links.append(
+        f"- [Images Search]({image_search})"
+    )
+
+    # -------------------------------------------------
+    # Research Papers
+    # -------------------------------------------------
+
+    research = f"https://www.google.com/search?q={encoded_species}+{encoded_country}+research+paper"
+
+    links.append(
+        f"- [Research Papers]({research})"
+    )
+
+    # -------------------------------------------------
+    # Conservation Information
+    # -------------------------------------------------
+
+    conservation = f"https://www.google.com/search?q={encoded_species}+{encoded_country}+conservation"
+
+    links.append(
+        f"- [Conservation Information]({conservation})"
+    )
 
     return links
 
 
-# =========================================
-# STREAMLIT UI
-# =========================================
+# =====================================================
+# STREAMLIT PAGE
+# =====================================================
+
+st.set_page_config(
+    page_title="Biodiversity GIS Explorer",
+    layout="wide"
+)
 
-st.set_page_config(page_title="Biodiversity GIS Explorer", layout="wide")
 st.title("🌍 Biodiversity GIS Explorer")
 
-country = st.text_input("Enter Country")
+st.write(
+    "Explore biodiversity species using GBIF + AI analysis."
+)
+
+# =====================================================
+# USER INPUTS
+# =====================================================
+
+country = st.text_input(
+    "Enter Country"
+)
 
 species_type = st.selectbox(
     "Select Species Type",
-    ["Birds", "Plants", "Mammals", "Reptiles", "Fish"]
+    [
+        "Birds",
+        "Plants",
+        "Mammals",
+        "Reptiles",
+        "Fish",
+        "Insects"
+    ]
 )
 
-# =========================================
-# MAIN BUTTON
-# =========================================
+# =====================================================
+# MAIN SEARCH BUTTON
+# =====================================================
 
 if st.button("Search Biodiversity"):
 
-    # GBIF API Search
-    url = f"https://api.gbif.org/v1/species/search?q={species_type}&limit=5"
-    response = requests.get(url)
-    data = response.json()
+    # -------------------------------------------------
+    # VALIDATION
+    # -------------------------------------------------
 
-    results = data.get("results", [])
+    if not country:
 
-    species_list = [
-        item.get("scientificName", "Unknown Species")
-        for item in results if "scientificName" in item
-    ]
+        st.warning(
+            "Please enter a country."
+        )
 
-    # عرض الأسماء الموجودة
-    species_text = "\n".join(species_list)
+        st.stop()
+
+    country_code = country_codes.get(country)
+
+    if not country_code:
+
+        st.error(
+            "Country not supported yet."
+        )
+
+        st.stop()
+
+    filter_value = species_filters.get(
+        species_type
+    )
+
+    # -------------------------------------------------
+    # FETCH SPECIES
+    # -------------------------------------------------
+
+    with st.spinner(
+        "Fetching biodiversity data..."
+    ):
+
+        species_list = get_species(
+            country_code,
+            filter_value
+        )
+
+    if not species_list:
+
+        st.warning(
+            "No species found."
+        )
+
+        st.stop()
+
+    # -------------------------------------------------
+    # DISPLAY SPECIES
+    # -------------------------------------------------
 
     st.subheader("🐾 Species Found")
-    st.text(species_text)
 
-    # Gemini
+    for sp in species_list:
+
+        st.write(f"• {sp}")
+
+    # -------------------------------------------------
+    # AI ANALYSIS
+    # -------------------------------------------------
+
+    species_text = "\n".join(species_list)
+
     prompt = f"""
-Country: {country}
-Species Type: {species_type}
+You are a biodiversity scientist.
+
+Country:
+{country}
 
-Species List:
+Species Type:
+{species_type}
+
+Species Found:
 {species_text}
+
+Write:
+1. Biodiversity summary
+2. Ecological importance
+3. Conservation challenges
+4. GIS/environment insights
 """
 
-    ai_text = call_gemini(prompt, os.getenv("GEMINI_API_KEY"))
+    with st.spinner(
+        "Generating AI analysis..."
+    ):
+
+        ai_text = call_gemini(
+            prompt,
+            os.getenv("GEMINI_API_KEY")
+        )
+
+    st.subheader(
+        "🤖 AI Biodiversity Analysis"
+    )
+
     st.write(ai_text)
 
-    # --- عرض الروابط النشطة والقابلة للضغط ---
-    st.subheader("🔗 Scientific Links")
-    
-    with st.spinner("Fetching scientific links..."):
-        all_links_text = ""
+    # =================================================
+    # LINKS SECTION
+    # =================================================
+
+    st.subheader(
+        "🔗 Scientific & Country Related Links"
+    )
+
+    with st.spinner(
+        "Generating links..."
+    ):
+
         for sp in species_list:
-            links = get_species_links(sp)
-            if links:
-                all_links_text += f"\n### 🔬 {sp}\n" + "\n".join(links) + "\n"
-        
-        if all_links_text.strip():
-            st.markdown(all_links_text)  # عرض الروابط بصيغة الـ Markdown النشطة
-        else:
-            st.warning("No specific links could be generated for these species names.")
+
+            st.markdown(f"## 🔬 {sp}")
+
+            links = get_species_links(
+                sp,
+                country
+            )
+
+            for link in links:
+
+                st.markdown(link)
+
+            st.markdown("---")