anpigon commited on
Commit
1c4aaba
β€’
1 Parent(s): 8eff983

refactor: Add input validation to chat_function

Browse files
Files changed (1) hide show
  1. app.py +12 -6
app.py CHANGED
@@ -78,9 +78,18 @@ def get_model_key(label: str) -> str:
78
  return next(key for key, value in AVAILABLE_MODELS.items() if value == label)
79
 
80
 
 
 
 
 
 
81
  def chat_function(
82
  message: str, history: List[Tuple[str, str]], model_label: str
83
  ) -> Generator[str, None, None]:
 
 
 
 
84
  model_key = get_model_key(model_label)
85
  if STREAMING:
86
  response = ""
@@ -113,19 +122,16 @@ with gr.Blocks(
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],
 
78
  return next(key for key, value in AVAILABLE_MODELS.items() if value == label)
79
 
80
 
81
+ def validate_input(message: str) -> bool:
82
+ """μž…λ ₯된 λ©”μ‹œμ§€κ°€ μœ νš¨ν•œμ§€ κ²€μ‚¬ν•©λ‹ˆλ‹€."""
83
+ return bool(message.strip())
84
+
85
+
86
  def chat_function(
87
  message: str, history: List[Tuple[str, str]], model_label: str
88
  ) -> Generator[str, None, None]:
89
+ if not validate_input(message):
90
+ yield "λ©”μ‹œμ§€λ₯Ό μž…λ ₯ν•΄μ£Όμ„Έμš”."
91
+ return
92
+
93
  model_key = get_model_key(model_label)
94
  if STREAMING:
95
  response = ""
 
122
  multimodal=False,
123
  examples=[
124
  [
125
+ "쀑고차 거래λ₯Ό ν–ˆλŠ”λ° λΆˆλŸ‰μœΌλ‘œ μ°¨ μˆ˜λ¦¬μ— 500λ§Œμ›μ΄ λ“€μ—ˆμŠ΅λ‹ˆλ‹€. νŒλ§€μžμ—κ²Œ 법적 μ±…μž„μ„ 물을 수 μžˆλ‚˜μš”? λΉ„μŠ·ν•œ νŒλ‘€λ₯Ό μ†Œκ°œν•΄μ£Όμ„Έμš”.",
126
  "GPT-4o",
 
127
  ],
128
  [
129
+ "μ•½ 2천 ν‰μ˜ 농지λ₯Ό κ΅¬λ§€ν–ˆλŠ”λ°, μ•Œκ³  λ³΄λ‹ˆ 주택을 지을 수 μ—†λŠ” λ•…μ΄μ—ˆμŠ΅λ‹ˆλ‹€. 이와 μœ μ‚¬ν•œ 뢀동산 사기 κ΄€λ ¨ νŒλ‘€λ₯Ό μ•Œλ €μ£Όμ„Έμš”.",
130
  "GPT-4o",
 
131
  ],
132
  [
133
+ "지인이 μž₯λ‚œμœΌλ‘œ νœ˜λ‘λ₯Έ 칼에 νŒ”μ΄ 20cm κ°€λŸ‰ μ°”λ ΈμŠ΅λ‹ˆλ‹€. μž₯λ‚œμ΄λΌκ³  μ£Όμž₯ν•˜λŠ”λ°, 이와 μœ μ‚¬ν•œ 상해 κ΄€λ ¨ νŒλ‘€λ₯Ό μ•Œλ €μ£Όμ„Έμš”.",
134
  "GPT-4o",
 
135
  ],
136
  ],
137
  additional_inputs=[model_dropdown],