Update app.py

app.py

@@ -15,6 +15,48 @@ client = Groq(
     api_key=os.environ.get("GROQ_API_KEY"),
 )
 
+# Language translations
+translations = {
+    "en": {
+        "app_title": "Bird Species Explorer: AI-Powered Identification for Researchers",
+        "app_description": "🔍 Upload a bird photo to instantly identify species and access comprehensive data on habitats, behaviors, and climate change impacts. A powerful tool for Tanzania-based ornithological research.",
+        "upload_label": "Upload Bird Image",
+        "identify_button": "Identify Bird",
+        "predictions_label": "Top 5 Predictions",
+        "bird_info_label": "Bird Information",
+        "research_questions": "Research Questions",
+        "question_placeholder": "Example: How has climate change affected this bird's migration pattern?",
+        "question_label": "Ask a question about this bird",
+        "submit_question": "Submit Question",
+        "clear_conversation": "Clear Conversation",
+        "upload_prompt": "Please upload an image",
+        "question_title": "Question:",
+        "answer_title": "Answer:",
+        "habitat_map_title": "Natural Habitat Map for",
+        "detailed_info_title": "Detailed Information",
+        "language_label": "Language / Lugha"
+    },
+    "sw": {
+        "app_title": "Mtafiti wa Ndege: Utambuzi wa Kiotomatiki kwa Watafiti",
+        "app_description": "🔍 Pakia picha ya ndege ili kutambua spishi mara moja na kupata data kamili kuhusu makazi, tabia, na athari za mabadiliko ya tabianchi. Zana yenye nguvu kwa utafiti wa ndege nchini Tanzania.",
+        "upload_label": "Pakia Picha ya Ndege",
+        "identify_button": "Tambua Ndege",
+        "predictions_label": "Utabiri Bora 5",
+        "bird_info_label": "Taarifa za Ndege",
+        "research_questions": "Maswali ya Utafiti",
+        "question_placeholder": "Mfano: Je, mabadiliko ya tabianchi yameathiri vipi mfumo wa uhamiaji wa ndege huyu?",
+        "question_label": "Uliza swali kuhusu ndege huyu",
+        "submit_question": "Wasilisha Swali",
+        "clear_conversation": "Futa Mazungumzo",
+        "upload_prompt": "Tafadhali pakia picha",
+        "question_title": "Swali:",
+        "answer_title": "Jibu:",
+        "habitat_map_title": "Ramani ya Makazi Asilia ya",
+        "detailed_info_title": "Taarifa za Kina",
+        "language_label": "Language / Lugha"
+    }
+}
+
 def clean_bird_name(name):
     """Clean bird name by removing numbers and special characters, and fix formatting"""
     # Remove numbers and dots at the beginning
@@ -140,14 +182,18 @@ def create_habitat_map(habitat_locations):
     map_html = m._repr_html_()
     return map_html
 
-def format_bird_info(raw_info):
+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+NOT TYPICALLY FOUND IN TANZANIA', 
-                        '<div class="alert alert-warning"><strong>⚠️ NOT TYPICALLY FOUND IN TANZANIA</strong></div>', 
+    formatted = re.sub(r'#+\s+' + warning_text, 
+                        f'<div class="alert alert-warning"><strong>⚠️ {warning_translation}</strong></div>', 
                         formatted)
     
     # Replace markdown headings with HTML headings for better control
@@ -163,10 +209,15 @@ def format_bird_info(raw_info):
     
     return formatted
 
-def get_bird_info(bird_name):
+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."
+    
     prompt = f"""
     Provide detailed information about the {clean_name} bird, including:
     1. Physical characteristics and appearance
@@ -177,7 +228,7 @@ def get_bird_info(bird_name):
     
     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.
     
-    Format your response in markdown for better readability.
+    Format your response in markdown for better readability.{lang_instruction}
     """
     
     try:
@@ -192,10 +243,14 @@ def get_bird_info(bird_name):
         )
         return chat_completion.choices[0].message.content
     except Exception as e:
-        return f"Error fetching information: {str(e)}"
+        error_msg = "Hitilafu katika kupata taarifa" if language == "sw" else "Error fetching information"
+        return f"{error_msg}: {str(e)}"
 
-def predict_and_get_info(img):
+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)
     
@@ -221,8 +276,8 @@ def predict_and_get_info(img):
     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)
+    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 = """
@@ -268,13 +323,13 @@ def predict_and_get_info(img):
     combined_info = f"""
     {custom_css}
     <div class="bird-container">
-        <h2>Natural Habitat Map for {clean_top_bird}</h2>
+        <h2>{t['habitat_map_title']} {clean_top_bird}</h2>
         <div class="map-container">
             {habitat_map_html}
         </div>
         
         <div class="info-container">
-            <h2>Detailed Information</h2>
+            <h2>{t['detailed_info_title']}</h2>
             {formatted_info}
         </div>
     </div>
@@ -282,10 +337,17 @@ def predict_and_get_info(img):
     
     return prediction_results, combined_info, clean_top_bird
 
