Thong bao

app.py

@@ -50,15 +50,29 @@ MAX_NEW_TOKENS = 62
 # === Hàm xử lý ===
 def summarize_article(article_text):
     """Tóm tắt bài viết và tạo prompt refinement"""
+    # Kiểm tra rỗng
     if not article_text.strip():
-        return "[Empty]", "[Empty]"
-    summary = summarizer(article_text, num_beams=NUM_BEAMS,
+        return gr.update(value="❌ <span style='color:red'>Bạn chưa nhập bài viết</span>"), "", ""
+    
+    # Kiểm tra số từ
+    word_count = len(article_text.split())
+    if word_count < 20 or word_count > 300:
+        return gr.update(value=f"❌ <span style='color:red'>Bài viết phải từ 20–300 từ (hiện tại: {word_count} từ)</span>"), "", ""
+    
+    # Nếu hợp lệ thì xóa cảnh báo
+    error_msg = gr.update(value="")
+
+    summary = summarizer(
+        article_text,
+        num_beams=NUM_BEAMS,
         no_repeat_ngram_size=NO_REPEAT_NGRAM_SIZE,
         length_penalty=LENGTH_PENALTY,
         min_new_tokens=MIN_NEW_TOKENS,
-        max_new_tokens=MAX_NEW_TOKENS, do_sample=False)[0]["summary_text"]
+        max_new_tokens=MAX_NEW_TOKENS,
+        do_sample=False
+    )[0]["summary_text"]
+
     title = summary.split(".")[0] + "."
-    # Prompt refinement
     input_ids = promptist_tokenizer(title.strip() + " Rephrase:", return_tensors="pt").input_ids.to(device)
     eos_id = promptist_tokenizer.eos_token_id
     outputs = promptist_model.generate(
@@ -73,7 +87,9 @@ def summarize_article(article_text):
     )
     output_texts = promptist_tokenizer.batch_decode(outputs, skip_special_tokens=True)
     prompt = output_texts[0].replace(title + " Rephrase:", "").strip()
-    return title, prompt
+    
+    return error_msg, title, prompt
+
 
 def generate_images(prompt, style, num_images=4):
     """Sinh nhiều ảnh"""
@@ -113,6 +129,7 @@ def create_app():
                 choices=["Art", "Anime", "Watercolor", "Cyberpunk"],
                 label="🎨 Phong cách ảnh", value="Art"
             )
+        error_box = gr.Markdown(value="", elem_id="error-msg")
         num_images_slider = gr.Slider(1, 8, value=4, step=1, label="🔢 Số lượng ảnh")
         btn_summary = gr.Button("📌 Sinh Tiêu đề & Prompt")
 
@@ -132,7 +149,7 @@ def create_app():
         btn_summary.click(
             fn=summarize_article,
             inputs=[article_input],
-            outputs=[title_output, prompt_output]
+            outputs=[error_box, title_output, prompt_output]
         )
 
         def update_gallery(prompt, style, num_images):