seawolf2357 commited on
Commit
e62103a
ยท
verified ยท
1 Parent(s): db9828e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -20
app.py CHANGED
@@ -20,14 +20,9 @@ def get_system_prompt():
20
  def chatbot_interface():
21
  st.title("AI ์–ด์‹œ์Šคํ„ดํŠธ์™€ ๋Œ€ํ™”ํ•˜๊ธฐ")
22
 
23
- # ๋ชจ๋ธ ์„ ํƒ
24
- model_options = [
25
- "claude-3-5-sonnet-20241022"
26
- ]
27
- model_selected = st.sidebar.selectbox("ํด๋กœ๋“œ ๋ชจ๋ธ ์„ ํƒ:", model_options)
28
-
29
- if "ai_model" not in st.session_state or st.session_state["ai_model"] != model_selected:
30
- st.session_state["ai_model"] = model_selected
31
 
32
  # ์„ธ์…˜ ์ƒํƒœ ์ดˆ๊ธฐํ™”
33
  if "messages" not in st.session_state:
@@ -55,14 +50,11 @@ def chatbot_interface():
55
  message_placeholder = st.empty()
56
  full_response = ""
57
 
58
- # ์‹œ์Šคํ…œ ํ”„๋กฌํ”„ํŠธ๋ฅผ ํฌํ•จํ•œ ๋ฉ”์‹œ์ง€ ์ค€๋น„
59
- messages = [{"role": "system", "content": get_system_prompt()}]
60
- messages.extend([{"role": m["role"], "content": m["content"]} for m in st.session_state.messages])
61
-
62
  # Claude API ํ˜ธ์ถœ
63
  with client.messages.stream(
64
  max_tokens=1024,
65
- messages=messages,
 
66
  model=st.session_state["ai_model"]
67
  ) as stream:
68
  for text in stream.text_stream:
@@ -84,13 +76,6 @@ def chatbot_interface():
84
  )
85
 
86
  def main():
87
- # ๋กœ๊ณ  ์ด๋ฏธ์ง€ ํ‘œ์‹œ
88
- logo_path = "logo.png"
89
- st.sidebar.image(logo_path, width=150)
90
-
91
- logo_path = "logo_transparent_background.png"
92
- st.sidebar.image(logo_path, width=250)
93
-
94
  chatbot_interface()
95
 
96
  if __name__ == "__main__":
 
20
  def chatbot_interface():
21
  st.title("AI ์–ด์‹œ์Šคํ„ดํŠธ์™€ ๋Œ€ํ™”ํ•˜๊ธฐ")
22
 
23
+ # ๋ชจ๋ธ ๊ณ ์ • ์„ค์ • (์„ ํƒ ๋ฐ•์Šค ์ œ๊ฑฐ)
24
+ if "ai_model" not in st.session_state:
25
+ st.session_state["ai_model"] = "claude-3-5-sonnet-20241022"
 
 
 
 
 
26
 
27
  # ์„ธ์…˜ ์ƒํƒœ ์ดˆ๊ธฐํ™”
28
  if "messages" not in st.session_state:
 
50
  message_placeholder = st.empty()
51
  full_response = ""
52
 
 
 
 
 
53
  # Claude API ํ˜ธ์ถœ
54
  with client.messages.stream(
55
  max_tokens=1024,
56
+ system=get_system_prompt(),
57
+ messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
58
  model=st.session_state["ai_model"]
59
  ) as stream:
60
  for text in stream.text_stream:
 
76
  )
77
 
78
  def main():
 
 
 
 
 
 
 
79
  chatbot_interface()
80
 
81
  if __name__ == "__main__":