babelAI commited on
Commit
53d95a9
β€’
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)