Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import torch
+from typing import List, Union
 from huggingface_hub import login
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
@@ -26,8 +27,7 @@ def build_prompt(history, new_message):
     prompt += f"User: {new_message}\nAI:"
     return prompt
 
-def chat(history, new_message):
-    # Đảm bảo history là list (khi nhập trực tiếp trên UI đôi khi là str)
+def chat(history: Union[str, List[str]], new_message: str) -> str:
     if isinstance(history, str):
         import ast
         try:
@@ -63,7 +63,6 @@ with gr.Blocks() as demo:
     btn = gr.Button("Gửi")
     btn.click(_chat_ui, inputs=[history, new_message], outputs=output)
 
-    # Định nghĩa API endpoint /api/chat_ai
     gr.api(chat, api_name="chat_ai")
 
 demo.launch()