cochi1706 commited on
Commit
a4a178e
·
1 Parent(s): 84a0118

Refactor message handling in response function to support both string and dictionary inputs. Added validation for empty messages to improve user experience by providing a clear error message when no input is received.

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -37,7 +37,18 @@ def respond(
37
  """
38
  Tạo phản hồi từ model coding assistant sử dụng pipeline
39
  """
40
- generated = text_generator(message, num_return_sequences=1)
 
 
 
 
 
 
 
 
 
 
 
41
 
42
  # Lấy câu trả lời từ kết quả
43
  answer = generated[0]['generated_text']
 
37
  """
38
  Tạo phản hồi từ model coding assistant sử dụng pipeline
39
  """
40
+ # Xử message - có thể là dict hoặc string
41
+ if isinstance(message, dict):
42
+ user_message = message.get("content", "")
43
+ else:
44
+ user_message = str(message) if message else ""
45
+
46
+ # Nếu không có message, trả về thông báo lỗi
47
+ if not user_message:
48
+ return "Xin lỗi, tôi không nhận được câu hỏi của bạn."
49
+
50
+ # Sử dụng pipeline để generate text
51
+ generated = text_generator(user_message, num_return_sequences=1)
52
 
53
  # Lấy câu trả lời từ kết quả
54
  answer = generated[0]['generated_text']