Upload auto_scheduler.py

auto_scheduler.py

@@ -1,7 +1,7 @@
-"""VNEWS Auto Scheduler - tự động đăng 3 bài rewrite AI + shorts từ 3 chủ đề HOT
+"""VNEWS Auto Scheduler - tự động đăng 5 bài rewrite AI + shorts từ 5 chủ đề HOT
 Vào các khung giờ: 7:00, 13:00, 19:00 (giờ Việt Nam)
 Mỗi bài: Rewrite AI từ nguồn báo + short video tự động
-FIX v7: Giữ nguyên tiêu đề gốc từng bài viết + thêm "Tin tóm tắt VNEWS 7h sáng/13h trưa/19h tối" ở đầu text
+FIX v8: Từ 3 lên 5 topics, thêm AI Thế Giới
 """
 import os, re, json, time, threading, asyncio, logging, random, hashlib, html as html_lib
 from datetime import datetime, timezone, timedelta, date
@@ -44,10 +44,9 @@ _STOP = set('và của các những một được trong với cho tại sau tr
 
 def _clean(s):
     s = html_lib.unescape(s or "")
-    # FIX: Remove malformed HTML artifacts (truncated tags without closing >)
     s = s.replace('<a href=" src="', '').replace("<a href=' src='", '')
     s = s.replace('<a href=" src=', '').replace("<a href=' src=", '')
-    s = re.sub(r'<[^>]+>', '', s)  # Remove all HTML tags
+    s = re.sub(r'<[^>]+>', '', s)
     return re.sub(r"\s+", " ", s).strip()
 
 def _get_hot_topics():
@@ -92,7 +91,7 @@ def _get_hot_topics():
         seen.add(key)
         topics.append({'label': '#' + re.sub(r'\s+', '', kw.title()), 'topic': kw, 'count': count})
         if len(topics) >= 20: break
-    for kw in ['World Cup 2026', 'Kinh tế Việt Nam', 'Bóng đá châu Âu', 'Công nghệ AI', 'Giá vàng', 'Thời tiết']:
+    for kw in ['World Cup 2026', 'Kinh tế Việt Nam', 'Bóng đá châu Âu', 'Công nghệ AI', 'AI Thế Giới', 'Giá vàng', 'Thời tiết']:
         if len(topics) >= 24: break
         if not any(kw.lower() in s for s in seen):
             topics.append({'label': '#' + re.sub(r'\s+', '', kw.title()), 'topic': kw, 'count': 0})
@@ -163,7 +162,6 @@ async def _create_ai_post(topic):
         LOG.warning(f"No articles for topic: {topic}. Fallback.")
         return await _create_fallback_post(topic, ai_ext, ai_patch)
     posts = []
-    # Get schedule time label for text intro (7h sáng, 13h trưa, 19h tối)
     now = datetime.now(VN_TZ)
     hour = now.hour
     time_label = "7h sáng" if hour == 7 else ("13h trưa" if hour == 13 else "19h tối")
@@ -176,13 +174,11 @@ async def _create_ai_post(topic):
             text = await ai_ext.qwen_generate(prompt, image_url=art.get('image'), max_tokens=1500)
             text = ai_patch._postprocess_ai_text(text, max_units=20)
             src = [art.get('source', {'title': art.get('title', topic), 'url': art.get('url', ''), 'via': art.get('via', '')})]
-            # Prepend time label intro to text (giữ nguyên title là tiêu đề gốc của bài báo)
             if text and not text.startswith(text_intro):
                 text = f"{text_intro}\n\n{text}"
             if 'Nguồn tham khảo:' not in (text or ''):
                 text = (text or '') + "\n\n" + ai_patch._source_line(src)
             img = art.get('image') or ai_ext.pollination_image_url(art.get('title', topic))
-            # Dùng art.get('title') GIỮ NGUYÊN tiêu đề gốc từ bài báo
             post = ai_ext.make_post(art.get('title', topic), text, img, art.get('url', ''), 'auto_scheduled', sources=src)
             try:
                 page_data = ai_patch._scrape_article_images(art.get('url', ''))
@@ -210,7 +206,6 @@ async def _create_ai_post(topic):
 async def _create_fallback_post(topic, ai_ext, ai_patch):
     LOG.info(f"Fallback: {topic}")
     try:
-        # Still add time label to fallback posts
         now = datetime.now(VN_TZ)
         hour = now.hour
         time_label = "7h sáng" if hour == 7 else ("13h trưa" if hour == 13 else "19h tối")
@@ -318,9 +313,10 @@ def _run_scheduled_posting():
             label = t.get('label', '')
             if label and label not in seen_labels:
                 seen_labels.add(label); selected.append(t['topic'])
-            if len(selected) >= 3: break
-        if len(selected) < 3:
-            selected = ['Thời sự Việt Nam', 'Kinh tế Việt Nam', 'Thể thao']
+            # Tăng từ 3 lên 5 topics
+            if len(selected) >= 5: break
+        if len(selected) < 5:
+            selected = ['Thời sự Việt Nam', 'Kinh tế Việt Nam', 'Công nghệ AI', 'Thể thao', 'Thế giới']
         LOG.info(f"Topics: {selected}")
         async def _do_all():
             results = []