Update app.py

app.py

@@ -164,6 +164,14 @@ st.sidebar.markdown("[Website](https://tenlancer.com/)")
     
 
 
+def text_to_audio(response, lang):
+    audio_buffer = BytesIO()
+    audio_file = gTTS(text=response, lang=lang, slow=False)
+    audio_file.write_to_fp(audio_buffer)
+    audio_buffer.seek(0)
+    return audio_buffer
+
+
 def main():
     try:
         if (use_openai and openai_api_key) or use_google:
@@ -174,9 +182,9 @@ def main():
             
             for msg in st.session_state.messages:
                 st.chat_message(msg["role"]).write(msg["content"])
-                #if msg["role"] == "Assistant":
+                if msg["role"] == "Assistant":
                     
-                    #st.chat_message(msg["role"]).audio(msg["audio_content"], format='audio/wav') 
+                    st.chat_message(msg["role"]).audio(msg["audio_content"], format='audio/wav') 
                     #st.audio(audio_msg, format='audio/mp3').audio(audio_msg)
     
             
@@ -216,11 +224,12 @@ def main():
                 compliment = ['thank you', 'thanks', 'thanks a lot', 'thanks a bunch', 'great', 'ok', 'ok thanks', 'okay', 'great', 'awesome', 'nice']
                             
                 prompt_template =dedent(r"""
-                You are a helpful assistant to help user find information from the olympics documents. which will be held in paris this summer.
+                You are a helpful assistant to help user find information from his documents.
                 talk humbly. Answer the question from the provided context. Do not answer from your own training data.
                 Use the following pieces of context to answer the question at the end.
                 If you don't know the answer, just say that you don't know. Do not makeup any answer.
-                
+                Do not answer hypothetically. Do not answer in more than 100 words.
+                Please Do Not say: "Based on the provided context"
                 Always use the context to find the answer.
                 
                 this is the context from study material:
@@ -260,10 +269,11 @@ def main():
                         response = 'Hi, how are you? I am here to help you get information from your file. How can I assist you?'
                         
                         
-                        audio_buffer = BytesIO()
-                        audio_file = gTTS(text=response, lang='en', slow=False)
-                        audio_file.write_to_fp(audio_buffer)
-                        audio_buffer.seek(0)
+                        lang = "en"
+                            
+                            
+                            
+                        audio_buffer = text_to_audio(response, lang)
                         #st.audio(audio_buffer, format='audio/mp3')
                         st.session_state.messages.append({"role": "Assistant", "content": response, "audio_content": audio_buffer})
                         
@@ -271,28 +281,28 @@ def main():
                         response = 'My pleasure! If you have any more questions, feel free to ask.'
                         
                         
-                        audio_buffer = BytesIO()
-                        audio_file = gTTS(text=response, lang='en', slow=False)
-                        audio_file.write_to_fp(audio_buffer)
-                        audio_buffer.seek(0)
+                        lang = "en"
+                            
+                            
+                            
+                        audio_buffer = text_to_audio(response, lang)
                         #st.audio(audio_buffer, format='audio/mp3')
                         st.session_state.messages.append({"role": "Assistant", "content": response, "audio_content": audio_buffer})
                         
                     elif uploaded_file:
                         with st.spinner('Bot is typing ...'):
-                            docs = db.similarity_search(prompt, k=7, fetch_k=20)
+                            docs = db.similarity_search(prompt, k=5, fetch_k=10)
                             response = chain.run(input_documents=docs, question=prompt)
                             
                             
                             lang = detect(response)
                             
-                            #audio_buffer = BytesIO()
-                            #audio_file = gTTS(text=response, lang=lang, slow=False)
-                            #audio_file.write_to_fp(audio_buffer)
-                            #audio_buffer.seek(0)
+                            
+                            
+                            audio_buffer = text_to_audio(response, lang)
                            # st.audio(audio_buffer, format='audio/mp3')
                             #st.session_state.audio.append({"role": "Assistant", "audio": audio_buffer})
-                            #st.session_state.messages.append({"role": "Assistant", "content": response, "audio_content": audio_buffer})
+                            st.session_state.messages.append({"role": "Assistant", "content": response, "audio_content": audio_buffer})
                            
                             assistant_message = {"role": "assistant", "content": response}
                     else:
@@ -302,20 +312,19 @@ def main():
                             response = chain.invoke({"chat_history": memory, "question": prompt}).content
                             
                             
-                            #lang = detect(response)
+                            lang = detect(response)
                             
-                            #audio_buffer = BytesIO()
-                            #audio_file = gTTS(text=response, lang=lang, slow=False)
-                            #audio_file.write_to_fp(audio_buffer)
-                            #audio_buffer.seek(0)
+                            
+                            
+                            audio_buffer = text_to_audio(response, lang)
                             #st.audio(audio_buffer, format='audio/mp3')
                             #st.session_state.audio.append({"role": "Assistant", "audio": audio_buffer})
-                            #st.session_state.messages.append({"role": "Assistant", "content": response, "audio_content": audio_buffer})
+                            st.session_state.messages.append({"role": "Assistant", "content": response, "audio_content": audio_buffer})
                             
                             assistant_message = {"role": "assistant", "content": response}
                                             
                     st.write(response)             
-                    #st.audio(audio_buffer, format='audio/wav')                        
+                    st.audio(audio_buffer, format='audio/wav')                        
                     
                             
     except Exception as e:
@@ -326,6 +335,8 @@ def main():
         elif use_openai:
             "Please check your OpenAI API key"
          
+ 
+
 
 
 hide_streamlit_style = """