audio added

app.py

@@ -3,7 +3,7 @@ from datetime import datetime
 
 import streamlit as st
 
-from helper import ChatBot, current_year, invoke_search_api
+from helper import ChatBot, current_year, invoke_search_api, save_to_audio
 
 # API KEY
 API_KEY = os.environ["API_KEY"]
@@ -75,7 +75,6 @@ with st.sidebar:
         (
             "New York, New York, United States",
             "Los Angeles, California, United States",
-            "Seattle, Washington, United States",
             "Chicago, Illinois, United States",
             "Houston, Texas, United States",
             "Phoenix, Arizona, United States",
@@ -91,6 +90,7 @@ with st.sidebar:
             "Charlotte, North Carolina, United States",
             "San Francisco, California, United States",
             "Indianapolis, Indiana, United States",
+            "Seattle, Washington, United States",
             "Denver, Colorado, United States",
             "Washington, D.C., United States",
             "Boston, Massachusetts, United States",
@@ -115,7 +115,6 @@ with st.sidebar:
             "Colorado Springs, Colorado, United States",
         ),
     )
-    do_not_use_internet = st.checkbox("Do not user internet. Chat only.")
 
     # Add a button to clear the session state
     if st.button("Clear Session"):
@@ -166,20 +165,14 @@ if prompt := st.chat_input(
     # API Call
     query = src_key_word + " " + prompt
     try:
-        if do_not_use_internet:
-            response = f"""
-            There is no search results here. Treat is as a standard conversation.
-            """
-            ref_table_string = """### No search triggered for this run."""
-        else:
-            md_data = invoke_search_api(
-                api_key=API_KEY, query=query, location=location, num=num
-            )
-            md_data = md_data["data"]
-            response = f"""
-            Please see search results below: \n{md_data}
-            """
-            ref_table_string = response
+        md_data = invoke_search_api(
+            api_key=API_KEY, query=query, location=location, num=num
+        )
+        md_data = md_data["data"]
+        response = f"""
+        Please see search results below: \n{md_data}
+        """
+        ref_table_string = response
 
         # API Call
         bot = ChatBot()
@@ -196,9 +189,14 @@ if prompt := st.chat_input(
         st.warning(f"Failed to fetch data: {e}")
         response = "We are fixing the API right now. Please check back later."
 
+    # Save audio
+    if response:
+        save_to_audio(response)
+
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
         st.markdown(response, unsafe_allow_html=True)
+        st.audio("output.mp3", format="audio/mpeg", loop=True)
         with st.expander("See references:", expanded=False):
             st.markdown(ref_table_string)
 

helper.py

@@ -6,6 +6,7 @@ from typing import Dict, List
 import requests
 import streamlit as st
 from openai import OpenAI
+from gtts import gTTS
 
 api_key = os.environ["API_KEY"]
 
@@ -69,6 +70,22 @@ def current_year() -> int:
     # Extract and return the current year
     return now.year
 
+# Save audio
+def save_to_audio(text: str) -> None:
+    """
+    Converts the given text to an audio file.
+
+    This function uses Google Text-to-Speech (gTTS) to convert the provided 
+    text into speech and saves it as an MP3 file named 'output.mp3'.
+
+    Parameters:
+    text (str): The text to be converted to speech.
+    """
+    # Create a gTTS object with the specified text and language ('en' for English)
+    tts = gTTS(text=text, lang='en')  # 'en' for English, change as needed
+
+    # Save the generated speech to an audio file named "output.mp3"
+    tts.save("output.mp3")
 
 class ChatBot:
     """

requirements.txt

@@ -1,4 +1,5 @@
 google-search-results
 openai
 streamlit
-serpapi
+serpapi
+gTTS