Update app.py

app.py

@@ -3,22 +3,15 @@ import gradio as gr
 import re
 import folium
 from fastai.vision.all import *
-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
 learn = load_learner('export.pkl')
 labels = learn.dls.vocab
 
-# Initialize Groq client
-client = Groq(
-    api_key=os.environ.get("GROQ_API_KEY"),
-)
-
 # Cache directory for API responses
 os.makedirs("cache", exist_ok=True)
 
@@ -72,7 +65,6 @@ 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
@@ -85,13 +77,11 @@ 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:
@@ -101,7 +91,6 @@ 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:
@@ -115,7 +104,6 @@ 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:
@@ -142,98 +130,43 @@ 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"""
+def get_bird_habitat_map(bird_name):
+    """Get habitat map locations for the bird - simplified without Groq"""
     clean_name = clean_bird_name(bird_name)
     
-    # Check cache for Tanzania check
-    tanzania_cache_key = f"{clean_name}_tanzania"
-    cached_tanzania = load_from_cache("tanzania_check", tanzania_cache_key)
-    if cached_tanzania is not None:
-        is_in_tanzania = cached_tanzania
-    else:
-        # 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()
-                # Cache result
-                save_to_cache("tanzania_check", tanzania_cache_key, is_in_tanzania)
-            except:
-                # Default to showing Tanzania if we can't determine
-                is_in_tanzania = True
-        else:
-            is_in_tanzania = True
-    
     # Check cache for habitat locations
     habitat_cache_key = f"{clean_name}_habitat"
     cached_habitat = load_from_cache("habitat", habitat_cache_key)
     if cached_habitat is not None:
-        return cached_habitat, is_in_tanzania
-    
-    # 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:
-    1. "name": Location name
-    2. "lat": Latitude (numeric value)
-    3. "lon": Longitude (numeric value)
-    4. "description": Brief description of why this is a key habitat (2-3 sentences)
-    
-    Example format:
-    [
-      {{"name": "Example Location", "lat": 12.34, "lon": 56.78, "description": "Brief description"}},
-      ...
+        # For already cached data, return with a default Tanzania presence value
+        return cached_habitat, True
+    
+    # Default habitat locations for demonstration purposes
+    default_locations = [
+        {
+            "name": "Serengeti National Park, Tanzania", 
+            "lat": -2.3333, 
+            "lon": 34.8333, 
+            "description": "Major savanna ecosystem with diverse bird habitats."
+        },
+        {
+            "name": "Ngorongoro Conservation Area, Tanzania", 
+            "lat": -3.2, 
+            "lon": 35.5, 
+            "description": "Crater highlands with varying altitudes supporting different bird species."
+        },
+        {
+            "name": "Tarangire National Park, Tanzania", 
+            "lat": -4.0, 
+            "lon": 36.0, 
+            "description": "Known for its baobab trees and seasonal swamps that attract various bird species."
+        }
     ]
     
-    {'' if is_in_tanzania else 'DO NOT include any locations in Tanzania as this bird is not native to or commonly found there.'}
-    """
+    # Cache the default result to avoid regenerating it
+    save_to_cache("habitat", habitat_cache_key, default_locations)
     
-    try:
-        chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
-            model="llama-3.3-70b-versatile",
-        )
-        response = chat_completion.choices[0].message.content
-        
-        # Extract JSON from response (in case there's additional text)
-        import json
-        import re
-        
-        # Find JSON pattern in response
-        json_match = re.search(r'\[.*\]', response, re.DOTALL)
-        if json_match:
-            locations = json.loads(json_match.group())
-        else:
-            # Fallback if JSON extraction fails
-            locations = [
-                {"name": "Primary habitat region", "lat": 0, "lon": 0, 
-                 "description": "Could not retrieve specific habitat information for this bird."}
-            ]
-        
-        # Cache the result
-        save_to_cache("habitat", habitat_cache_key, locations)
-        
-        return locations, is_in_tanzania
-        
-    except Exception as e:
-        return [{"name": "Error retrieving data", "lat": 0, "lon": 0, 
-                "description": "Please try again or check your connection."}], False
-
+    return default_locations, True  # Assume bird is in Tanzania for simplicity
 
 def create_habitat_map(habitat_locations):
     """Create a folium map with the habitat locations"""
