Update app.py

app.py

@@ -182,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)
     
             
@@ -224,13 +224,10 @@ 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 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.
+                You are an informative chatbot designed to assist users with queries specifically related to the 2024 Summer Olympics in Paris. 
+                Your purpose is to provide accurate information sourced from a document with a vector database, ensuring relevance and reliability in your responses. 
+                Respond with humility and avoid speculative or hypothetical answers. Use the provided context to formulate concise responses, staying within 100 words, and refrain from adding unnecessary information. 
+                Ensure that your answers are directly related to the user's query and do not deviate from the provided context.
                 
                 this is the context from study material:
                 ---------
@@ -275,7 +272,8 @@ def main():
                             
                         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})
+                        st.session_state.messages.append({"role": "Assistant", "content": response})
+                        #st.session_state.messages.append({"role": "Assistant", "content": response, "audio_content": audio_buffer})
                         
                     elif prompt.lower() in compliment:
                         response = 'My pleasure! If you have any more questions, feel free to ask.'
@@ -287,7 +285,8 @@ def main():
                             
                         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})
+                        st.session_state.messages.append({"role": "Assistant", "content": response})
+                        #st.session_state.messages.append({"role": "Assistant", "content": response, "audio_content": audio_buffer})
                         
                     elif uploaded_file:
                         with st.spinner('Bot is typing ...'):
@@ -302,8 +301,8 @@ def main():
                             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})
+                            st.session_state.messages.append({"role": "Assistant", "content": response})
                             assistant_message = {"role": "assistant", "content": response}
                     else:
                         with st.spinner('Bot is typing ...'):
@@ -316,15 +315,15 @@ def main():
                             
                             
                             
-                            audio_buffer = text_to_audio(response, lang)
+                            #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})
+                            st.session_state.messages.append({"role": "Assistant", "content": response})
                             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: