Update app.py

app.py

@@ -5,14 +5,7 @@ Created on Sun Oct 01 10:49:43 2023
 
 import streamlit as st
 import sentencepiece
-from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
-
-model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/AI_Chaperone").to("cpu")
-tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/AI_Chaperone")
-
-pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
-
-greetings=["Hello!","I am AI Chaperone. Your special virtual assistant for your needs.", "Feel free to ask me anything. I will do what I can."]
+from interaction import generate_response, audio_response
 
 st.set_page_config(page_title='AI Chaperone', layout='wide')
 
@@ -39,13 +32,13 @@ if prompt:=st.chat_input("Enter your query"):
         'role':'user',
         'content': prompt})
 
-    out=pipe(prompt)
-    response=out[0]['generated_text']
-    
-    # response = f"Analysis: {response}"
-    
+    response=generate_response(prompt, st.session_state.messages)    
     with st.chat_message("assistant"):
         st.markdown(response)
+
+    audio_tag = audio_response(response)
+
+    st.markdown(audio_tag, unsafe_allow_html=True)
         
     st.session_state.messages.append({
         'role':'assistant',