Update app.py

app.py

@@ -241,24 +241,54 @@ for i, challenge in enumerate(st.session_state.daily_challenges):
 st.session_state.points = len(completed_challenges) * 10  # 10 points per challenge
 st.markdown(f"<h2 style='text-align: center;'>💰 Points: {st.session_state.points}</h2>", unsafe_allow_html=True)
 
-# Check if all challenges are completed
+# Function to fetch endangered species data for a region
+def fetch_all_endangered_species(city: str) -> list:
+    prompt = (
+        f"Provide a list of endangered species found near {city}, "
+        "including their names, population estimates, and descriptions. "
+        "Return the data in JSON format as a list of dictionaries."
+    )
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[{"role": "user", "content": prompt}],
+        max_tokens=300,
+        temperature=0.8
+    )
+    response_content = response.choices[0].message['content'].strip()
+    
+    try:
+        species_list = eval(response_content)  # Assuming the response is a JSON-like structure
+        if isinstance(species_list, list) and all(isinstance(species, dict) for species in species_list):
+            return species_list
+        else:
+            st.error("Unexpected response format. Could not parse species data.")
+            return []
+    except Exception as e:
+        st.error(f"Error fetching endangered species data: {e}")
+        return []
+
+# Display the endangered species section
 if len(completed_challenges) == len(st.session_state.daily_challenges):
     st.success("All challenges completed! 🎉 You've unlocked the secret section!")
 
     # Fetch endangered species data for the user's city
-    species_data = fetch_endangered_species(city)
-    if species_data:
-        st.subheader("🦋 Endangered Species Unlocked!")
-        st.write(f"**Species**: {species_data.get('name', 'Unknown')}")
-        st.write(f"**Population**: {species_data.get('population', 'Unknown')}")
-        st.write(f"**Description**: {species_data.get('description', 'No description available')}")
-
-        # Generate an image of the endangered species
-        image_description = f"Generate an image of the endangered species: {species_data.get('name', 'Unknown')}"
-        species_image_bytes = generate_image(image_description)
-        if species_image_bytes:
-            species_image = Image.open(BytesIO(species_image_bytes))
-            st.image(species_image, caption=f"Endangered Species: {species_data.get('name', 'Unknown')}", use_column_width=True)
+    species_data_list = fetch_all_endangered_species(city)
+    if species_data_list:
+        st.subheader("🦋 Endangered Species in Your Region")
+        
+        for species_data in species_data_list:
+            species_name = species_data.get('name', 'Unknown')
+            st.write(f"**Species**: {species_name}")
+            st.write(f"**Population**: {species_data.get('population', 'Unknown')}")
+            st.write(f"**Description**: {species_data.get('description', 'No description available')}")
+
+            # Generate an image of each endangered species
+            image_description = f"Generate an image of the endangered species: {species_name}"
+            species_image_bytes = generate_image(image_description)
+            if species_image_bytes:
+                species_image = Image.open(BytesIO(species_image_bytes))
+                st.image(species_image, caption=f"Endangered Species: {species_name}", use_column_width=True)
+            st.write("---")
 
 # User's Environmental Interests Section
 st.subheader("🌍 Get Involved!")