anpigon commited on
Commit
0a85bef
Β·
1 Parent(s): 03d55cf

refactor: Update app.py to improve chat interface functionality

Browse files
Files changed (1) hide show
  1. app.py +26 -24
app.py CHANGED
@@ -92,7 +92,9 @@ def chat_function(
92
  yield response
93
 
94
 
95
- with gr.Blocks() as demo:
 
 
96
  gr.Markdown("# λŒ€λ²•μ› νŒλ‘€ 상담 λ„μš°λ―Έ")
97
  gr.Markdown(
98
  "μ•ˆλ…•ν•˜μ„Έμš”! λŒ€λ²•μ› νŒλ‘€μ— κ΄€ν•œ μ§ˆλ¬Έμ— λ‹΅λ³€ν•΄λ“œλ¦¬λŠ” AI 상담 λ„μš°λ―Έμž…λ‹ˆλ‹€. νŒλ‘€ 검색, 해석, 적용 등에 λŒ€ν•΄ κΆκΈˆν•˜μ‹  점이 있으면 μ–Έμ œλ“  λ¬Όμ–΄λ³΄μ„Έμš”."
@@ -104,30 +106,30 @@ with gr.Blocks() as demo:
104
  value=list(AVAILABLE_MODELS.values())[0],
105
  )
106
 
107
- chatbot = gr.Chatbot(height="50vh")
108
- msg = gr.Textbox(label="λ©”μ‹œμ§€ μž…λ ₯")
109
- with gr.Row():
110
- clear = gr.Button("λŒ€ν™” λ‚΄μš© μ§€μš°κΈ°")
111
- send_btn = gr.Button("전솑")
112
-
113
- def user(user_message, history):
114
- return "", history + [[user_message, None]]
115
-
116
- def bot(history, model_label):
117
- model_key = get_model_key(model_label)
118
- user_message = history[-1][0]
119
- bot_message = ""
120
- for chunk in respond_stream(user_message, history[:-1], model_key):
121
- bot_message += chunk
122
- yield history[:-1] + [[user_message, bot_message]]
123
-
124
- msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
125
- bot, [chatbot, model_dropdown], chatbot
 
 
 
 
126
  )
127
- send_btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
128
- bot, [chatbot, model_dropdown], chatbot
129
- )
130
- clear.click(lambda: None, None, chatbot, queue=False)
131
 
132
  if __name__ == "__main__":
133
  demo.launch()
 
92
  yield response
93
 
94
 
95
+ with gr.Blocks(
96
+ fill_height=True,
97
+ ) as demo:
98
  gr.Markdown("# λŒ€λ²•μ› νŒλ‘€ 상담 λ„μš°λ―Έ")
99
  gr.Markdown(
100
  "μ•ˆλ…•ν•˜μ„Έμš”! λŒ€λ²•μ› νŒλ‘€μ— κ΄€ν•œ μ§ˆλ¬Έμ— λ‹΅λ³€ν•΄λ“œλ¦¬λŠ” AI 상담 λ„μš°λ―Έμž…λ‹ˆλ‹€. νŒλ‘€ 검색, 해석, 적용 등에 λŒ€ν•΄ κΆκΈˆν•˜μ‹  점이 있으면 μ–Έμ œλ“  λ¬Όμ–΄λ³΄μ„Έμš”."
 
106
  value=list(AVAILABLE_MODELS.values())[0],
107
  )
108
 
109
+ chatbot = gr.ChatInterface(
110
+ fn=chat_function,
111
+ autofocus=True,
112
+ fill_height=True,
113
+ multimodal=False,
114
+ examples=[
115
+ [
116
+ "쀑고차 거래λ₯Ό ν–ˆμ–΄. 그런데 λΆˆλŸ‰μ„ λ°›μ•„ μ°¨μˆ˜λ¦¬μ— 500λ§Œμ›μ΄ λ“€μ—ˆμ–΄. νŒλ§€μžλŠ” 사기꾼인가? λΉ„μŠ·ν•œ 사둀λ₯Ό λͺ‡ 개 μ†Œκ°œν•΄μ€˜!!",
117
+ "GPT-4o",
118
+ "gpt_4o",
119
+ ],
120
+ [
121
+ "논밭은 μ•½ 2μ²œν‰μ„ μƒ€λŠ”λ°, μ•Œκ³  λ³΄λ‹ˆ 집을 지을 수 μ—†λŠ” 땅이야. 이런 사기와 λΉ„μŠ·ν•œ κ±Έ μ•Œλ €μ€˜!",
122
+ "GPT-4o",
123
+ "gpt_4o",
124
+ ],
125
+ [
126
+ "지인이 μž₯λ‚œν•˜λ‹€κ°€ λ“€μ˜€ μžˆλŠ” 칼을 νœ˜λ‘˜λŸ¬ λ‚΄ νŒ”μ΄ 20cmκ°€λŸ‰ μžμƒμ„ μž…μ—ˆμ–΄. μžκΈ°λŠ” μž₯λ‚œμ΄λΌλŠ”λ°, λΉ„μŠ·ν•œ 사둀λ₯Ό μ•Œλ €μ€˜!",
127
+ "GPT-4o",
128
+ "gpt_4o",
129
+ ],
130
+ ],
131
+ additional_inputs=[model_dropdown],
132
  )
 
 
 
 
133
 
134
  if __name__ == "__main__":
135
  demo.launch()