@@ -274,7 +207,6 @@ 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
@@ -302,9 +234,8 @@ 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"""
+    """Get detailed information about a bird - simplified without Groq"""
     clean_name = clean_bird_name(bird_name)
     
     # Check cache first
@@ -313,42 +244,70 @@ def get_bird_info(bird_name, language="en"):
     if cached_info is not None:
         return cached_info
     
-    # Adjust language for the prompt
-    lang_instruction = ""
-    if language == "sw":
-        lang_instruction = " Provide your response in Swahili language."
+    # Generate basic info based on the bird name
+    # This is a placeholder - in a real application, you would replace this with 
+    # a database lookup or another API call
     
-    prompt = f"""
-    Provide detailed information about the {clean_name} bird, including:
-    1. Physical characteristics and appearance
-    2. Habitat and distribution
-    3. Diet and behavior
-    4. Migration patterns (emphasize if this pattern has changed in recent years due to climate change)
-    5. Conservation status
+    is_in_tanzania = True  # Default assumption for demonstration
     
-    If this bird is not commonly found in Tanzania, explicitly flag that this bird is "NOT TYPICALLY FOUND IN TANZANIA" at the beginning of your response and explain why its presence might be unusual.
+    if language == "sw":
+        bird_info = f"""
+## Maelezo ya {clean_name}
+
+### Sifa za Kimwili na Mwonekano
+• Ndege huyu ana umbile la wastani na rangi nyeupe na nyeusi.
+• Ana mdomo mrefu na miguu myeupe.
+
+### Makazi na Usambazaji
+• Hupatikana sana katika misitu na viunga vya maji nchini Tanzania.
+• Hupendelea maeneo yenye miti mingi na vyanzo vya maji.
+
+### Lishe na Tabia
+• Hula wadudu na matunda madogo madogo.
+• Hutengeneza viota vyake juu ya miti.
+
+### Mifumo ya Uhamiaji
+• Sio ndege mhamiaji, hupatikana nchini Tanzania mwaka mzima.
+• Husafiri katika makundi madogo madogo.
+
+### Hali ya Uhifadhi
+• Haijaathiriwa sana na mabadiliko ya tabianchi.
+• Idadi yake ipo salama kwa sasa.
+"""
+    else:
+        bird_info = f"""
+## About the {clean_name}
+
+### Physical Characteristics and Appearance
+• This bird has a medium build with distinctive white and black coloration.
+• It has a long beak and white legs.
+
+### Habitat and Distribution
+• Commonly found in forests and near water bodies in Tanzania.
+• Prefers areas with abundant trees and water sources.
+
+### Diet and Behavior
+• Feeds on insects and small fruits.
+• Builds nests high up in trees.
+
+### Migration Patterns
+• Not a migratory bird, present in Tanzania year-round.
+• Travels in small flocks.
+
+### Conservation Status
+• Not significantly affected by climate change yet.
+• Population remains stable.
+"""
     
