babelAI commited on
Commit
53d95a9
·
verified ·
1 Parent(s): 96b9b91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -34
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
- # From here down is all the Streamlit UI.
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
- #if 'sessionMessages' not in st.session_state:
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= input)
43
  return input_text
44
 
 
 
45
 
46
- chat = ChatOpenAI(temperature=0)
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)