Add language detection

app.py

@@ -27,6 +27,7 @@ with (gr.Blocks() as app):
                     radio = gr.Radio(
                         choices=['small', 'big'], value='small', label='GPU', info='Select the size of GPU'
                     )
+                    language = gr.Checkbox(label='Predict language')
 
             with gr.Column():
                 with gr.Row():
@@ -64,13 +65,13 @@ with (gr.Blocks() as app):
 
     button_text.click(
         get_answer_text,
-        [text, chat, user_id, checkbox, radio],
+        [text, chat, user_id, checkbox, radio, language],
         [video, output_audio, chat, text]
     )
 
     button_audio.click(
         get_answer_audio,
-        [audio, chat, user_id, checkbox, radio],
+        [audio, chat, user_id, checkbox, radio, language],
         [video, output_audio, chat, audio]
     )
 

functions.py

@@ -69,6 +69,7 @@ def init_chatbot(chatbot: list[tuple[str, str]]):
     # Call API with the following json
     inputs = {"inputs": {
         "date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        'language': 'es',
         'get_video': True
     }}
     output = _query(inputs, 'small')
@@ -80,21 +81,25 @@ def init_chatbot(chatbot: list[tuple[str, str]]):
 
 
 def get_answer_text(
-        question: str, chatbot: list[tuple[str, str]], user_id: str, checkbox: bool, size_gpu: str
+        question: str, chatbot: list[tuple[str, str]], user_id: str, checkbox: bool, size_gpu: str,
+        detect_language: bool
 ):
     """
     Gets the answer of the chatbot
     """
+    get_language = None if detect_language is False else True
+
     # Create json and send it to the API
     inputs = {'inputs': {
-        'text': question, 'user_id': user_id, 'get_video': checkbox
+        'text': question, 'user_id': user_id, 'get_video': checkbox, 'get_language': get_language
     }}
     output = _query(inputs, size_gpu)
     return _update_elements(question, chatbot, output, checkbox, '')
 
 
 def get_answer_audio(
-        audio_path, chatbot: list[tuple[str, str]], user_id: str, checkbox: bool, size_gpu: str
+        audio_path, chatbot: list[tuple[str, str]], user_id: str, checkbox: bool, size_gpu: str,
+        detect_language: bool
 ):
     """
     Gets the answer of the chatbot
@@ -104,9 +109,12 @@ def get_answer_audio(
         audio_data = audio_file.read()
     encoded_audio = base64.b64encode(audio_data).decode('utf-8')
 
+    get_language = None if detect_language is False else True
+
     # Create json and send it to the API
     inputs = {'inputs': {
-        'is_audio': True, 'audio': encoded_audio, 'user_id': user_id, 'get_video': checkbox
+        'is_audio': True, 'audio': encoded_audio, 'user_id': user_id, 'get_video': checkbox,
+        'get_language': get_language
     }}
     output = _query(inputs, size_gpu)