Update app.py

app.py

@@ -14,6 +14,7 @@ client = Groq(
     api_key=os.environ.get("GROQ_API_KEY"),
 )
 
+# Your existing functions remain the same
 def clean_disease_name(name):
     """Clean disease name by removing numbers and special characters, and fix formatting"""
     # Remove numbers and dots at the beginning
@@ -409,11 +410,18 @@ 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
+        )
+   
     # Main identification section
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(type="pil", label=text_dict["swahili"]["upload_label"])
-            lang_btn = gr.Button(text_dict["swahili"]["lang_btn"])
             submit_btn = gr.Button(text_dict["swahili"]["identify_btn"], variant="primary")
        
         with gr.Column(scale=2):
@@ -442,6 +450,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     # Function to switch language
     def switch_language(current_lang):
         new_lang = "english" if current_lang == "swahili" else "swahili"
+        
+        # Debug message to verify the language switch is happening
+        print(f"Switching language from {current_lang} to {new_lang}")
+        
         return (
             new_lang,
             text_dict[new_lang]["title"],
@@ -464,6 +476,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             return None, text_dict[lang]["error_no_image"], "", ""
        
         try:
+            # Print debug message to verify language
+            print(f"Processing image with language: {lang}")
+            
             pred_results, info, clean_disease_name = predict_and_get_info(img, lang)
             return pred_results, info, clean_disease_name, ""
         except Exception as e: