Update app.py

app.py

@@ -18,8 +18,8 @@ client = Groq(
 # Language translations
 translations = {
     "en": {
-        "app_title": "Bird Species Identification for Researchers",
-        "app_description": "Upload an image to identify bird species and get detailed information relevant to research in Tanzania and climate change studies.",
+        "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",
@@ -33,11 +33,12 @@ translations = {
         "question_title": "Question:",
         "answer_title": "Answer:",
         "habitat_map_title": "Natural Habitat Map for",
-        "detailed_info_title": "Detailed Information"
+        "detailed_info_title": "Detailed Information",
+        "language_label": "Language / Lugha"
     },
     "sw": {
-        "app_title": "Utambuzi wa Spishi za Ndege kwa Watafiti",
-        "app_description": "Pakia picha ili kutambua spishi za ndege na kupata taarifa muhimu zinazohusiana na utafiti nchini Tanzania na masomo ya mabadiliko ya tabianchi.",
+        "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",
@@ -51,7 +52,8 @@ translations = {
         "question_title": "Swali:",
         "answer_title": "Jibu:",
         "habitat_map_title": "Ramani ya Makazi Asilia ya",
-        "detailed_info_title": "Taarifa za Kina"
+        "detailed_info_title": "Taarifa za Kina",
+        "language_label": "Language / Lugha"
     }
 }
 
@@ -375,156 +377,149 @@ 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)}"
 
-def update_conversation(question, bird_name, history, language="en"):
-    """Update the conversation history with new Q&A"""
-    t = translations[language]
+# 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']}")
+        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=translations['en']['upload_label'])
+            submit_btn = gr.Button(translations['en']['identify_button'], variant="primary")
+        
+        with gr.Column(scale=2):
+            prediction_output = gr.Label(label=translations['en']['predictions_label'], num_top_classes=5)
+            bird_info_output = gr.HTML(label=translations['en']['bird_info_label'])
     
-    if not question.strip():
-        return history
+    # Clear divider
+    gr.Markdown("---")
     
-    answer = follow_up_question(question, bird_name, language)
+    # Follow-up question section with improved UI
+    questions_header = gr.Markdown(f"## {translations['en']['research_questions']}")
     
