Update app.py

app.py

@@ -8,14 +8,12 @@ from fpdf import FPDF
 # =========================================
 
 def call_gemini(prompt, api_key):
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-3.1-flash-lite:generateContent?key={api_key}"
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={api_key}"
 
     payload = {
         "contents": [
             {
-                "parts": [
-                    {"text": prompt}
-                ]
+                "parts": [{"text": prompt}]
             }
         ]
     }
@@ -36,7 +34,7 @@ def call_gemini(prompt, api_key):
 def get_species_links(species_name):
     links = []
 
-    # GBIF official page
+    # GBIF
     gbif_url = f"https://api.gbif.org/v1/species/match?name={species_name}"
     r = requests.get(gbif_url).json()
 
@@ -46,7 +44,7 @@ def get_species_links(species_name):
     # Wikipedia
     links.append(f"https://en.wikipedia.org/wiki/{species_name.replace(' ', '_')}")
 
-    # Google Scholar (REAL research results)
+    # Google Scholar
     links.append(f"https://scholar.google.com/scholar?q={species_name}")
 
     return links
@@ -58,11 +56,6 @@ def get_species_links(species_name):
 
 st.set_page_config(page_title="Biodiversity GIS Explorer", layout="wide")
 st.title("🌍 Biodiversity GIS Explorer")
-st.write("GBIF + Gemini AI + Scientific Links + PDF Export")
-
-# =========================
-# INPUTS
-# =========================
 
 country = st.text_input("Enter Country")
 
@@ -71,9 +64,9 @@ species_type = st.selectbox(
     ["Birds", "Plants", "Mammals", "Reptiles", "Fish"]
 )
 
-# =========================
-# BUTTON
-# =========================
+# =========================================
+# MAIN BUTTON
+# =========================================
 
 if st.button("Search Biodiversity"):
 
@@ -98,6 +91,7 @@ if st.button("Search Biodiversity"):
         for item in results
     ]
 
+    # ✅ IMPORTANT: define this before using it
     species_text = "\n".join(species_list)
 
     st.subheader("🐾 Species Found")
@@ -128,7 +122,7 @@ Species List:
     st.write(ai_text)
 
     # =========================
-    # SCIENTIFIC LINKS
+    # LINKS
     # =========================
 
     st.subheader("📚 Research Links")
@@ -145,11 +139,11 @@ Species List:
 
         all_links_text += f"\n{sp}:\n" + "\n".join(links) + "\n"
 
-# =========================
-# PDF EXPORT (FIXED)
-# =========================
+    # =========================
+    # PDF EXPORT
+    # =========================
 
-final_text = f"""
+    final_text = f"""
 BIODIVERSITY REPORT
 
 Country: {country}
@@ -165,21 +159,20 @@ RESEARCH LINKS:
 {all_links_text}
 """
 
-pdf = FPDF()
-pdf.add_page()
-pdf.set_font("Arial", size=12)
-
-for line in final_text.split("\n"):
-    pdf.multi_cell(0, 10, line)
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
 
-pdf_file = "biodiversity_report.pdf"
+    for line in final_text.split("\n"):
+        pdf.multi_cell(0, 10, line)
 
-pdf.output(pdf_file)
+    pdf_file = "biodiversity_report.pdf"
+    pdf.output(pdf_file)
 
-with open(pdf_file, "rb") as f:
-    st.download_button(
-        label="📄 Download PDF Report",
-        data=f,
-        file_name="biodiversity_report.pdf",
-        mime="application/pdf"
-    )
+    with open(pdf_file, "rb") as f:
+        st.download_button(
+            label="📄 Download PDF Report",
+            data=f,
+            file_name="biodiversity_report.pdf",
+            mime="application/pdf"
+        )