Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,59 +1,36 @@
|
|
1 |
-
|
2 |
import streamlit as st
|
3 |
-
|
4 |
-
|
5 |
from langchain_openai import ChatOpenAI
|
6 |
-
from langchain.schema import
|
7 |
-
AIMessage,
|
8 |
-
HumanMessage,
|
9 |
-
SystemMessage
|
10 |
-
)
|
11 |
|
12 |
-
#
|
13 |
st.set_page_config(page_title="데이터스트림즈 챗 모델", page_icon=":robot:")
|
14 |
st.header("안녕하세요. 저는 데이터스트림즈에서 제작한 고혈압 전문 상담 AI 대화 봇이에요. 저는 말씀하시는 것을 기억하니 계속해서 질문해주시면 되요.")
|
15 |
|
16 |
-
|
17 |
-
|
18 |
if "sessionMessages" not in st.session_state:
|
19 |
st.session_state.sessionMessages = [
|
20 |
SystemMessage(content="You are a renowned doctor specializing in hypertension.")
|
21 |
]
|
22 |
|
23 |
-
#
|
24 |
-
# st.session_state.sessionMessages = []
|
25 |
-
|
26 |
-
# Now you can append to it safely
|
27 |
-
#st.session_state.sessionMessages.append(HumanMessage(content=question))
|
28 |
-
|
29 |
-
|
30 |
def load_answer(question):
|
31 |
-
|
32 |
-
st.session_state.sessinMessages.append(HumanMessage(content=question))
|
33 |
-
|
34 |
assistant_answer = chat(st.session_state.sessionMessages)
|
35 |
-
|
36 |
st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
|
37 |
-
|
38 |
return assistant_answer.content
|
39 |
|
40 |
-
|
41 |
def get_text():
|
42 |
-
input_text = st.text_input("상담자: ", key=
|
43 |
return input_text
|
44 |
|
|
|
|
|
45 |
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
user_input=get_text()
|
52 |
submit = st.button('Generate')
|
53 |
|
54 |
if submit:
|
55 |
-
|
56 |
response = load_answer(user_input)
|
57 |
st.subheader("Answer:")
|
58 |
-
|
59 |
-
st.write(response,key=1)
|
|
|
|
|
1 |
import streamlit as st
|
|
|
|
|
2 |
from langchain_openai import ChatOpenAI
|
3 |
+
from langchain.schema import AIMessage, HumanMessage, SystemMessage
|
|
|
|
|
|
|
|
|
4 |
|
5 |
+
# Streamlit UI 설정
|
6 |
st.set_page_config(page_title="데이터스트림즈 챗 모델", page_icon=":robot:")
|
7 |
st.header("안녕하세요. 저는 데이터스트림즈에서 제작한 고혈압 전문 상담 AI 대화 봇이에요. 저는 말씀하시는 것을 기억하니 계속해서 질문해주시면 되요.")
|
8 |
|
9 |
+
# 세션 메시지 초기화
|
|
|
10 |
if "sessionMessages" not in st.session_state:
|
11 |
st.session_state.sessionMessages = [
|
12 |
SystemMessage(content="You are a renowned doctor specializing in hypertension.")
|
13 |
]
|
14 |
|
15 |
+
# 대답 로드 함수
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
def load_answer(question):
|
17 |
+
st.session_state.sessionMessages.append(HumanMessage(content=question))
|
|
|
|
|
18 |
assistant_answer = chat(st.session_state.sessionMessages)
|
|
|
19 |
st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
|
|
|
20 |
return assistant_answer.content
|
21 |
|
22 |
+
# 사용자 입력 받기
|
23 |
def get_text():
|
24 |
+
input_text = st.text_input("상담자: ", key="input")
|
25 |
return input_text
|
26 |
|
27 |
+
# ChatOpenAI 객체 생성
|
28 |
+
chat = ChatOpenAI(temperature=0.7)
|
29 |
|
30 |
+
user_input = get_text()
|
|
|
|
|
|
|
|
|
|
|
31 |
submit = st.button('Generate')
|
32 |
|
33 |
if submit:
|
|
|
34 |
response = load_answer(user_input)
|
35 |
st.subheader("Answer:")
|
36 |
+
st.write(response, key=1)
|
|