Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 
 # Import modules from other files
 from chatbot import chatbot, model_inference, BOT_AVATAR, EXAMPLES, model_selector, decoding_strategy, temperature, max_new_tokens, repetition_penalty, top_p
-from voice_chat import respond
+from voice_chat import respond, model, transcribe, search
 from live_chat import videochat
 
 # Define Gradio theme
@@ -76,10 +76,10 @@ with gr.Blocks(
 # Voice chat block
 with gr.Blocks() as voice:
     with gr.Row():
-        input = gr.Audio(label="Voice Chat (BETA)", sources="microphone", type="filepath", waveform_options=False)
-        output = gr.Audio(label="JARVIS", type="filepath", interactive=False, autoplay=True, elem_classes="audio")
-        gr.Interface(fn=respond, inputs=[input], outputs=[output])
-        input.stop_recording(fn=respond, inputs=[input], outputs=[output])
+        web_search = gr.Checkbox(label="Web Search", value=False)
+        input = gr.Audio(label="Voice Chat", sources="microphone")
+        output = gr.Audio(label="AI",autoplay=True)
+        gr.Interface(fn=respond, inputs=[input, web_search], outputs=[output], live=True, batch=True, max_batch_size=20, delete_cache=(60,60))
         
 
 # Live chat block