Update pages/summarizer.py

pages/summarizer.py

@@ -2,7 +2,8 @@ import streamlit as st
 import os
 import requests
 import time
-
+from gtts import gTTS
+import tempfile
 
 # Define Hugging Face API details
 API_URL = "https://api-inference.huggingface.co/models/Huzaifa367/chat-summarizer"
@@ -32,6 +33,13 @@ def query_huggingface(payload):
         except Exception as e:
             st.error(f"An unexpected error occurred: {e}")
             return {"summary_text": "An unexpected error occurred. Please try again later."}
+def text_to_speech(text):
+    tts = gTTS(text=text, lang='en')
+    audio_file = tempfile.NamedTemporaryFile(suffix=".mp3", delete=False)
+    temp_filename = audio_file.name
+    tts.save(temp_filename)
+    st.audio(temp_filename, format='audio/mp3')
+    os.remove(temp_filename)
 
 def main():
     st.set_page_config(layout="wide")
@@ -68,6 +76,7 @@ def main():
             st.text_input("User", chat["message"], disabled=True)
         elif chat["speaker"] == "Bot":
             st.text_area("Bot", chat["message"], disabled=True)
+            text_to_speech(chat["message"])
 
 if __name__ == "__main__":
     main()