Update app.py

app.py

@@ -7,6 +7,7 @@ from groq import Groq
 from PIL import Image
 import time
 import json
+import numpy as np
 from functools import lru_cache
 
 # Load the trained model
@@ -71,6 +72,7 @@ translations = {
     }
 }
 
+
 def clean_bird_name(name):
     """Clean bird name by removing numbers and special characters, and fix formatting"""
     # Remove numbers and dots at the beginning
@@ -83,11 +85,13 @@ def clean_bird_name(name):
     cleaned = ' '.join(cleaned.split())
     return cleaned
 
+
 def get_cache_path(function_name, key):
     """Generate a cache file path"""
     safe_key = re.sub(r'[^\w]', '_', key)
     return f"cache/{function_name}_{safe_key}.json"
 
+
 def save_to_cache(function_name, key, data):
     """Save API response to cache"""
     try:
@@ -97,6 +101,7 @@ def save_to_cache(function_name, key, data):
     except Exception as e:
         print(f"Error saving to cache: {e}")
 
+
 def load_from_cache(function_name, key, max_age=86400):  # Default max age: 1 day
     """Load API response from cache if it exists and is not too old"""
     try:
@@ -110,6 +115,7 @@ def load_from_cache(function_name, key, max_age=86400):  # Default max age: 1 da
         print(f"Error loading from cache: {e}")
     return None
 
+
 def is_likely_bird_image(img):
     """Basic check to see if the image might contain a bird"""
     try:
@@ -136,6 +142,7 @@ def is_likely_bird_image(img):
         # If any error occurs during the check, assume it might be a bird
         return True
 
+
 def get_bird_habitat_map(bird_name, check_tanzania=True):
     """Get habitat map locations for the bird using Groq API with caching"""
     clean_name = clean_bird_name(bird_name)
@@ -227,6 +234,7 @@ def get_bird_habitat_map(bird_name, check_tanzania=True):
         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"""
     # Find center point based on valid coordinates
@@ -266,6 +274,7 @@ def create_habitat_map(habitat_locations):
     map_html = m._repr_html_()
     return map_html
 
+
 def format_bird_info(raw_info, language="en"):
     """Improve the formatting of bird information"""
     # Add proper line breaks between sections and ensure consistent heading levels
@@ -293,6 +302,7 @@ def format_bird_info(raw_info, language="en"):
     
     return formatted
 
+
 def get_bird_info(bird_name, language="en"):
     """Get detailed information about a bird using Groq API with caching"""
     clean_name = clean_bird_name(bird_name)
@@ -339,14 +349,14 @@ def get_bird_info(bird_name, language="en"):
         error_msg = "Hitilafu katika kupata taarifa" if language == "sw" else "Error fetching information"
         return f"{error_msg}: {str(e)}"
 
+
 def create_message_html(message, icon="🔍", language="en"):
     """Create a styled message container for notifications"""
     custom_css = """
     <style>
     .message-container {
         font-family: Arial, sans-serif;
-        padding:
-        20px;
+        padding: 20px;
         background-color: #f8f9fa;
         border-radius: 8px;
         text-align: center;
@@ -372,6 +382,7 @@ def create_message_html(message, icon="🔍", language="en"):
     """
     return html
 
+
 def predict_and_get_info(img, language="en"):
     """Predict bird species and get detailed information"""
     # Get translations
@@ -502,6 +513,7 @@ def predict_and_get_info(img, language="en"):
         error_msg = "Hitilafu katika kuchakata picha" if language == "sw" else "Error processing image"
         return None, create_message_html(f"{error_msg}: {str(e)}", "⚠️", language), "", ""
 
+
 def follow_up_question(question, bird_name, language="en"):
     """Allow researchers to ask follow-up questions about the identified bird"""
     t = translations[language]
@@ -551,6 +563,7 @@ def follow_up_question(question, bird_name, language="en"):
         error_msg = "Hitilafu katika kupata taarifa" if language == "sw" else "Error fetching information"
         return f"{error_msg}: {str(e)}"
 
+
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     # Current language and bird state