Refactor response handling in chatbot application to support optional system messages and improve message processing. Added checks for empty responses to enhance user experience.

app.py

@@ -32,7 +32,7 @@ print(f"Model đã sẵn sàng! Device: {device}")
 def respond(
     message,
     history: list[dict[str, str]],
-    system_message,
+    system_message=None,
     max_tokens=None,
     temperature=None,
     top_p=None,
@@ -48,10 +48,25 @@ def respond(
     if top_p is None:
         top_p = 0.95
     
-    # Chuẩn bị prompt với chat template
-    messages = [{"role": "system", "content": system_message}]
+    # Xử lý message - có thể là dict hoặc string
+    if isinstance(message, dict):
+        user_message = message.get("content", "")
+    else:
+        user_message = str(message) if message else ""
+    
+    # Chuẩn bị messages từ history và message hiện tại
+    messages = []
+    
+    # Thêm system message nếu có
+    if system_message:
+        messages.append({"role": "system", "content": system_message})
+    
+    # Thêm history (đã có format đúng)
     messages.extend(history)
-    messages.append({"role": "user", "content": message})
+    
+    # Thêm message hiện tại từ user
+    if user_message:
+        messages.append({"role": "user", "content": user_message})
     
     # Format messages với chat template
     prompt = tokenizer.apply_chat_template(
@@ -78,6 +93,11 @@ def respond(
     if prompt in answer:
         answer = answer.replace(prompt, "").strip()
     
+    # Đảm bảo trả về string không rỗng
+    if not answer or len(answer.strip()) == 0:
+        answer = "Xin lỗi, tôi không thể tạo phản hồi."
+    
+    print(f"Generated answer length: {len(answer)}")
     return answer
 
 
@@ -88,14 +108,7 @@ chatbot = gr.ChatInterface(
     respond,
     title="🤖 Coding Assistant",
     description="Chatbot hỗ trợ lập trình",
-    type="messages",
-    additional_inputs=[
-        gr.Textbox(
-            value="You are a helpful coding assistant. Provide clear, concise, and accurate code solutions and explanations.",
-            label="System message",
-            lines=3,
-        )
-    ]
+    type="messages"
 )
 
 demo = chatbot