Update app.py

app.py

@@ -59,13 +59,9 @@ 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
     cleaned = re.sub(r'^\d+\.', '', name)
-    # Replace underscores with spaces
     cleaned = cleaned.replace('_', ' ')
-    # Remove any remaining special characters
     cleaned = re.sub(r'[^\w\s]', '', cleaned)
-    # Fix spacing
     cleaned = ' '.join(cleaned.split())
     return cleaned
 
@@ -73,7 +69,6 @@ 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? 
@@ -87,12 +82,10 @@ def get_bird_habitat_map(bird_name, check_tanzania=True):
             )
             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:
@@ -112,98 +105,70 @@ def get_bird_habitat_map(bird_name, check_tanzania=True):
     
     try:
         chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
+            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."}
             ]
-        
         return locations, is_in_tanzania
-        
-    except Exception as e:
+    except:
         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
     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:
         name = location.get("name", "Unknown")
         lat = location.get("lat", 0)
         lon = location.get("lon", 0)
         description = location.get("description", "No description available")
         
-        # Skip invalid coordinates
         if lat == 0 and lon == 0:
             continue
             
-        # Add marker
         folium.Marker(
             location=[lat, lon],
             popup=folium.Popup(f"<b>{name}</b><br>{description}", max_width=300),
             tooltip=name
         ).add_to(m)
     
-    # Save map to HTML
     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
     formatted = raw_info
     
-    # Get translation of warning text based on language
     warning_text = "NOT TYPICALLY FOUND IN TANZANIA"
     warning_translation = "HAPATIKANI SANA TANZANIA" if language == "sw" else warning_text
     
-    # Fix heading levels (make all main sections h3)
     formatted = re.sub(r'#+\s+' + warning_text, 
-                        f'<div class="alert alert-warning"><strong>⚠️ {warning_translation}</strong></div>', 
-                        formatted)
+                       f'<div class="alert alert-warning"><strong>⚠️ {warning_translation}</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>')
     
@@ -213,10 +178,7 @@ def get_bird_info(bird_name, language="en"):
     """Get detailed information about a bird using Groq API"""
     clean_name = clean_bird_name(bird_name)
     
-    # Adjust language for the prompt
-    lang_instruction = ""
-    if language == "sw":
-        lang_instruction = " Provide your response in Swahili language."
+    lang_instruction = "" if language == "en" else " Provide your response in Swahili language."
     
     prompt = f"""
     Provide detailed information about the {clean_name} bird, including:
@@ -233,12 +195,7 @@ def get_bird_info(bird_name, language="en"):
     
     try:
         chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
+            messages=[{"role": "user", "content": prompt}],
             model="llama-3.3-70b-versatile",
         )
         return chat_completion.choices[0].message.content
@@ -248,38 +205,63 @@ def get_bird_info(bird_name, language="en"):
 
 def predict_and_get_info(img, language="en"):
     """Predict bird species and get detailed information"""
-    # Get translations
     t = translations[language]
     
-    # Process the image
     img = PILImage.create(img)
-    
-    # Get prediction
     pred, pred_idx, probs = learn.predict(img)
     
-    # Get top 5 predictions (or all if less than 5)
     num_classes = min(5, len(labels))
     top_indices = probs.argsort(descending=True)[:num_classes]
     top_probs = probs[top_indices]
     top_labels = [labels[i] for i in top_indices]
     
-    # Format as dictionary with cleaned names for display
     prediction_results = {clean_bird_name(top_labels[i]): float(top_probs[i]) for i in range(num_classes)}
     
-    # Get top prediction (original format for info retrieval)
     top_bird = str(pred)
-    # Also keep a clean version for display
     clean_top_bird = clean_bird_name(top_bird)
     
-    # Get habitat locations and create map
+    if top_bird.lower() == "other" and float(probs[pred_idx]) > 0.8:
+        message = (
+            "This image does not appear to match any bird species in our trained dataset. "
+            "Please upload a clear image of a bird for accurate identification."
+        ) if language == "en" else (
+            "Picha hii haionekani kulingana na spishi yoyote ya ndege katika seti yetu ya mafunzo. "
+            "Tafadhali pakia picha ya wazi ya ndege kwa utambuzi sahihi."
+        )
+        
+        custom_css = """
+        <style>
+        .bird-container {
+            font-family: Arial, sans-serif;
+            padding: 10px;
+        }
+        .alert-info {
+            background-color: #d9edf7;
+            border: 1px solid #bce8f1;
+            color: #31708f;
+            padding: 10px;
+            margin-bottom: 15px;
+            border-radius: 4px;
+        }
+        </style>
+        """
+        
+        combined_info = f"""
+        {custom_css}
+        <div class="bird-container">
+            <div class="alert-info">
+                <strong>ℹ️ {message}</strong>
+            </div>
+        </div>
+        """
+        return prediction_results, combined_info, ""
+    
     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, language)
     formatted_info = format_bird_info(bird_info, language)
     
-    # Create combined info with map at the top and properly formatted information
     custom_css = """
     <style>
     .bird-container {
@@ -344,10 +326,7 @@ def follow_up_question(question, bird_name, language="en"):
     if not question.strip() or not bird_name:
         return "Please identify a bird first and ask a specific question about it." if language == "en" else "Tafadhali tambua ndege kwanza na uulize swali maalum kuhusu ndege huyo."
     
-    # Adjust language for the prompt
-    lang_instruction = ""
-    if language == "sw":
-        lang_instruction = " Provide your response in Swahili language."
+    lang_instruction = "" if language == "en" else " Provide your response in Swahili language."
     
     prompt = f"""
     The researcher is asking about the {bird_name} bird: "{question}"
@@ -364,12 +343,7 @@ def follow_up_question(question, bird_name, language="en"):
     
     try:
         chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
+            messages=[{"role": "user", "content": prompt}],
             model="llama-3.3-70b-versatile",
         )
         return chat_completion.choices[0].message.content
@@ -379,11 +353,9 @@ def follow_up_question(question, bird_name, language="en"):
 
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    # Current language and bird state
     current_lang = gr.State("en")
     current_bird = gr.State("")
     
-    # Header with language switcher
     with gr.Row():
         with gr.Column(scale=3):
             title_md = gr.Markdown(f"# {translations['en']['app_title']}")
@@ -394,10 +366,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
                 value="English"
             )
     
-    # App description
     description_md = gr.Markdown(f"{translations['en']['app_description']}")
     
-    # Main identification section
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(type="pil", label=translations['en']['upload_label'])
@@ -407,10 +377,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             prediction_output = gr.Label(label=translations['en']['predictions_label'], num_top_classes=5)
             bird_info_output = gr.HTML(label=translations['en']['bird_info_label'])
     
-    # Clear divider
     gr.Markdown("---")
     
-    # Follow-up question section with improved UI
     questions_header = gr.Markdown(f"## {translations['en']['research_questions']}")
     
     conversation_history = gr.Markdown("")
@@ -426,7 +394,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         follow_up_btn = gr.Button(translations['en']['submit_question'], variant="primary")
         clear_btn = gr.Button(translations['en']['clear_conversation'])
     
-    # Functions for event handlers
     def process_image(img, lang):
         if img is None:
             return None, translations[lang]['upload_prompt'], "", ""
@@ -446,7 +413,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         
         answer = follow_up_question(question, bird_name, lang)
         
-        # Format the conversation with clear separation
         new_exchange = f"""
 ### {t['question_title']}
 {question}
@@ -454,18 +420,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
 {answer}
 ---
 """
-        updated_history = new_exchange + history
-        return updated_history
+        return new_exchange + history
     
     def clear_conversation_history():
         return ""
     
     def update_language(choice):
-        # Convert selection to language code
         lang = "sw" if choice == "Kiswahili" else "en"
         t = translations[lang]
         
-        # Return updated UI components based on selected language
         return (
             lang, 
             f"# {t['app_title']}", 
@@ -481,7 +444,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             t['clear_conversation']
         )
     
-    # Set up event handlers
     language_selector.change(
         update_language,
         inputs=[language_selector],
@@ -521,5 +483,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         outputs=[conversation_history]
     )
 
-# Launch the app
 app.launch(share=True)