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)
|
|
|