-    Format your response in markdown for better readability.{lang_instruction}
-    """
+    # Add warning if not in Tanzania (for demonstration purposes, we'll assume random birds aren't in Tanzania)
+    import random
+    if not is_in_tanzania or random.random() > 0.7:  # 30% chance to be marked as not in Tanzania
+        warning = "NOT TYPICALLY FOUND IN TANZANIA\n\n" if language == "en" else "HAPATIKANI SANA TANZANIA\n\n"
+        bird_info = warning + bird_info
     
-    try:
-        chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
-            model="llama-3.3-70b-versatile",
-        )
-        response = chat_completion.choices[0].message.content
-        # Cache the result
-        save_to_cache("bird_info", cache_key, response)
-        return response
-    except Exception as e:
-        error_msg = "Hitilafu katika kupata taarifa" if language == "sw" else "Error fetching information"
-        return f"{error_msg}: {str(e)}"
-
+    # Cache the result
+    save_to_cache("bird_info", cache_key, bird_info)
+    return bird_info
 
 def create_message_html(message, icon="🔍", language="en"):
     """Create a styled message container for notifications"""
@@ -382,7 +341,6 @@ 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
@@ -513,7 +471,6 @@ 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]
@@ -527,42 +484,54 @@ def follow_up_question(question, bird_name, language="en"):
     if cached_answer is not None:
         return cached_answer
     
-    # Adjust language for the prompt
-    lang_instruction = ""
-    if language == "sw":
-        lang_instruction = " Provide your response in Swahili language."
-    
-    prompt = f"""
-    The researcher is asking about the {bird_name} bird: "{question}"
-    
-    Provide a detailed, scientific answer focusing on accurate ornithological information. 
-    If the question relates to Tanzania or climate change impacts, emphasize those aspects in your response.
-    
-    IMPORTANT: Do not repeat basic introductory information about the bird that would have already been provided in a general description.
-    Do not start your answer with phrases like "Introduction to the {bird_name}" or similar repetitive headers.
-    Directly answer the specific question asked.
+    # Generate a simple response based on the question
+    # This is a placeholder - in a real application, you would replace this with
+    # a database lookup or another API call
+    
+    clean_bird = clean_bird_name(bird_name)
+    
+    response_templates = {
+        "en": {
+            "habitat": f"The {clean_bird} typically inhabits forested areas and wetlands in Tanzania. They prefer areas with abundant tree cover and proximity to water sources for feeding and nesting.",
+            "diet": f"The {clean_bird} primarily feeds on insects, small fruits, and seeds. They forage in trees and sometimes on the ground, using their specialized beak to extract food from various sources.",
+            "migration": f"The {clean_bird} is generally a resident species in Tanzania, though some local movements may occur based on seasonal food availability. Climate change has not significantly altered their distribution patterns yet.",
+            "conservation": f"The {clean_bird} currently has a stable population in Tanzania. Conservation efforts focus on habitat preservation, especially protecting the forests and wetlands where they breed.",
+            "behavior": f"The {clean_bird} is known for its distinctive calls and social behavior. They often form small flocks and are more active during early morning and late afternoon hours.",
+            "default": f"The {clean_bird} is an interesting bird species found in Tanzania. While specific information on your question is limited, ongoing research continues to expand our understanding of this species."
+        },
+        "sw": {
+            "habitat": f"Ndege {clean_bird} huishi katika maeneo ya misitu na maeneo yenye maji nchini Tanzania. Hupendelea maeneo yenye miti mingi na karibu na vyanzo vya maji kwa ajili ya chakula na viota.",
+            "diet": f"Ndege {clean_bird} hula wadudu, matunda madogo, na mbegu. Hutafuta chakula kwenye miti na wakati mwingine ardhini, akitumia mdomo wake maalum kutoa chakula kutoka vyanzo mbalimbali.",
+            "migration": f"Ndege {clean_bird} kwa kawaida huishi Tanzania mwaka mzima, ingawa baadhi ya harakati za ndani hufanyika kulingana na upatikanaji wa chakula kwa msimu. Mabadiliko ya tabianchi hayajabadilisha sana mienendo yao ya usambazaji.",
+            "conservation": f"Ndege {clean_bird} kwa sasa ana idadi imara nchini Tanzania. Juhudi za uhifadhi zinalenga kuhifadhi makazi, hasa kulinda misitu na nyanda za maji ambapo wanazaliana.",
+            "behavior": f"Ndege {clean_bird} anajulikana kwa mwito wake wa kipekee na tabia ya kijamii. Mara nyingi huunda makundi madogo na wana shughuli zaidi wakati wa asubuhi na mchana.",
+            "default": f"Ndege {clean_bird} ni spishi ya ndege ya kuvutia inayopatikana Tanzania. Ingawa taarifa mahususi kuhusu swali lako ni ndogo, utafiti unaoendelea unaendelea kupanua uelewa wetu wa spishi hii."
+        }
+    }
     
-    Format your response in markdown for better readability.{lang_instruction}
-    """
+    # Simple keyword matching for response selection
+    question_lower = question.lower()
+    if language in response_templates:
+        templates = response_templates[language]
+        if any(word in question_lower for word in ["habitat", "live", "found", "where", "makazi", "anaishi", "anapatikana"]):
+            response = templates["habitat"]
+        elif any(word in question_lower for word in ["eat", "food", "diet", "feeding", "chakula", "kula", "lishe"]):
+            response = templates["diet"]
+        elif any(word in question_lower for word in ["migration", "move", "travel", "season", "uhamiaji", "safiri", "msimu"]):
+            response = templates["migration"]
+        elif any(word in question_lower for word in ["conserve", "protect", "endangered", "status", "uhifadhi", "linda", "hatarini"]):
+            response = templates["conservation"]
+        elif any(word in question_lower for word in ["behavior", "behaviour", "habit", "social", "tabia", "mazoea", "jamii"]):
+            response = templates["behavior"]
+        else:
+            response = templates["default"]
+    else:
+        # Fallback to English
+        response = response_templates["en"]["default"]
     
-    try:
-        chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
-            model="llama-3.3-70b-versatile",
-        )
-        response = chat_completion.choices[0].message.content
-        # Cache the result
-        save_to_cache("follow_up", cache_key, response)
-        return response
-    except Exception as e:
-        error_msg = "Hitilafu katika kupata taarifa" if language == "sw" else "Error fetching information"
-        return f"{error_msg}: {str(e)}"
-
+    # Cache the result
+    save_to_cache("follow_up", cache_key, response)
+    return response
 
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as app: