Add new API and more videos

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.mp4 filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -3,5 +3,4 @@ __pycache__/
 
 data.json
 audio.wav
-video.mp4
-infinite_loop.mp4
+video.mp4

app.py

@@ -9,15 +9,16 @@ with (gr.Blocks() as app):
     with gr.Tab('Test Chats'):
         with gr.Row() as select_author:
             author = gr.Dropdown(authors, value=authors[0], label='Author', interactive=True)
+            language = gr.Dropdown(
+                value='Español', choices=['Español', 'English', 'Português'], label='Language'
+            )
             chat_btn = gr.Button(value='Start chat')
 
         # ------------------------------------- Chat -------------------------------------------
         with gr.Row(visible=False) as chatbot:
             with gr.Column():
                 with gr.Row():
-                    video = gr.Video(
-                        interactive=False, label='Video', autoplay=True, value='infinite_loop.mp4'
-                    )
+                    video = gr.Video(interactive=False, label='Video', autoplay=True)
                 with gr.Row():
                     output_audio = gr.Audio(interactive=False, label='Audio', autoplay=True)
                 with gr.Row():
@@ -25,9 +26,11 @@ with (gr.Blocks() as app):
                         label='Get video', info='Remember that this has a cost'
                     )
                     radio = gr.Radio(
-                        choices=['small', 'big'], value='small', label='GPU', info='Select the size of GPU'
+                        choices=['small', 'big'], value='small',
+                        label='GPU', info='Select the size of GPU (at the moment they are the same)',
+                        visible=False
                     )
-                    language = gr.Checkbox(label='Predict language')
+                    predict_language = gr.Checkbox(label='Predict language')
 
             with gr.Column():
                 with gr.Row():
@@ -60,18 +63,24 @@ with (gr.Blocks() as app):
     ).then(
         make_visible, None, [chatbot, scores_row]
     ).then(
-        init_chatbot, [chat], [video, chat, user_id]
+        init_chatbot, [chat, language], [video, chat, user_id]
     )
-
+    import time
     button_text.click(
+        play_searching, language, video
+    ).then(
+        lambda : time.sleep(5), None, None
+    ).then(
         get_answer_text,
-        [text, chat, user_id, checkbox, radio, language],
+        [text, chat, user_id, checkbox, radio, predict_language],
         [video, output_audio, chat, text]
     )
 
     button_audio.click(
+        play_searching, language, video
+    ).then(
         get_answer_audio,
-        [audio, chat, user_id, checkbox, radio, language],
+        [audio, chat, user_id, checkbox, radio, predict_language],
         [video, output_audio, chat, audio]
     )
 
@@ -82,7 +91,7 @@ with (gr.Blocks() as app):
     )
 
     video.end(
-        lambda: 'infinite_loop.mp4', None, video
+        lambda: 'videos/waiting.mp4', None, video
     )
 
     output_audio.stop(

functions.py

@@ -35,6 +35,19 @@ def make_visible():
     return gr.Row.update(visible=True), gr.Row.update(visible=True)
 
 
+def play_searching(language: str):
+    """
+    """
+    if language == 'Español':
+        l = 'es'
+    elif language == 'English':
+        l = 'en'
+    else:
+        l = 'pt'
+
+    return f'videos/searching_{l}.mp4'
+
+
 def _query(payload, size_gpu: str):
     """
     Returns the json from a post request. It is done to the BellaAPI
@@ -61,15 +74,23 @@ def _download_media(url: str, type_media: str) -> None:
         shutil.copyfileobj(r.raw, f)
 
 
-def init_chatbot(chatbot: list[tuple[str, str]]):
+def init_chatbot(chatbot: list[tuple[str, str]], language: str):
     """
     Returns a greeting video, with its transcription and the user_id that
     will be used later in the other requests
     """
+    # Select language
+    if language == 'Español':
+        l = 'es'
+    elif language == 'English':
+        l = 'en'
+    else:
+        l = 'pt'
+
     # Call API with the following json
     inputs = {"inputs": {
         "date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-        'language': 'es',
+        'language': l,
         'get_video': True
     }}
     output = _query(inputs, 'small')
@@ -135,7 +156,7 @@ def _update_elements(question, chatbot, output, checkbox, clean):
 
     else:
         _download_media(link_media, 'audio')
-        return 'infinite_loop.mp4', 'audio.wav', chatbot, clean
+        return 'videos/waiting.mp4', 'audio.wav', chatbot, clean
 
 
 def save_scores(author: gr.Dropdown, history: gr.Chatbot, opinion: gr.Textbox, *score_values):

videos/searching_en.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4e19acf4521b44bb6b2f6024a3b01c331a75aadc65b9b7b8663013211b2ec850
+size 657344

videos/searching_es.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c9b7c0f9f5f2e55e4c707bf89f4a7786c0171305bda2f168bb8e51b7562feae6
+size 759678

videos/searching_pt.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a7760a516d7da9e1e31f2502613c636a0ed33ff2567d19f5f40168a582b6ffc6
+size 1077705

videos/waiting.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7fb959d7ae5fb25116b403293a2314c9ac96d4ec10f6996ce0f40ebfaef8b22
+size 495429