-    # Format the conversation with clear separation
-    new_exchange = f"""
+    conversation_history = gr.Markdown("")
+    
+    with gr.Row():
+        follow_up_input = gr.Textbox(
+            label=translations['en']['question_label'], 
+            placeholder=translations['en']['question_placeholder'],
+            lines=2
+        )
+    
+    with gr.Row():
+        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'], "", ""
+        
+        try:
+            pred_results, info, clean_bird_name = predict_and_get_info(img, lang)
+            return pred_results, info, clean_bird_name, ""
+        except Exception as 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, lang):
+        t = translations[lang]
+        
+        if not question.strip():
+            return history
+        
+        answer = follow_up_question(question, bird_name, lang)
+        
+        # Format the conversation with clear separation
+        new_exchange = f"""
 ### {t['question_title']}
 {question}
 ### {t['answer_title']}
 {answer}
 ---
 """
-    updated_history = new_exchange + history
-    return updated_history
-
-# Create the Gradio interface
-def create_interface(language="en"):
-    t = translations[language]
+        updated_history = new_exchange + history
+        return updated_history
     
-    with gr.Blocks(theme=gr.themes.Soft()) as app:
-        # Language selector
-        with gr.Row():
-            with gr.Column(scale=3):
-                gr.Markdown(f"# {t['app_title']}")
-            with gr.Column(scale=1):
-                language_selector = gr.Radio(
-                    choices=["English", "Kiswahili"], 
-                    label="Language / Lugha", 
-                    value="English" if language == "en" else "Kiswahili"
-                )
-        
-        gr.Markdown(f"{t['app_description']}")
-        
-        # Store the current bird and language for context
-        current_bird = gr.State("")
-        current_lang = gr.State(language)
-        
-        # Main identification section
-        with gr.Row():
-            with gr.Column(scale=1):
-                input_image = gr.Image(type="pil", label=t['upload_label'])
-                submit_btn = gr.Button(t['identify_button'], variant="primary")
-            
-            with gr.Column(scale=2):
-                prediction_output = gr.Label(label=t['predictions_label'], num_top_classes=5)
-                bird_info_output = gr.HTML(label=t['bird_info_label'])
-        
-        # Clear divider
-        gr.Markdown("---")
-        
-        # Follow-up question section with improved UI
-        gr.Markdown(f"## {t['research_questions']}")
-        
-        conversation_history = gr.Markdown("")
-        
-        with gr.Row():
-            follow_up_input = gr.Textbox(
-                label=t['question_label'], 
-                placeholder=t['question_placeholder'],
-                lines=2
-            )
-        
-        with gr.Row():
-            follow_up_btn = gr.Button(t['submit_question'], variant="primary")
-            clear_btn = gr.Button(t['clear_conversation'])
-        
-        # Set up event handlers
-        def process_image(img, lang):
-            if img is None:
-                return None, translations[lang]['upload_prompt'], "", ""
-            
-            try:
-                pred_results, info, clean_bird_name = predict_and_get_info(img, lang)
-                return pred_results, info, clean_bird_name, ""
-            except Exception as e:
-                error_msg = "Hitilafu katika kuchakata picha" if lang == "sw" else "Error processing image"
-                return None, f"{error_msg}: {str(e)}", "", ""
-        
-        def clear_conversation_history():
-            return ""
-        
-        def switch_language(choice):
-            new_lang = "sw" if choice == "Kiswahili" else "en"
-            return new_lang
-        
-        # Connect language selector to recreate the interface
-        language_selector.change(
-            switch_language,
-            inputs=[language_selector],
-            outputs=[current_lang]
-        ).then(
-            lambda lang: gr.Blocks.update(visible=False),
-            inputs=[current_lang],
-            outputs=[app]
-        ).then(
-            lambda: None,
-            None,
-            None,
-            _js=f"""() => {{
-                // Reload the page with the new language parameter
-                const url = new URL(window.location.href);
-                const lang = document.querySelector('input[name="language-selector"]:checked').value;
-                url.searchParams.set('language', lang === 'Kiswahili' ? 'sw' : 'en');
-                window.location.href = url.toString();
-            }}"""
-        )
-        
-        submit_btn.click(
-            process_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, current_lang],
-            outputs=[conversation_history]
-        ).then(
-            lambda: "", 
-            outputs=follow_up_input
-        )
+    def clear_conversation_history():
+        return ""
+    
+    def update_language(choice):
+        # Convert selection to language code
+        lang = "sw" if choice == "Kiswahili" else "en"
+        t = translations[lang]
         
-        clear_btn.click(
-            clear_conversation_history,
-            outputs=[conversation_history]
+        # 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']
         )
-        
-    return app
-
-# Get language from URL parameter or default to English
-def launch_app():
-    import sys
-    
-    # Check if running in Gradio's development environment
-    if len(sys.argv) > 1 and sys.argv[1] == "run":
-        # Default to English when running locally
-        language = "en"
-    else:
-        # Get language from URL parameter
-        import os
-        language_param = os.environ.get("GRADIO_LANGUAGE", "en")
-        language = language_param if language_param in ["en", "sw"] else "en"
     
-    app = create_interface(language)
-    app.launch(share=True)
+    # 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, 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, current_lang],
+        outputs=[conversation_history]
+    ).then(
+        lambda: "", 
+        outputs=follow_up_input
+    )
+    
+    clear_btn.click(
+        clear_conversation_history,
+        outputs=[conversation_history]
+    )
 
-if __name__ == "__main__":
-    launch_app()
+# Launch the app
+app.launch(share=True)