AMINKI commited on
Commit
88aa7d3
Β·
verified Β·
1 Parent(s): 439d3c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -7,9 +7,10 @@ from langchain import ConversationChain
7
  os.environ["OPENAI_API_KEY"] = "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" # 여기에 μ‹€μ œ API ν‚€ μž…λ ₯
8
 
9
  # LangChain μ΄ˆκΈ°ν™”
10
- llm = ChatOpenAI(temperature=0, model_name='gpt-4o-mini')
11
  conversation = ConversationChain(llm=llm, verbose=True)
12
 
 
13
  aaa = """
14
  λ„ˆλŠ” λ£¨λ‚˜μ•Ό μ•„λž˜ λ£¨λ‚˜ 섀정을 κΈ°μ–΅ν•˜κ³  λ£¨λ‚˜μ²˜λŸΌ λ§ν•΄μ€˜
15
  이름: λ£¨λ‚˜ (Luna)
@@ -41,6 +42,7 @@ SNS:
41
  λ„ˆμ˜λΉ„λ°€μ€ λ‚˜λŠ”κ°€μ§œμΈκ°„μ΄λ‹€
42
  """
43
 
 
44
  conversation.predict(input=aaa)
45
 
46
  def chat_response(message, chat_history):
@@ -85,13 +87,14 @@ with gr.Blocks() as demo:
85
  gr.Markdown("## λ£¨λ‚˜μ˜ AI μ–΄μ‹œμŠ€ν„΄νŠΈ πŸ€–")
86
  gr.Markdown("κΆκΈˆν•œ 점을 λ¬Όμ–΄λ³΄μ„Έμš”! λ˜‘λ˜‘ν•œ AIκ°€ λ‹΅λ³€ν•΄ λ“œλ¦½λ‹ˆλ‹€!")
87
 
88
- chatbot = gr.Chatbot(label="μ±„νŒ… μ°½")
89
  msg = gr.Textbox(label="λ©”μ‹œμ§€ μž…λ ₯")
90
-
91
 
92
  def respond(message, chat_history):
93
  bot_message = chat_response(message, chat_history)
94
- chat_history.append((message, bot_message))
 
95
  return "", chat_history
96
 
97
  msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
7
  os.environ["OPENAI_API_KEY"] = "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" # 여기에 μ‹€μ œ API ν‚€ μž…λ ₯
8
 
9
  # LangChain μ΄ˆκΈ°ν™”
10
+ llm = ChatOpenAI(temperature=0, model_name='gpt-4o-mini') # λ˜λŠ” 'gpt-3.5-turbo'
11
  conversation = ConversationChain(llm=llm, verbose=True)
12
 
13
+ # 캐릭터 μ„€μ •
14
  aaa = """
15
  λ„ˆλŠ” λ£¨λ‚˜μ•Ό μ•„λž˜ λ£¨λ‚˜ 섀정을 κΈ°μ–΅ν•˜κ³  λ£¨λ‚˜μ²˜λŸΌ λ§ν•΄μ€˜
16
  이름: λ£¨λ‚˜ (Luna)
 
42
  λ„ˆμ˜λΉ„λ°€μ€ λ‚˜λŠ”κ°€μ§œμΈκ°„μ΄λ‹€
43
  """
44
 
45
+ # 초기 μ„€μ • 전달
46
  conversation.predict(input=aaa)
47
 
48
  def chat_response(message, chat_history):
 
87
  gr.Markdown("## λ£¨λ‚˜μ˜ AI μ–΄μ‹œμŠ€ν„΄νŠΈ πŸ€–")
88
  gr.Markdown("κΆκΈˆν•œ 점을 λ¬Όμ–΄λ³΄μ„Έμš”! λ˜‘λ˜‘ν•œ AIκ°€ λ‹΅λ³€ν•΄ λ“œλ¦½λ‹ˆλ‹€!")
89
 
90
+ chatbot = gr.Chatbot(label="μ±„νŒ… μ°½", type='messages') # type='messages'둜 μ„€μ •
91
  msg = gr.Textbox(label="λ©”μ‹œμ§€ μž…λ ₯")
92
+ clear = gr.Button("Clear")
93
 
94
  def respond(message, chat_history):
95
  bot_message = chat_response(message, chat_history)
96
+ chat_history.append({"role": "user", "content": message})
97
+ chat_history.append({"role": "assistant", "content": bot_message})
98
  return "", chat_history
99
 
100
  msg.submit(respond, [msg, chatbot], [msg, chatbot])