Update app.py

app.py

@@ -362,8 +362,28 @@ def follow_up_question(question, disease_name, language="swahili"):
         error_msg = "Hitilafu kupata taarifa" if language.lower() == "swahili" else "Error fetching information"
         return f"{error_msg}: {str(e)}"
 
+# CSS for styling the language toggle
+toggle_css = """
+<style>
+.language-toggle-container {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    margin-bottom: 20px;
+}
+.toggle-label {
+    font-size: 16px;
+    font-weight: 600;
+    margin-right: 10px;
+}
+</style>
+"""
+
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as app:
+    # Add the CSS for the toggle
+    gr.HTML(toggle_css)
+    
     # State for current language
     current_language = gr.State("swahili")
    
@@ -381,7 +401,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             "question_placeholder": "Mfano: Je, ni hatua gani bora za kuzuia ugonjwa huu?",
             "submit_btn": "Wasilisha Swali",
             "clear_btn": "Futa Mazungumzo",
-            "lang_btn": "Switch to English",
+            "toggle_label": "Lugha/Language:",
             "error_no_image": "Tafadhali pakia picha",
             "error_processing": "Hitilafu katika kuchakata picha"
         },
@@ -397,7 +417,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             "question_placeholder": "Example: What are the best preventative measures for this disease?",
             "submit_btn": "Submit Question",
             "clear_btn": "Clear Conversation",
-            "lang_btn": "Badilisha lugha kuwa Kiswahili",
+            "toggle_label": "Language/Lugha:",
             "error_no_image": "Please upload an image",
             "error_processing": "Error processing image"
         }
@@ -410,12 +430,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     # Store the current disease for context
     current_disease = gr.State("")
    
-    # Add prominent language switch button at the top
-    with gr.Row():
-        lang_btn = gr.Button(
-            text_dict["swahili"]["lang_btn"], 
-            variant="secondary",
-            scale=1
+    # Add language toggle at the top
+    with gr.Row(elem_classes="language-toggle-container"):
+        gr.Markdown(text_dict["swahili"]["toggle_label"], elem_classes="toggle-label")
+        language_toggle = gr.Radio(
+            ["Swahili", "English"], 
+            value="Swahili", 
+            label="", 
+            interactive=True,
+            elem_id="language-toggle"
         )
    
     # Main identification section
@@ -447,13 +470,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         follow_up_btn = gr.Button(text_dict["swahili"]["submit_btn"], variant="primary")
         clear_btn = gr.Button(text_dict["swahili"]["clear_btn"])
    
-    # Function to switch language
-    def switch_language(current_lang):
-        new_lang = "english" if current_lang == "swahili" else "swahili"
+    # Function to switch language based on toggle
+    def switch_language(selected_language, current_lang):
+        # Map the radio value to our language code
+        new_lang = "swahili" if selected_language == "Swahili" else "english"
+        
+        # If language hasn't changed, don't update anything
+        if new_lang == current_lang:
+            return [current_lang] + [gr.update() for _ in range(11)]
         
-        # Debug message to verify the language switch is happening
+        # Debug message to verify the language switch
         print(f"Switching language from {current_lang} to {new_lang}")
         
+        # Return updated values
         return (
             new_lang,
             text_dict[new_lang]["title"],
@@ -466,8 +495,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             text_dict[new_lang]["question_label"],
             text_dict[new_lang]["question_placeholder"],
             text_dict[new_lang]["submit_btn"],
-            text_dict[new_lang]["clear_btn"],
-            text_dict[new_lang]["lang_btn"]
+            text_dict[new_lang]["clear_btn"]
         )
    
     # Set up event handlers
@@ -507,10 +535,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     def clear_conversation_history():
         return ""
    
-    # Connect the language switch button
-    lang_btn.click(
+    # Connect the language toggle
+    language_toggle.change(
         switch_language,
-        inputs=[current_language],
+        inputs=[language_toggle, current_language],
         outputs=[
             current_language,
             title_md,
@@ -523,8 +551,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             follow_up_input,
             follow_up_input,
             follow_up_btn,
-            clear_btn,
-            lang_btn
+            clear_btn
         ]
     )