Update app.py

app.py

@@ -24,6 +24,8 @@ else:
     dp.mkdir(exist_ok=True)
     dataDir = '/data/'
 
+speak_file = dataDir + "speek.wav"
+
 client = OpenAI(api_key = key)
 
 def genUsageStats(do_reset=False):
@@ -174,6 +176,24 @@ def transcribe(user, pwd, fpath):
 def pause_message():
     return "Audio input is paused.  Resume or Stop as desired"
 
+def gen_output_audio(txt):
+    if len(txt) < 10:
+        txt = "This dialog is too short to mess with!"
+    response = client.audio.speech.create(model="tts-1", voice="fable", input=txt)
+    with open(speak_file, 'wb') as fp:
+        fp.write(response.content)
+    return speak_file
+
+def set_speak_button(txt):
+    vis = False
+    if len(txt) > 10:
+        vis = True
+    return gr.Button(visible=vis)
+
+def delete_speak_file():
+    if os.path.exists(speak_file):
+        os.remove(speak_file)
+
 with gr.Blocks() as demo:
     history = gr.State([])
     password = gr.State("")
@@ -196,6 +216,7 @@ with gr.Blocks() as demo:
         # gpt_chooser=gr.Radio(choices=[("GPT-3.5","gpt-3.5-turbo"),("GPT-4o","gpt-4o-mini")],
         #                      value="gpt-3.5-turbo", label="GPT Model", interactive=True)
         submit_window = gr.Button(value="Submit Prompt/Question")
+        speak_output = gr.Button(value="Speak Dialog", visible=False)
     prompt_window = gr.Textbox(label = "Prompt or Question")
     output_window = gr.Textbox(label = "Dialog")
     submit_window.click(chat, inputs=[prompt_window, user_window, password, history, output_window, model],
@@ -205,4 +226,8 @@ with gr.Blocks() as demo:
                                 outputs=[prompt_window])
     audio_widget.pause_recording(fn=pause_message, outputs=[prompt_window])
     reset_button.add(audio_widget)
+    audio_out = gr.Audio(autoplay=True, visible=False)
+    speak_output.click(gen_output_audio, output_window, audio_out)
+    output_window.change(fn=set_speak_button, inputs=output_window,outputs=speak_output)
+    demo.unload(delete_speak_file)
 demo.launch(share=True)