Update app.py

app.py

@@ -29,13 +29,7 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history):
     # internal_history là [{"role": "user", ...}, {"role": "assistant", ...}]
     if internal_history is None:
         internal_history = []
-        
-    expected_key = os.environ.get("hf_key")
-    if expected_key not in prompt:
-        print("❌ Invalid key.")
-        return None
-    prompt = prompt.replace(expected_key, "")
-    
+
     # 2. Xây dựng toàn bộ lịch sử để đưa cho model
     # Bắt đầu với System Prompt (luôn lấy cái mới nhất từ Textbox)
     messages_for_model = [{"role": "system", "content": system_prompt}]
@@ -57,7 +51,7 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history):
     # 4. Generate
     output_tokens = model.generate(
         inputs,
-        max_new_tokens=5120, # Tăng số token tối đa lên một chút
+        max_new_tokens=512, # Tăng số token tối đa lên một chút
         do_sample=True,
         temperature=0.7,
         top_p=0.9
@@ -86,6 +80,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     # "Bộ nhớ" ẩn để lưu lịch sử ChatML (list of dicts)
     internal_history = gr.State()
 
+    gr.Markdown("# 💬 Xyntrai-Mistral 2.5 7B (Có trí nhớ)")
     
     with gr.Row():
         with gr.Column(scale=3):
@@ -112,7 +107,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
             # Ô System Prompt
             system_prompt_box = gr.Textbox(
                 label="System Prompt (AI's Role & Rules)",
-                value="",
+                value=SYSTEM_PROMPT_DEFAULT,
                 lines=30
             )