Update app.py

app.py

@@ -166,19 +166,19 @@ def translate_speech(audio_file, target_language):
 def clear_history():
     global conversation_history
     conversation_history = []
-    return None, None
+    return None, None, None, None
 
-def update_description(tab):
-    if tab == "Voice Assistant":
-        return "# <center><b>Hello, I am Optimus Prime your personal AI voice assistant</b></center>"
-    elif tab == "Speech Translation":
-        return "# <center><b>Hear how you sound in another language</b></center>"
+def voice_assistant_tab():
+    return "# <center><b>Hello, I am Optimus Prime your personal AI voice assistant</b></center>"
+
+def speech_translation_tab():
+    return "# <center><b>Hear how you sound in another language</b></center>"
 
 with gr.Blocks(css="style.css") as demo:
     description = gr.Markdown("# <center><b>Hello, I am Optimus Prime your personal AI voice assistant</b></center>")
     
     with gr.Tabs() as tabs:
-        with gr.TabItem("Voice Assistant"):
+        with gr.TabItem("Voice Assistant") as voice_assistant:
             select = gr.Dropdown([
                 'Llama 3B Service',
                 'Mixtral 8x7B',
@@ -210,7 +210,7 @@ with gr.Blocks(css="style.css") as demo:
                 live=True
             )
         
-        with gr.TabItem("Speech Translation"):
+        with gr.TabItem("Speech Translation") as speech_translation:
             input_audio = gr.Audio(label="User", sources="microphone", type="filepath", waveform_options=False)
             target_lang = gr.Dropdown(
                 choices=list(LANGUAGE_CODES.keys()),
@@ -233,11 +233,12 @@ with gr.Blocks(css="style.css") as demo:
     clear_button.click(
         fn=clear_history,
         inputs=[],
-        outputs=[input, output],
+        outputs=[input, output, input_audio, output_audio],
         api_name="clear"
     )
 
-    tabs.change(fn=update_description, inputs=[tabs], outputs=[description])
+    voice_assistant.select(fn=voice_assistant_tab, inputs=None, outputs=description)
+    speech_translation.select(fn=speech_translation_tab, inputs=None, outputs=description)
 
 if __name__ == "__main__":
     demo.queue(max_size=200).launch()