Update app.py

app.py

@@ -4,6 +4,8 @@ from langchain.chat_models import ChatOpenAI
 from langchain.chains import LLMChain
 from langchain.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate
 import hashlib
+from gtts import gTTS
+import tempfile
 
 # Set up OpenAI API key
 os.environ["OPENAI_API_KEY"] = os.environ.get("OPENAI_API_KEY")
@@ -65,6 +67,12 @@ def respond(message, history, stage):
     
     return response, new_stage
 
+def text_to_speech(text):
+    tts = gTTS(text=text, lang='en')
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as fp:
+        tts.save(fp.name)
+        return fp.name
+
 css = """
 .chatbot-header {
     display: flex;
@@ -107,10 +115,9 @@ with gr.Blocks(css=css) as demo:
         bot_message, new_stage = respond(history[-1][0], history[:-1], stage)
         history[-1][1] = bot_message
         
-        # Generate a unique, short filename for the audio
-        filename = hashlib.md5(bot_message.encode()).hexdigest()[:10] + ".wav"
+        audio_file = text_to_speech(bot_message)
         
-        return history, new_stage, (filename, bot_message)
+        return history, new_stage, audio_file
 
     msg.submit(user, [msg, chatbot, state], [msg, chatbot, state], queue=False).then(
         bot, [chatbot, state], [chatbot, state, audio_output]