Update app.py

app.py

@@ -27,10 +27,30 @@ def clean_bird_name(name):
     cleaned = ' '.join(cleaned.split())
     return cleaned
 
-def get_bird_habitat_map(bird_name):
+def get_bird_habitat_map(bird_name, check_tanzania=True):
     """Get habitat map locations for the bird using Groq API"""
     clean_name = clean_bird_name(bird_name)
     
+    # First check if the bird is endemic to Tanzania
+    if check_tanzania:
+        tanzania_check_prompt = f"""
+        Is the {clean_name} bird native to or commonly found in Tanzania? 
+        Answer with ONLY "yes" or "no".
+        """
+        
+        try:
+            tanzania_check = client.chat.completions.create(
+                messages=[{"role": "user", "content": tanzania_check_prompt}],
+                model="llama-3.3-70b-versatile",
+            )
+            is_in_tanzania = "yes" in tanzania_check.choices[0].message.content.lower()
+        except:
+            # Default to showing Tanzania if we can't determine
+            is_in_tanzania = True
+    else:
+        is_in_tanzania = True
+    
+    # Now get the habitat locations
     prompt = f"""
     Provide a JSON array of the main habitat locations for the {clean_name} bird in the world. 
     Return ONLY a JSON array with 3-5 entries, each containing:
@@ -45,7 +65,7 @@ def get_bird_habitat_map(bird_name):
       ...
     ]
     
-    If this bird is not commonly found in Tanzania, include at least one location in Tanzania with a note about its unusual presence there.
+    {'' if is_in_tanzania else 'DO NOT include any locations in Tanzania as this bird is not native to or commonly found there.'}
     """
     
     try:
@@ -71,20 +91,32 @@ def get_bird_habitat_map(bird_name):
         else:
             # Fallback if JSON extraction fails
             locations = [
-                {"name": "Habitat information unavailable", "lat": 0, "lon": 0, 
-                 "description": "Could not retrieve habitat information for this bird."}
+                {"name": "Primary habitat region", "lat": 0, "lon": 0, 
+                 "description": "Could not retrieve specific habitat information for this bird."}
             ]
         
-        return locations
+        return locations, is_in_tanzania
         
     except Exception as e:
-        return [{"name": f"Error retrieving habitat data: {str(e)}", "lat": 0, "lon": 0, 
-                 "description": "Please try again or check your connection."}]
+        return [{"name": "Error retrieving data", "lat": 0, "lon": 0, 
+                "description": "Please try again or check your connection."}], False
 
 def create_habitat_map(habitat_locations):
     """Create a folium map with the habitat locations"""
-    # Create a world-centered map
-    m = folium.Map(location=[20, 0], zoom_start=2)
+    # Find center point based on valid coordinates
+    valid_coords = [(loc.get("lat", 0), loc.get("lon", 0)) 
+                    for loc in habitat_locations 
+                    if loc.get("lat", 0) != 0 or loc.get("lon", 0) != 0]
+    
+    if valid_coords:
+        # Calculate the average of the coordinates
+        avg_lat = sum(lat for lat, _ in valid_coords) / len(valid_coords)
+        avg_lon = sum(lon for _, lon in valid_coords) / len(valid_coords)
+        # Create map centered on the average coordinates
+        m = folium.Map(location=[avg_lat, avg_lon], zoom_start=3)
+    else:
+        # Default world map if no valid coordinates
+        m = folium.Map(location=[20, 0], zoom_start=2)
     
     # Add markers for each habitat location
     for location in habitat_locations:
@@ -108,6 +140,29 @@ def create_habitat_map(habitat_locations):
     map_html = m._repr_html_()
     return map_html
 
+def format_bird_info(raw_info):
+    """Improve the formatting of bird information"""
+    # Add proper line breaks between sections and ensure consistent heading levels
+    formatted = raw_info
+    
+    # Fix heading levels (make all main sections h3)
+    formatted = re.sub(r'#+\s+NOT TYPICALLY FOUND IN TANZANIA', 
+                        '<div class="alert alert-warning"><strong>⚠️ NOT TYPICALLY FOUND IN TANZANIA</strong></div>', 
+                        formatted)
+    
+    # Replace markdown headings with HTML headings for better control
+    formatted = re.sub(r'#+\s+(.*)', r'<h3>\1</h3>', formatted)
+    
+    # Add paragraph tags for better spacing
+    formatted = re.sub(r'\n\*\s+(.*)', r'<p>• \1</p>', formatted)
+    formatted = re.sub(r'\n([^<\n].*)', r'<p>\1</p>', formatted)
+    
+    # Remove any duplicate paragraph tags
+    formatted = formatted.replace('<p><p>', '<p>')
+    formatted = formatted.replace('</p></p>', '</p>')
+    
+    return formatted
+
 def get_bird_info(bird_name):
     """Get detailed information about a bird using Groq API"""
     clean_name = clean_bird_name(bird_name)
@@ -162,22 +217,68 @@ def predict_and_get_info(img):
     clean_top_bird = clean_bird_name(top_bird)
     
     # Get habitat locations and create map
-    habitat_locations = get_bird_habitat_map(top_bird)
+    habitat_locations, is_in_tanzania = get_bird_habitat_map(top_bird)
     habitat_map_html = create_habitat_map(habitat_locations)
     
     # Get detailed information about the top predicted bird
     bird_info = get_bird_info(top_bird)
+    formatted_info = format_bird_info(bird_info)
+    
+    # Create combined info with map at the top and properly formatted information
+    custom_css = """
+    <style>
+    .bird-container {
+        font-family: Arial, sans-serif;
+        padding: 10px;
+    }
+    .map-container {
+        height: 400px;
+        width: 100%;
+        border: 1px solid #ddd;
+        border-radius: 8px;
+        overflow: hidden;
+        margin-bottom: 20px;
+    }
+    .info-container {
+        line-height: 1.6;
+    }
+    .info-container h3 {
+        margin-top: 20px;
+        margin-bottom: 10px;
+        color: #2c3e50;
+        border-bottom: 1px solid #eee;
+        padding-bottom: 5px;
+    }
+    .info-container p {
+        margin-bottom: 10px;
+    }
+    .alert {
+        padding: 10px;
+        margin-bottom: 15px;
+        border-radius: 4px;
+    }
+    .alert-warning {
+        background-color: #fcf8e3;
+        border: 1px solid #faebcc;
+        color: #8a6d3b;
+    }
+    </style>
+    """
     
-    # Create combined info with map at the top
     combined_info = f"""
-## Natural Habitat Map for {clean_top_bird}
-<div style="height: 400px; width: 100%; border: 1px solid #ddd; border-radius: 8px; overflow: hidden; margin-bottom: 20px;">
-{habitat_map_html}
-</div>
-
-## Detailed Information
-{bird_info}
-"""
+    {custom_css}
+    <div class="bird-container">
+        <h2>Natural Habitat Map for {clean_top_bird}</h2>
+        <div class="map-container">
+            {habitat_map_html}
+        </div>
+        
+        <div class="info-container">
+            <h2>Detailed Information</h2>
+            {formatted_info}
+        </div>
+    </div>
+    """
     
     return prediction_results, combined_info, clean_top_bird