Update app.py

app.py

@@ -17,6 +17,7 @@ from huggingface_hub import InferenceClient
 # Bhashini API translation function
 api_key = getenv('API_KEY')
 user_id = getenv('USER_ID')
+global history
 def bhashini_translate(text: str, from_code: str = "en", to_code: str = "hi") -> dict:
     """Translates text from source language to target language using the Bhashini API."""
     
@@ -98,9 +99,10 @@ def add_text(history, text):
     return history, gr.Textbox(value="", interactive=False)
 
 def bot(history, cross_encoder):
+    global history
     top_rerank = 25
     top_k_rank = 20
-    query = history[-1][0]
+    query = history[-1][0] if history else ''
 
     if not query:
         gr.Warning("Please submit a non-empty string as a prompt")
@@ -156,7 +158,10 @@ def bot(history, cross_encoder):
             history[-1][1] = character
             yield history, prompt_html
 
-def translate_text(response_text, selected_language):
+#def translate_text(response_text, selected_language):
+    
+def translate_text(selected_language):
+    global history
     iso_language_codes = {
         "Hindi": "hi",
         "Gom": "gom",
@@ -183,8 +188,37 @@ def translate_text(response_text, selected_language):
     }
     
     to_code = iso_language_codes[selected_language]
+    response_text = history[-1][1] if history else ''
     translation = bhashini_translate(response_text, to_code=to_code)
     return translation['translated_content']
+    # iso_language_codes = {
+    #     "Hindi": "hi",
+    #     "Gom": "gom",
+    #     "Kannada": "kn",
+    #     "Dogri": "doi",
+    #     "Bodo": "brx",
+    #     "Urdu": "ur",
+    #     "Tamil": "ta",
+    #     "Kashmiri": "ks",
+    #     "Assamese": "as",
+    #     "Bengali": "bn",
+    #     "Marathi": "mr",
+    #     "Sindhi": "sd",
+    #     "Maithili": "mai",
+    #     "Punjabi": "pa",
+    #     "Malayalam": "ml",
+    #     "Manipuri": "mni",
+    #     "Telugu": "te",
+    #     "Sanskrit": "sa",
+    #     "Nepali": "ne",
+    #     "Santali": "sat",
+    #     "Gujarati": "gu",
+    #     "Odia": "or"
+    # }
+    
+    # to_code = iso_language_codes[selected_language]
+    # translation = bhashini_translate(response_text, to_code=to_code)
+    # return translation['translated_content']
 
 # Gradio interface
 with gr.Blocks(theme='NoCrypt/miku') as CHATBOT:
@@ -228,28 +262,25 @@ with gr.Blocks(theme='NoCrypt/miku') as CHATBOT:
     )
     
     prompt_html = gr.HTML()
-    translated_textbox = gr.Textbox(label="Translated Response")
     
-    # Click event handler for submit button
+    translated_textbox = gr.Textbox(label="Translated Response")
+
     txt_btn.click(
         add_text, [chatbot, txt], [chatbot, txt], queue=False
     ).then(
-        bot, [chatbot, cross_encoder], [chatbot, prompt_html]
+        bot, [cross_encoder], [chatbot, prompt_html]
     ).then(
-        lambda history: history[-1][1], [chatbot], translated_textbox  # Extract chatbot output
-    ).then(
-        translate_text, [txt, language_dropdown], translated_textbox
+        translate_text, [language_dropdown], translated_textbox
     )
 
-    # Submit event handler for pressing Enter key
     txt.submit(
         add_text, [chatbot, txt], [chatbot, txt], queue=False
     ).then(
-        bot, [chatbot, cross_encoder], [chatbot, prompt_html]
-    ).then(
-        lambda history: history[-1][1], [chatbot], translated_textbox  # Extract chatbot output
+        bot, [cross_encoder], [chatbot, prompt_html]
     ).then(
-        translate_text, [txt, language_dropdown], translated_textbox
+        translate_text, [language_dropdown], translated_textbox
+    )
+
     )
     # txt_msg = txt_btn.click(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
     #     bot, [chatbot, cross_encoder], [chatbot, prompt_html]).then(