Update app.py

app.py

@@ -74,7 +74,6 @@ def get_bird_habitat_map(bird_name, check_tanzania=True):
         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}],
@@ -119,7 +118,7 @@ def get_bird_habitat_map(bird_name, check_tanzania=True):
                  "description": "Could not retrieve specific habitat information for this bird."}
             ]
         return locations, is_in_tanzania
-    except:
+    except Exception as e:
         return [{"name": "Error retrieving data", "lat": 0, "lon": 0, 
                 "description": "Please try again or check your connection."}], False
 
@@ -157,7 +156,6 @@ def create_habitat_map(habitat_locations):
 def format_bird_info(raw_info, language="en"):
     """Improve the formatting of bird information"""
     formatted = raw_info
-    
     warning_text = "NOT TYPICALLY FOUND IN TANZANIA"
     warning_translation = "HAPATIKANI SANA TANZANIA" if language == "sw" else warning_text
     
@@ -171,7 +169,6 @@ def format_bird_info(raw_info, language="en"):
     
     formatted = formatted.replace('<p><p>', '<p>')
     formatted = formatted.replace('</p></p>', '</p>')
-    
     return formatted
 
 def get_bird_info(bird_name, language="en"):
@@ -220,6 +217,7 @@ def predict_and_get_info(img, language="en"):
     top_bird = str(pred)
     clean_top_bird = clean_bird_name(top_bird)
     
+    # Check if top prediction is "Other" with high confidence (>80%)
     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. "
@@ -420,7 +418,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
 {answer}
 ---
 """
-        return new_exchange + history
+        updated_history = new_exchange + history
+        return updated_history
     
     def clear_conversation_history():
         return ""
@@ -483,4 +482,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         outputs=[conversation_history]
     )
 
+# Launch the app
 app.launch(share=True)