File size: 1,127 Bytes
944d166
 
 
 
 
 
804707d
07a5e28
944d166
4dddcfe
944d166
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
07a5e28
944d166
804707d
07a5e28
 
 
 
944d166
 
 
804707d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
"""
Created on Sun Oct 01 10:49:43 2023
@author: Loges
"""

import streamlit as st
import sentencepiece
from interaction import generate_response, audio_response

st.set_page_config(page_title='AI Chaperone', layout='wide')

if 'messages' not in st.session_state:
    st.session_state.messages=[]

    for gr in greetings:
        st.session_state.messages.append({
            'role':'assistant',
            'content': gr})
    

st.subheader("AI Chaperone")

for message in st.session_state.messages:
    with st.chat_message(message['role']):
        st.markdown(message['content'])

if prompt:=st.chat_input("Enter your query"):
    with st.chat_message("user"):
        st.markdown(prompt)

    st.session_state.messages.append({
        'role':'user',
        'content': prompt})

    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',
        'content': response})