-def follow_up_question(question, bird_name):
+def follow_up_question(question, bird_name, language="en"):
     """Allow researchers to ask follow-up questions about the identified bird"""
+    t = translations[language]
+    
     if not question.strip() or not bird_name:
-        return "Please identify a bird first and ask a specific question about it."
+        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."
     
     prompt = f"""
     The researcher is asking about the {bird_name} bird: "{question}"
@@ -297,7 +359,7 @@ def follow_up_question(question, bird_name):
     Do not start your answer with phrases like "Introduction to the {bird_name}" or similar repetitive headers.
     Directly answer the specific question asked.
     
-    Format your response in markdown for better readability.
+    Format your response in markdown for better readability.{lang_instruction}
     """
     
     try:
@@ -312,67 +374,83 @@ def follow_up_question(question, bird_name):
         )
         return chat_completion.choices[0].message.content
     except Exception as e:
-        return f"Error fetching information: {str(e)}"
+        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:
-    gr.Markdown("# Bird Species Identification for Researchers")
-    gr.Markdown("Upload an image to identify bird species and get detailed information relevant to research in Tanzania and climate change studies.")
-    
-    # Store the current bird for context
+    # 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']}")
+        with gr.Column(scale=1):
+            language_selector = gr.Radio(
+                choices=["English", "Kiswahili"], 
+                label=translations['en']['language_label'], 
+                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="Upload Bird Image")
-            submit_btn = gr.Button("Identify Bird", variant="primary")
+            input_image = gr.Image(type="pil", label=translations['en']['upload_label'])
+            submit_btn = gr.Button(translations['en']['identify_button'], variant="primary")
         
         with gr.Column(scale=2):
-            prediction_output = gr.Label(label="Top 5 Predictions", num_top_classes=5)
-            bird_info_output = gr.HTML(label="Bird Information")
+            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
-    gr.Markdown("## Research Questions")
+    questions_header = gr.Markdown(f"## {translations['en']['research_questions']}")
     
     conversation_history = gr.Markdown("")
     
     with gr.Row():
         follow_up_input = gr.Textbox(
-            label="Ask a question about this bird", 
-            placeholder="Example: How has climate change affected this bird's migration pattern?",
+            label=translations['en']['question_label'], 
+            placeholder=translations['en']['question_placeholder'],
             lines=2
         )
     
     with gr.Row():
-        follow_up_btn = gr.Button("Submit Question", variant="primary")
-        clear_btn = gr.Button("Clear Conversation")
+        follow_up_btn = gr.Button(translations['en']['submit_question'], variant="primary")
+        clear_btn = gr.Button(translations['en']['clear_conversation'])
     
-    # Set up event handlers
-    def process_image(img):
+    # Functions for event handlers
+    def process_image(img, lang):
         if img is None:
-            return None, "Please upload an image", "", ""
+            return None, translations[lang]['upload_prompt'], "", ""
         
         try:
-            pred_results, info, clean_bird_name = predict_and_get_info(img)
+            pred_results, info, clean_bird_name = predict_and_get_info(img, lang)
             return pred_results, info, clean_bird_name, ""
         except Exception as e:
-            return None, f"Error processing image: {str(e)}", "", ""
+            error_msg = "Hitilafu katika kuchakata picha" if lang == "sw" else "Error processing image"
+            return None, f"{error_msg}: {str(e)}", "", ""
     
-    def update_conversation(question, bird_name, history):
+    def update_conversation(question, bird_name, history, lang):
+        t = translations[lang]
+        
         if not question.strip():
             return history
         
-        answer = follow_up_question(question, bird_name)
+        answer = follow_up_question(question, bird_name, lang)
         
         # Format the conversation with clear separation
         new_exchange = f"""
-### Question:
+### {t['question_title']}
 {question}
-### Answer:
+### {t['answer_title']}
 {answer}
 ---
 """
@@ -382,15 +460,56 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     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']}", 
+            f"{t['app_description']}", 
+            t['upload_label'],
+            t['identify_button'],
+            t['predictions_label'],
+            t['bird_info_label'],
+            f"## {t['research_questions']}",
+            t['question_label'],
+            t['question_placeholder'],
+            t['submit_question'],
+            t['clear_conversation']
+        )
+    
+    # Set up event handlers
+    language_selector.change(
+        update_language,
+        inputs=[language_selector],
+        outputs=[
+            current_lang,
+            title_md,
+            description_md,
+            input_image,
+            submit_btn,
+            prediction_output,
+            bird_info_output,
+            questions_header,
+            follow_up_input,
+            follow_up_input,
+            follow_up_btn,
+            clear_btn
+        ]
+    )
+    
     submit_btn.click(
         process_image,
-        inputs=[input_image],
+        inputs=[input_image, current_lang],
         outputs=[prediction_output, bird_info_output, current_bird, conversation_history]
     )
     
     follow_up_btn.click(
         update_conversation,
-        inputs=[follow_up_input, current_bird, conversation_history],
+        inputs=[follow_up_input, current_bird, conversation_history, current_lang],
         outputs=[conversation_history]
     ).then(
         lambda: "",