Spaces:
Running
Running
File size: 1,783 Bytes
9b6d204 |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import streamlit as st
from streamlit_chat import message
from dotenv import load_dotenv
from langchain.chat_models import ChatOpenAI
from langchain.schema import SystemMessage, HumanMessage,AIMessage
##################################################################################
##variable d'environement
load_dotenv()
##################################################################################
##streamlit page
def start():
st.set_page_config(page_title="heeloo")
st.header('chatbot')
st.divider()
start()
#message('how are you')
#message('how are you', is_user=True)
###################################################################################
def main():
chat = ChatOpenAI(temperature='0.5')
if "messages" not in st.session_state:
st.session_state.messages=[
SystemMessage(content="You are an assistant")
]
#create input
input_user= st.text_input("Ask Question ", placeholder ="Ask Question")
#create button
button=st.button('Ask' ,use_container_width=True)
if button:
#add message to message history
st.session_state.messages.append(HumanMessage(content=input_user))
#send message history to chat
resposne =chat(st.session_state.messages)
#display response
message(resposne, is_user=False)
#add response to message hisotry
st.session_state.messages.append(AIMessage(content=resposne.content))
#get list of messages
messages = st.session_state.get('messages',[])
#:1 start with index 1
for i, msg in enumerate(messages):
if i%2 ==0 :
message(msg.content, is_user=True,key=str(i)+'_user')
else:
message(msg.content, is_user=False,key=str(i)+'_ai')
main()
|