first commit
Browse files- README.md +1 -1
- app.py +0 -4
- gen_clear_view.py +24 -0
- pages_helpers.py +572 -0
- pages_notebook.py +258 -0
- requirements.txt +7 -0
README.md
CHANGED
|
@@ -5,7 +5,7 @@ colorFrom: purple
|
|
| 5 |
colorTo: green
|
| 6 |
sdk: streamlit
|
| 7 |
sdk_version: 1.38.0
|
| 8 |
-
app_file:
|
| 9 |
pinned: false
|
| 10 |
short_description: Tóm tắt và chat với các nội dung từ các links được cung cấp
|
| 11 |
---
|
|
|
|
| 5 |
colorTo: green
|
| 6 |
sdk: streamlit
|
| 7 |
sdk_version: 1.38.0
|
| 8 |
+
app_file: pages_notebook.py
|
| 9 |
pinned: false
|
| 10 |
short_description: Tóm tắt và chat với các nội dung từ các links được cung cấp
|
| 11 |
---
|
app.py
DELETED
|
@@ -1,4 +0,0 @@
|
|
| 1 |
-
import streamlit as st
|
| 2 |
-
|
| 3 |
-
x = st.slider('Select a value')
|
| 4 |
-
st.write(x, 'squared is', x * x)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
gen_clear_view.py
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os, sys, json
|
| 2 |
+
from pprint import pprint
|
| 3 |
+
|
| 4 |
+
import llm
|
| 5 |
+
|
| 6 |
+
|
| 7 |
+
try: filename = sys.argv[1]
|
| 8 |
+
except: filename = ".cache/https:/vietnamnet.vn/muc-nuoc-song-cau-o-thai-nguyen-giam-dan-nguoi-dan-tat-bat-don-dep-sau-lu-2321461.html"
|
| 9 |
+
|
| 10 |
+
clean_view_filename = filename + "__clean_view.txt"
|
| 11 |
+
|
| 12 |
+
# if not os.path.exists(clean_view_filename):
|
| 13 |
+
text_filename = filename + ".txt"
|
| 14 |
+
|
| 15 |
+
text = open(text_filename, "rt").read()
|
| 16 |
+
|
| 17 |
+
print(f"Chuẩn bị tạo nội dung cleanview {clean_view_filename}", flush = True)
|
| 18 |
+
clean_view = llm.gen_clean_view(text)
|
| 19 |
+
|
| 20 |
+
with open(clean_view_filename, "wt") as f:
|
| 21 |
+
f.write(clean_view)
|
| 22 |
+
|
| 23 |
+
print("\n- - - CLEAN VIEW - - -\n")
|
| 24 |
+
print(clean_view, flush = True)
|
pages_helpers.py
ADDED
|
@@ -0,0 +1,572 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import trafilatura
|
| 2 |
+
import requests
|
| 3 |
+
import lzma
|
| 4 |
+
import os
|
| 5 |
+
import re
|
| 6 |
+
import time
|
| 7 |
+
from datetime import datetime
|
| 8 |
+
import json
|
| 9 |
+
from pprint import pprint
|
| 10 |
+
import subprocess
|
| 11 |
+
|
| 12 |
+
import config
|
| 13 |
+
|
| 14 |
+
from utils import *
|
| 15 |
+
|
| 16 |
+
from text_utils import *
|
| 17 |
+
|
| 18 |
+
from llm import *
|
| 19 |
+
|
| 20 |
+
from mode_llm import llm_html_to_md, md_to_text, get_html_body_with_soup
|
| 21 |
+
|
| 22 |
+
from crawl4ai import WebCrawler # pip install "crawl4ai @ git+https://github.com/unclecode/crawl4ai.git"
|
| 23 |
+
|
| 24 |
+
# Create an instance of WebCrawler
|
| 25 |
+
crawler = WebCrawler()
|
| 26 |
+
|
| 27 |
+
# Warm up the crawler (load necessary models)
|
| 28 |
+
crawler.warmup()
|
| 29 |
+
|
| 30 |
+
## Cách lấy cookies và headers sử dụng https://curlconverter.com
|
| 31 |
+
cookies = {
|
| 32 |
+
'ASP.NET_SessionId': '42i3ivvgk14yd2tnxmddybvq',
|
| 33 |
+
'Culture': 'vi',
|
| 34 |
+
'Cookie_VB': 'close',
|
| 35 |
+
'ruirophaply-covi19': '24',
|
| 36 |
+
'SLG_G_WPT_TO': 'vi',
|
| 37 |
+
'G_ENABLED_IDPS': 'google',
|
| 38 |
+
'SLG_GWPT_Show_Hide_tmp': '1',
|
| 39 |
+
'SLG_wptGlobTipTmp': '1',
|
| 40 |
+
'__zlcmid': '1NOmxyopHgawxjN',
|
| 41 |
+
'45C5EF': '96780c17-dee3-49b2-9bf7-6335c4348d4f',
|
| 42 |
+
'vqc': '0',
|
| 43 |
+
}
|
| 44 |
+
|
| 45 |
+
headers = {
|
| 46 |
+
'accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
|
| 47 |
+
'accept-language': 'en-US,en;q=0.9',
|
| 48 |
+
'cache-control': 'max-age=0',
|
| 49 |
+
'priority': 'u=0, i',
|
| 50 |
+
'sec-ch-ua': '"Opera GX";v="111", "Chromium";v="125", "Not.A/Brand";v="24"',
|
| 51 |
+
'sec-ch-ua-mobile': '?0',
|
| 52 |
+
'sec-ch-ua-platform': '"Windows"',
|
| 53 |
+
'sec-fetch-dest': 'document',
|
| 54 |
+
'sec-fetch-mode': 'navigate',
|
| 55 |
+
'sec-fetch-site': 'none',
|
| 56 |
+
'sec-fetch-user': '?1',
|
| 57 |
+
'sec-gpc': '1',
|
| 58 |
+
'upgrade-insecure-requests': '1',
|
| 59 |
+
'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36 OPR/111.0.0.0',
|
| 60 |
+
# 'User-Agent': "Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Mobile Safari/537.36",
|
| 61 |
+
}
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
def norm_url_and_gen_filename(url):
|
| 65 |
+
url = url.strip() # loại bỏ ký tự trống ở đầu và cuối
|
| 66 |
+
if url[-1] == "/": url = url[:-1] # loại bỏ "/" ở cuối
|
| 67 |
+
|
| 68 |
+
# init filename và tạo sub folder nếu chưa có
|
| 69 |
+
filename = f'.cache/{url}'
|
| 70 |
+
return url, filename
|
| 71 |
+
|
| 72 |
+
|
| 73 |
+
def reset_content(url):
|
| 74 |
+
_, filename = norm_url_and_gen_filename(url)
|
| 75 |
+
|
| 76 |
+
text_filename = filename + ".txt"
|
| 77 |
+
json_filename = filename + ".json"
|
| 78 |
+
html_filename = filename + ".html"
|
| 79 |
+
xz_filename = filename + ".xz"
|
| 80 |
+
|
| 81 |
+
for filename in [
|
| 82 |
+
text_filename,
|
| 83 |
+
json_filename,
|
| 84 |
+
html_filename,
|
| 85 |
+
xz_filename,
|
| 86 |
+
]:
|
| 87 |
+
cmd = f"rm -rf '{filename}'"; print(cmd)
|
| 88 |
+
subprocess.run(cmd, shell = True)
|
| 89 |
+
|
| 90 |
+
|
| 91 |
+
# from functools import lru_cache
|
| 92 |
+
# @lru_cache(maxsize=128)
|
| 93 |
+
def url_content(url, update_text=None):
|
| 94 |
+
url, filename = norm_url_and_gen_filename(url)
|
| 95 |
+
parts = filename.split("/")
|
| 96 |
+
|
| 97 |
+
for i in range(1, len(parts)):
|
| 98 |
+
path = "/".join(parts[:i])
|
| 99 |
+
# print(path) # DEBUG
|
| 100 |
+
os.makedirs(path, exist_ok=True)
|
| 101 |
+
|
| 102 |
+
|
| 103 |
+
text_filename = filename + ".txt"
|
| 104 |
+
json_filename = filename + ".json"
|
| 105 |
+
html_filename = filename + ".html"
|
| 106 |
+
xz_filename = filename + ".xz"
|
| 107 |
+
|
| 108 |
+
|
| 109 |
+
# Nếu được yêu cầu update nội dung text của url thì update
|
| 110 |
+
if update_text is not None:
|
| 111 |
+
|
| 112 |
+
print("\nUPDATE TEXT", filename)
|
| 113 |
+
|
| 114 |
+
text, noise_ratio, max_noise = normalize_text(update_text, get_noise_info = True)
|
| 115 |
+
|
| 116 |
+
# update meta, Lưu lại text và meta
|
| 117 |
+
meta = json.load(open(json_filename))
|
| 118 |
+
|
| 119 |
+
meta["updated_at"] = str(datetime.now())
|
| 120 |
+
meta["updates_count"] += 1
|
| 121 |
+
|
| 122 |
+
meta["noise_ratio"] = noise_ratio
|
| 123 |
+
meta["max_noise"] = max_noise
|
| 124 |
+
|
| 125 |
+
# Cắt ngắn text nếu đầu vào quá dài
|
| 126 |
+
text = cut_if_too_long(text, meta)
|
| 127 |
+
|
| 128 |
+
with open(json_filename, "wt") as f:
|
| 129 |
+
f.write(json.dumps(meta, ensure_ascii = False))
|
| 130 |
+
|
| 131 |
+
with open(text_filename, "wt") as f:
|
| 132 |
+
f.write(text)
|
| 133 |
+
|
| 134 |
+
# gọi gen_clear_view.py (chạy chậm) ở process riêng
|
| 135 |
+
get_clear_view(filename)
|
| 136 |
+
|
| 137 |
+
# Re-gen llm contents
|
| 138 |
+
get_llm_gen_contents(url, use_cache = False)
|
| 139 |
+
|
| 140 |
+
print(CYAN,end=""); print(meta); print(RESET,end="", flush=True)
|
| 141 |
+
|
| 142 |
+
|
| 143 |
+
# Nếu tồn tại text_filename thì trả về,
|
| 144 |
+
# đây có thể là nội dung text đã được chỉnh sửa nên ưu tiên trả về trước
|
| 145 |
+
if os.path.exists(text_filename):
|
| 146 |
+
print("\nGOT TEXT", filename, flush=True)
|
| 147 |
+
|
| 148 |
+
norm_text = open(text_filename, "rt").read()
|
| 149 |
+
return norm_text
|
| 150 |
+
|
| 151 |
+
|
| 152 |
+
html = None
|
| 153 |
+
# Thử đọc nội dung html đã được cache
|
| 154 |
+
if os.path.exists(xz_filename):
|
| 155 |
+
try:
|
| 156 |
+
html = lzma.open(xz_filename,"rt").read()
|
| 157 |
+
print("\nGOT HTML", filename, flush=True)
|
| 158 |
+
except:
|
| 159 |
+
pass
|
| 160 |
+
|
| 161 |
+
|
| 162 |
+
|
| 163 |
+
blacklist = """
|
| 164 |
+
Your connection is not private
|
| 165 |
+
-----BEGIN CERTIFICATE-----
|
| 166 |
+
""".strip().split("\n")
|
| 167 |
+
|
| 168 |
+
|
| 169 |
+
## Thử các cách lấy HTML: requests vs crawl4ai vs reader
|
| 170 |
+
get_html_method = "requests"
|
| 171 |
+
if html is None:
|
| 172 |
+
# Thử lần 1 bằng requests
|
| 173 |
+
print("\nGET HTML", filename, flush=True)
|
| 174 |
+
|
| 175 |
+
try:
|
| 176 |
+
html = requests.get(url, cookies=cookies, headers=headers)
|
| 177 |
+
html = str(html.text)
|
| 178 |
+
|
| 179 |
+
# Lưu lại nội dung html vào xz_filename
|
| 180 |
+
with lzma.open(xz_filename, "wt") as f: f.write(html)
|
| 181 |
+
|
| 182 |
+
except Exception as e:
|
| 183 |
+
print(f"!!! REQUESTS Error {e} !!!")
|
| 184 |
+
|
| 185 |
+
if isinstance(html, str):
|
| 186 |
+
for x in blacklist:
|
| 187 |
+
if x in html:
|
| 188 |
+
print(f"--{x}--")
|
| 189 |
+
i = html.find(x)
|
| 190 |
+
print(f"{RED}!!! REQUESTS đọc lỗi {html[i-30:i+200]} !!!{RESET}")
|
| 191 |
+
html = None
|
| 192 |
+
break
|
| 193 |
+
|
| 194 |
+
|
| 195 |
+
meta = None
|
| 196 |
+
if html is None or len(html) < 500:
|
| 197 |
+
# Thử lần 2 bằng CRAWL4AI
|
| 198 |
+
print("GET HTML CRAWL4AI", filename, flush=True)
|
| 199 |
+
get_html_method = "crawl4ai"
|
| 200 |
+
|
| 201 |
+
try:
|
| 202 |
+
result = crawler.run(url=url)
|
| 203 |
+
html = result.html
|
| 204 |
+
|
| 205 |
+
# Lưu lại nội dung html vào xz_filename
|
| 206 |
+
with lzma.open(xz_filename, "wt") as f: f.write(html)
|
| 207 |
+
|
| 208 |
+
# {'title': 'Ngập úng và thiệt hại trên 202.000 ha lúa | baotintuc.vn', 'description': 'Thống kê từ Bộ Nông nghiệp và Phát triển nông thôn, tính đến sáng 13/9, có trên 202.000 ha lúa, gần 39.300 ha hoa màu bị ngập úng, thiệt hại do ảnh hưởng của bão số 3.', 'keywords': None, 'author': 'baotintuc.vn', 'og:type': 'article', 'og:url': 'https://baotintuc.vn/xa-hoi/ngap-ung-va-thiet-hai-tren-202000-ha-lua-20240913095621343.htm', 'og:image': 'https://cdnmedia.baotintuc.vn/Upload/EqV5H9rWgvy9oNikwkHLXA/files/13092024-bao-1.jpg', 'og:image:url': 'https://cdnmedia.baotintuc.vn/Upload/EqV5H9rWgvy9oNikwkHLXA/files/13092024-bao-1.jpg', 'og:image:secure_url': 'https://cdnmedia.baotintuc.vn/Upload/EqV5H9rWgvy9oNikwkHLXA/files/13092024-bao-1.jpg', 'og:image:width': '460', 'og:image:height': '345', 'og:title': 'Ngập úng và thiệt hại trên 202.000 ha lúa', 'og:description': 'Thống kê từ Bộ Nông nghiệp và Phát triển nông thôn, đến sáng 13/9, có trên 202.000 ha lúa, gần 39.300 ha hoa màu bị ngập úng, thiệt hại do ảnh hưởng của bão số 3.', 'twitter: card': 'summary_large_image', 'twitter: image': 'https://cdnmedia.baotintuc.vn/Upload/EqV5H9rWgvy9oNikwkHLXA/files/13092024-bao-1.jpg'}
|
| 209 |
+
meta = dict(result.metadata)
|
| 210 |
+
|
| 211 |
+
for key in result.metadata.keys():
|
| 212 |
+
if "og:" in key or "twitter:" in key:
|
| 213 |
+
meta.pop(key)
|
| 214 |
+
|
| 215 |
+
except Exception as e:
|
| 216 |
+
print(f"!!! CRAWL4AI Error {e} !!!")
|
| 217 |
+
|
| 218 |
+
if isinstance(html, str):
|
| 219 |
+
for x in blacklist:
|
| 220 |
+
if x in html:
|
| 221 |
+
i = html.find(x)
|
| 222 |
+
print(f"{RED}!!! CRAWL4AI đọc lỗi {html[i-30:i+200]} !!!{RESET}")
|
| 223 |
+
html = None
|
| 224 |
+
meta = {}
|
| 225 |
+
break
|
| 226 |
+
|
| 227 |
+
|
| 228 |
+
if html is None or len(html) < 500:
|
| 229 |
+
# Thử lần 3 bằng reader api
|
| 230 |
+
print("GET HTML READER", filename, flush=True)
|
| 231 |
+
get_html_method = "reader"
|
| 232 |
+
|
| 233 |
+
try:
|
| 234 |
+
reader_url = "https://r.jina.ai/" + url
|
| 235 |
+
# Use below header make https://jina.ai/reader return text not markdown by default
|
| 236 |
+
html = requests.get(reader_url, headers = { 'X-Return-Format': 'html', }).text
|
| 237 |
+
|
| 238 |
+
# Lưu lại nội dung html vào xz_filename
|
| 239 |
+
with lzma.open(xz_filename, "wt") as f: f.write(html)
|
| 240 |
+
|
| 241 |
+
except Exception as e:
|
| 242 |
+
print(f"!!! READER Error {e} !!!")
|
| 243 |
+
|
| 244 |
+
|
| 245 |
+
if isinstance(html, str):
|
| 246 |
+
for x in blacklist:
|
| 247 |
+
if x in html:
|
| 248 |
+
i = html.find(x)
|
| 249 |
+
print(f"{RED}!!! READER đọc lỗi {html[i-30:i+200]} !!!{RESET}")
|
| 250 |
+
html = None
|
| 251 |
+
break
|
| 252 |
+
|
| 253 |
+
|
| 254 |
+
## Thử các cách extract text: trafilatura vs llm vs reader
|
| 255 |
+
extract_method = "trafilatura"
|
| 256 |
+
# https://trafilatura.readthedocs.io/en/latest/corefunctions.html#extract
|
| 257 |
+
try:
|
| 258 |
+
text = trafilatura.extract(html,
|
| 259 |
+
# favor_recall = True,
|
| 260 |
+
include_tables = True,
|
| 261 |
+
include_comments = False,
|
| 262 |
+
with_metadata = False,
|
| 263 |
+
)
|
| 264 |
+
except:
|
| 265 |
+
text = ""
|
| 266 |
+
|
| 267 |
+
if meta is None: # Có thể meta đã đc lấy ở crawl4ai
|
| 268 |
+
try:
|
| 269 |
+
meta = trafilatura.extract(html, only_with_metadata = True)
|
| 270 |
+
if meta and len(meta) > 0:
|
| 271 |
+
# print(meta); input() # DEBUG
|
| 272 |
+
meta = meta.split("---")[1]
|
| 273 |
+
splits = re.findall(r'\S+: [^\n]+', meta)
|
| 274 |
+
meta = { x.split(": ", 1)[0].strip() : x.split(": ", 1)[1].strip() for x in splits }
|
| 275 |
+
else:
|
| 276 |
+
meta = {}
|
| 277 |
+
except:
|
| 278 |
+
meta = {}
|
| 279 |
+
|
| 280 |
+
|
| 281 |
+
# Chuẩn hóa text
|
| 282 |
+
if text is None: text = ""
|
| 283 |
+
text, noise_ratio, max_noise = normalize_text(text, get_noise_info = True)
|
| 284 |
+
print(f">>> {RED}noise_ratio {pretty_num(noise_ratio)}, max_noise {max_noise}{RESET}")
|
| 285 |
+
|
| 286 |
+
|
| 287 |
+
MEANINGFUL = 500
|
| 288 |
+
MAX_NOISE_RATIO = 0.3
|
| 289 |
+
|
| 290 |
+
too_short = ( len(text) < MEANINGFUL )
|
| 291 |
+
too_noise = ( noise_ratio > MAX_NOISE_RATIO or max_noise > MEANINGFUL )
|
| 292 |
+
|
| 293 |
+
# ko lấy đc text hoặc text quá ngắn (cào trượt), hoặc text quá noise
|
| 294 |
+
if text is None or too_short or too_noise:
|
| 295 |
+
# Lấy text thông qua phương pháp khác
|
| 296 |
+
print("!!! Đoạn text dưới do trafilatura triết xuất có vấn đề?")
|
| 297 |
+
print("too short", too_short)
|
| 298 |
+
print("too noise", too_noise)
|
| 299 |
+
print("- - - "*6)
|
| 300 |
+
print(f"{YELLOW}{text}{RESET}")
|
| 301 |
+
print("- - - "*6)
|
| 302 |
+
|
| 303 |
+
print("!!! Dùng Jina Reader ...")
|
| 304 |
+
reader_url = "https://r.jina.ai/" + url
|
| 305 |
+
# Use below header make https://jina.ai/reader return text not markdown by default
|
| 306 |
+
reader_text = requests.get(reader_url, headers = { 'X-Return-Format': 'text', }).text
|
| 307 |
+
reader_text, reader_noise_ratio, reader_max_noise = normalize_text(reader_text, get_noise_info = True)
|
| 308 |
+
|
| 309 |
+
# Chuẩn hóa text
|
| 310 |
+
reader_text, reader_noise_ratio, reader_max_noise = normalize_text(reader_text, get_noise_info = True)
|
| 311 |
+
|
| 312 |
+
reader_too_noise = ( reader_noise_ratio > MAX_NOISE_RATIO or reader_max_noise > MEANINGFUL )
|
| 313 |
+
|
| 314 |
+
print(f">>> {RED}reader_noise_ratio {pretty_num(reader_noise_ratio)}, reader_max_noise {reader_max_noise}{RESET}")
|
| 315 |
+
print(f">>> {RED}reader_too_noise {reader_too_noise}{RESET}")
|
| 316 |
+
|
| 317 |
+
signal = int( len(text) * (1 - noise_ratio) ) + 1
|
| 318 |
+
reader_signal = int( len(reader_text) * (1 - reader_noise_ratio) ) + 1
|
| 319 |
+
|
| 320 |
+
samesame = ( abs(signal - reader_signal) / reader_signal ) < 0.2
|
| 321 |
+
print(f">>> {RED}samesame {samesame}, original signal {pretty_num(signal)}, reader_signal {pretty_num(reader_signal)}{RESET}")
|
| 322 |
+
|
| 323 |
+
# Nếu bản gốc quá ngắn nhưng bản reader quá noise thì thà chọn ngắn còn hơn
|
| 324 |
+
original_too_shot_but_reader_too_noise = (
|
| 325 |
+
too_short and (samesame or reader_noise_ratio >= 0.5 )
|
| 326 |
+
)
|
| 327 |
+
|
| 328 |
+
original_too_noise_but_reader_even_more_noise = (
|
| 329 |
+
too_noise and noise_ratio < reader_noise_ratio and max_noise < reader_max_noise
|
| 330 |
+
)
|
| 331 |
+
|
| 332 |
+
if original_too_shot_but_reader_too_noise:
|
| 333 |
+
print("!!! reader quá noise, chọn bản trafilatura too_short còn hơn.")
|
| 334 |
+
|
| 335 |
+
if original_too_noise_but_reader_even_more_noise:
|
| 336 |
+
print("!!! reader còn noise hơn bản trafilatura, bỏ qua.")
|
| 337 |
+
|
| 338 |
+
|
| 339 |
+
if not original_too_shot_but_reader_too_noise and \
|
| 340 |
+
not original_too_noise_but_reader_even_more_noise:
|
| 341 |
+
|
| 342 |
+
choose_original_text = False
|
| 343 |
+
|
| 344 |
+
if reader_too_noise: # vẫn còn noisy lắm, thử dùng readability.js
|
| 345 |
+
|
| 346 |
+
if html is not None and len(html) > 200:
|
| 347 |
+
|
| 348 |
+
html_filename = filename + ".html"
|
| 349 |
+
with open(html_filename, "wt") as f:
|
| 350 |
+
f.write(html)
|
| 351 |
+
|
| 352 |
+
abi_text = subprocess.run(
|
| 353 |
+
f"node node_readability.js '{html_filename}' '{url}'",
|
| 354 |
+
shell=True,
|
| 355 |
+
capture_output=True,
|
| 356 |
+
).stdout.decode('utf-8')
|
| 357 |
+
|
| 358 |
+
abi_text, abi_noise_ratio, abi_max_noise = \
|
| 359 |
+
normalize_text(abi_text, get_noise_info = True)
|
| 360 |
+
|
| 361 |
+
if abi_max_noise < reader_max_noise:
|
| 362 |
+
print(GREEN, ">>>", abi_text, "<<<", RESET)
|
| 363 |
+
|
| 364 |
+
if len(abi_text) < len(reader_text) and len(text) < len(reader_text): # chuộng text ngắn
|
| 365 |
+
choose_original_text = True
|
| 366 |
+
|
| 367 |
+
|
| 368 |
+
if not choose_original_text:
|
| 369 |
+
|
| 370 |
+
extract_method = "reader"
|
| 371 |
+
text = reader_text
|
| 372 |
+
noise_ratio = reader_noise_ratio
|
| 373 |
+
max_noise = reader_max_noise
|
| 374 |
+
|
| 375 |
+
|
| 376 |
+
# update meta, Lưu lại text và meta
|
| 377 |
+
meta["url"] = url
|
| 378 |
+
meta["get_html_method"] = get_html_method
|
| 379 |
+
meta["extract_method"] = extract_method
|
| 380 |
+
meta["created_at"] = str(datetime.now())
|
| 381 |
+
meta["updates_count"] = 0
|
| 382 |
+
meta["noise_ratio"] = noise_ratio
|
| 383 |
+
meta["max_noise"] = max_noise
|
| 384 |
+
meta["text_origin_len"] = len(text)
|
| 385 |
+
|
| 386 |
+
if "hostname" in meta: meta.pop("hostname")
|
| 387 |
+
if "sitename" in meta: meta.pop("sitename")
|
| 388 |
+
|
| 389 |
+
# Thêm title và description vào text (nếu có)
|
| 390 |
+
norm_text = normalize_text(text)
|
| 391 |
+
text = add_title_desc_to_text(norm_text, meta)
|
| 392 |
+
|
| 393 |
+
# Cắt ngắn text nếu đầu vào quá dài
|
| 394 |
+
text = cut_if_too_long(text, meta)
|
| 395 |
+
|
| 396 |
+
print(CYAN,end=""); print(meta); print(RESET,end="")
|
| 397 |
+
|
| 398 |
+
with open(json_filename, "wt") as f:
|
| 399 |
+
f.write(json.dumps(meta, ensure_ascii = False))
|
| 400 |
+
|
| 401 |
+
with open(text_filename, "wt") as f:
|
| 402 |
+
f.write(text)
|
| 403 |
+
|
| 404 |
+
get_clear_view(filename)
|
| 405 |
+
get_llm_gen_contents(url, use_cache = False)
|
| 406 |
+
|
| 407 |
+
return text
|
| 408 |
+
|
| 409 |
+
|
| 410 |
+
def get_clear_view(filename):
|
| 411 |
+
# gọi gen_clear_view.py (chạy chậm) ở process riêng
|
| 412 |
+
subprocess.run(f"nohup python3 gen_clear_view.py '{filename}' &", shell = True)
|
| 413 |
+
import time; time.sleep(1) # chờ 1 giây
|
| 414 |
+
|
| 415 |
+
|
| 416 |
+
def cut_if_too_long(text, meta, max_words = config.text_max_words):
|
| 417 |
+
words = text.split()
|
| 418 |
+
|
| 419 |
+
if len(words) > max_words:
|
| 420 |
+
|
| 421 |
+
words = words[ : max_words]
|
| 422 |
+
threshold = len(" ".join(words))
|
| 423 |
+
|
| 424 |
+
meta["text_cutoff"] = True
|
| 425 |
+
meta["text_cutoff_len"] = threshold
|
| 426 |
+
|
| 427 |
+
return text[ : threshold ]
|
| 428 |
+
|
| 429 |
+
else:
|
| 430 |
+
return text
|
| 431 |
+
|
| 432 |
+
|
| 433 |
+
def add_title_desc_to_text(text, meta):
|
| 434 |
+
content = []
|
| 435 |
+
title = meta["title"] if "title" in meta else None
|
| 436 |
+
description = meta["description"] if "description" in meta else None
|
| 437 |
+
|
| 438 |
+
if title is not None and len(title) > 5:
|
| 439 |
+
content.append(f"**title**: {title}")
|
| 440 |
+
|
| 441 |
+
if description is not None and len(description) > 10:
|
| 442 |
+
content.append(f"**description**: {description}")
|
| 443 |
+
|
| 444 |
+
content.append(text)
|
| 445 |
+
return "\n\n".join(content)
|
| 446 |
+
|
| 447 |
+
|
| 448 |
+
|
| 449 |
+
def normalize_text(text, get_noise_info = False):
|
| 450 |
+
text = text.strip()
|
| 451 |
+
chunks = re.split(r'\s*(?:\n\s*)+', text, flags = re.MULTILINE)
|
| 452 |
+
text = "\n\n".join([ x for x in chunks if len(x) > 20 ])
|
| 453 |
+
|
| 454 |
+
if get_noise_info:
|
| 455 |
+
noise_len = 1
|
| 456 |
+
total_len = 1
|
| 457 |
+
|
| 458 |
+
max_noise = 0
|
| 459 |
+
continuous_noise = 0
|
| 460 |
+
for x in chunks:
|
| 461 |
+
n = len(x)
|
| 462 |
+
total_len += n
|
| 463 |
+
|
| 464 |
+
if n < 80:
|
| 465 |
+
noise_len += n
|
| 466 |
+
continuous_noise += n
|
| 467 |
+
if continuous_noise > max_noise:
|
| 468 |
+
max_noise = continuous_noise
|
| 469 |
+
else:
|
| 470 |
+
continuous_noise = 0
|
| 471 |
+
|
| 472 |
+
noise_ratio = noise_len / total_len
|
| 473 |
+
return text, noise_ratio, max_noise
|
| 474 |
+
else:
|
| 475 |
+
return text
|
| 476 |
+
|
| 477 |
+
|
| 478 |
+
def get_clean_view(url):
|
| 479 |
+
url, filename = norm_url_and_gen_filename(url)
|
| 480 |
+
clean_view_filename = filename + "__clean_view.txt"
|
| 481 |
+
|
| 482 |
+
if os.path.exists(clean_view_filename):
|
| 483 |
+
return open(clean_view_filename, "rt").read()
|
| 484 |
+
else:
|
| 485 |
+
return None
|
| 486 |
+
|
| 487 |
+
|
| 488 |
+
def get_meta(url):
|
| 489 |
+
url, filename = norm_url_and_gen_filename(url)
|
| 490 |
+
json_filename = filename + ".json"
|
| 491 |
+
return json.load(open(json_filename))
|
| 492 |
+
|
| 493 |
+
|
| 494 |
+
TAGS = "keyphrases figures summary".split()
|
| 495 |
+
###
|
| 496 |
+
def get_llm_gen_contents(url, use_cache = True):
|
| 497 |
+
url, filename = norm_url_and_gen_filename(url)
|
| 498 |
+
json_filename = filename + ".json"
|
| 499 |
+
text_filename = filename + ".txt"
|
| 500 |
+
|
| 501 |
+
if os.path.exists(json_filename):
|
| 502 |
+
meta = json.load(open(json_filename, "rt"))
|
| 503 |
+
|
| 504 |
+
generated = ( "llm_generated" in meta )
|
| 505 |
+
|
| 506 |
+
if not use_cache or not generated:
|
| 507 |
+
|
| 508 |
+
text = open(text_filename, "rt").read()
|
| 509 |
+
marked_text, chunks = add_chunk_markers(text, para = True)
|
| 510 |
+
|
| 511 |
+
raw = extract_keyphrases_figures_summary(marked_text)
|
| 512 |
+
result = extract_xmls(raw, TAGS)
|
| 513 |
+
|
| 514 |
+
result["raw"] = raw
|
| 515 |
+
meta["llm_generated"] = result
|
| 516 |
+
|
| 517 |
+
with open(json_filename, "wt") as f:
|
| 518 |
+
f.write(json.dumps(meta, ensure_ascii = False))
|
| 519 |
+
|
| 520 |
+
return meta["llm_generated"]
|
| 521 |
+
|
| 522 |
+
else:
|
| 523 |
+
return {
|
| 524 |
+
"summary": "Tóm tắt nội dung ... văn bản nói về ...",
|
| 525 |
+
"keyphrases": ["keywords 1", "keywords 2", "keywords 3"]
|
| 526 |
+
}
|
| 527 |
+
|
| 528 |
+
|
| 529 |
+
default_urls_input = """
|
| 530 |
+
https://thuvienphapluat.vn/phap-luat/ho-so-dien-tu-thuc-hien-thu-tuc-hanh-chinh-la-gi-huong-dan-chuan-bi-va-nop-ho-so-dien-tu-khi-thuc-h-155754-140107.html
|
| 531 |
+
https://video.vnexpress.net/bon-ngay-chong-choi-lu-ngap-gan-3-m-cua-nguoi-dan-thai-nguyen-4791440.html
|
| 532 |
+
http://danvan.vn/Home/Tin-hoat-dong/Ban-dan-van/18706/Ban-Dan-van-Trung-uong-va-Hoi-Chu-thap-do-Viet-Nam-tham-tang-qua-nhan-dan-bi-anh-huong-bao-so-3-tai-Thai-Nguyen
|
| 533 |
+
https://baodauthau.vn/thai-nguyen-144-ty-dong-nang-cap-duong-cach-mang-thang-8-tp-song-cong-post164486.html
|
| 534 |
+
https://baothainguyen.vn/chinh-tri/202409/chu-tich-quoc-hoi-tran-thanh-man-lam-viec-voi-tinh-thai-nguyen-ve-cong-tackhac-phuc-hau-qua-bao-so-3-3f9253f/
|
| 535 |
+
https://baothainguyen.vn/giao-duc/202409/dam-bao-dieu-kien-de-hoc-sinh-tro-lai-truong-cham-nhat-ngay-16-9-9742985/
|
| 536 |
+
https://baothainguyen.vn/tai-nguyen-moi-truong/202409/khu-khuan-dien-rong-nhung-vung-bi-ngap-lut-tai-tp-thai-nguyen-585273d/
|
| 537 |
+
https://baothainguyen.vn/thoi-su-thai-nguyen/202409/dien-luc-tp-thai-nguyen-no-luccap-dien-tro-lai-cho-tren-2000-hotrong-ngay-12-9-da21a20/
|
| 538 |
+
https://baothainguyen.vn/xa-hoi/202409/tao-sinh-ke-giam-ngheo-vung-dong-bao-dan-toc-thieu-so-b8f041c/
|
| 539 |
+
https://baotintuc.vn/xa-hoi/ngap-ung-va-thiet-hai-tren-202000-ha-lua-20240913095621343.htm
|
| 540 |
+
https://daidoanket.vn/thai-nguyen-hai-nguoi-tu-vong-thiet-hai-hon-600-ty-dong-do-bao-yagi-10290104.html
|
| 541 |
+
https://dangcongsan.vn/xay-dung-dang/thai-nguyen-cong-bo-cac-quyet-dinh-ve-cong-tac-can-bo-677747.html
|
| 542 |
+
https://danviet.vn/62-y-bac-si-cua-binh-dinh-den-thai-nguyen-yen-bai-quyet-tam-cung-dong-bao-vuot-qua-kho-khan-20240913101402511.htm
|
| 543 |
+
https://laodong.vn/thoi-su/chu-tich-quoc-hoi-kiem-tra-cong-tac-khac-phuc-hau-qua-mua-lu-o-thai-nguyen-1393445.ldo
|
| 544 |
+
https://nhandan.vn/anh-chu-tich-quoc-hoi-tran-thanh-man-kiem-tra-cong-tac-khac-phuc-hau-qua-bao-so-3-tai-tinh-thai-nguyen-post830447.html
|
| 545 |
+
https://nld.com.vn/toi-7-gio-13-9-336-nguoi-chet-va-mat-tich-hon-130-ngan-nguoi-dan-phai-di-doi-do-bao-lu-196240913101124546.htm
|
| 546 |
+
https://phunuvietnam.vn/thai-nguyen-hoi-vien-phu-nu-chung-tay-khac-phuc-hau-qua-ngap-lut-20240912154801867.htm
|
| 547 |
+
https://phunuvietnam.vn/thai-nguyen-trien-khai-cong-tac-phong-chong-dich-sau-thien-tai-20240912174641866.htm
|
| 548 |
+
https://thainguyen.dcs.vn/hoat-dong-cua-cac-dang-bo/dang-bo-tp-thai-nguyen/hoi-nghi-ban-thuong-vu-thanh-uy-thai-nguyen-lan-thu-102-857.html
|
| 549 |
+
https://thainguyen.dms.gov.vn/tin-chi-tiet/-/chi-tiet/thai-nguyen-%C4%91am-bao-nguon-hang-hoa-phuc-vu-nhan-dan-89820-1404.html
|
| 550 |
+
https://thuonghieucongluan.com.vn/thai-nguyen-tiep-nhan-5-tan-gao-ho-tro-nhan-dan-bi-anh-huong-ngap-lut-a235642.html
|
| 551 |
+
https://tienphong.vn/nam-thanh-nien-o-thai-nguyen-bi-lu-cuon-khi-di-bat-ca-post1672693.tpo
|
| 552 |
+
https://tienphong.vn/ngan-hang-dau-tien-cong-bo-giam-lai-suat-cho-vay-sau-bao-so-3-post1672728.tpo
|
| 553 |
+
https://tuoitre.vn/chu-tich-quoc-hoi-tran-thanh-man-trao-30-ti-dong-ho-tro-khac-phuc-bao-lu-tai-thai-nguyen-20240912191724375.htm
|
| 554 |
+
https://tuoitre.vn/sau-lu-nguoi-dan-thai-nguyen-noi-chua-bao-gio-bun-ngap-nhieu-den-vay-202409121653144.htm
|
| 555 |
+
https://vietnamnet.vn/muc-nuoc-song-cau-o-thai-nguyen-giam-dan-nguoi-dan-tat-bat-don-dep-sau-lu-2321461.html
|
| 556 |
+
https://vtcnews.vn/trieu-nu-cuoi-huong-ve-thai-nguyen-sau-con-bao-ar895714.html
|
| 557 |
+
""".strip()
|
| 558 |
+
|
| 559 |
+
default_urls_input = """
|
| 560 |
+
https://vnexpress.net/sam-altman-ai-thong-minh-hon-con-nguoi-trong-vai-nghin-ngay-toi-4796649.html
|
| 561 |
+
https://vnexpress.net/may-tram-chay-ai-gia-tram-trieu-dong-tai-viet-nam-4796490.html
|
| 562 |
+
https://www.vngcloud.vn/blog/what-are-large-language-models
|
| 563 |
+
https://arxiv.org/html/2408.16737v1
|
| 564 |
+
https://arxiv.org/html/2409.15700v1
|
| 565 |
+
https://arxiv.org/html/2409.09916v1
|
| 566 |
+
https://arxiv.org/html/2409.06903v1
|
| 567 |
+
https://arxiv.org/html/2409.12558v1
|
| 568 |
+
https://arxiv.org/html/2409.10516v2
|
| 569 |
+
https://rlhflow.github.io/posts/2024-05-29-multi-objective-reward-modeling
|
| 570 |
+
https://arxiv.org/html/2405.07863v2
|
| 571 |
+
https://arxiv.org/html/2406.12845
|
| 572 |
+
""".strip()
|
pages_notebook.py
ADDED
|
@@ -0,0 +1,258 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import re
|
| 2 |
+
import html
|
| 3 |
+
import uuid
|
| 4 |
+
import urllib
|
| 5 |
+
|
| 6 |
+
import streamlit as st
|
| 7 |
+
from pages.sticky_header import sticky_container
|
| 8 |
+
|
| 9 |
+
import sys; sys.path.append('../')
|
| 10 |
+
from pages_helpers import *
|
| 11 |
+
from text_utils import *
|
| 12 |
+
|
| 13 |
+
import config
|
| 14 |
+
|
| 15 |
+
# Kích hoạt chế độ toàn màn hình (full width)
|
| 16 |
+
st.set_page_config(layout="wide")
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
###################################################################
|
| 20 |
+
|
| 21 |
+
# Define HTML and CSS for the tooltip
|
| 22 |
+
html_content = """
|
| 23 |
+
<style>
|
| 24 |
+
.tooltip {
|
| 25 |
+
position: relative;
|
| 26 |
+
display: inline-block;
|
| 27 |
+
cursor: pointer;
|
| 28 |
+
}
|
| 29 |
+
|
| 30 |
+
.tooltip .tooltiptext {
|
| 31 |
+
font-weight: normal;
|
| 32 |
+
visibility: hidden;
|
| 33 |
+
width: 320px;
|
| 34 |
+
background-color: #eadef9;
|
| 35 |
+
color: black;
|
| 36 |
+
text-align: left;
|
| 37 |
+
border-radius: 6px;
|
| 38 |
+
padding: 0.6em;
|
| 39 |
+
position: absolute;
|
| 40 |
+
z-index: 1;
|
| 41 |
+
top: 100%;
|
| 42 |
+
left: 50%;
|
| 43 |
+
margin-left: -60px;
|
| 44 |
+
opacity: 20%;
|
| 45 |
+
transition: opacity 0.3s;
|
| 46 |
+
}
|
| 47 |
+
|
| 48 |
+
.tooltip:hover .tooltiptext {
|
| 49 |
+
visibility: visible;
|
| 50 |
+
opacity: 1;
|
| 51 |
+
}
|
| 52 |
+
</style>
|
| 53 |
+
"""
|
| 54 |
+
st.markdown(html_content, unsafe_allow_html=True)
|
| 55 |
+
|
| 56 |
+
def javascript(source: str) -> None:
|
| 57 |
+
div_id = uuid.uuid4()
|
| 58 |
+
|
| 59 |
+
st.markdown(f"""
|
| 60 |
+
<div style="display:none" id="{div_id}">
|
| 61 |
+
<iframe src="javascript: \
|
| 62 |
+
var script = document.createElement('script'); \
|
| 63 |
+
script.type = 'text/javascript'; \
|
| 64 |
+
script.text = {html.escape(repr(source))}; \
|
| 65 |
+
var div = window.parent.document.getElementById('{div_id}'); \
|
| 66 |
+
div.appendChild(script); \
|
| 67 |
+
div.parentElement.parentElement.parentElement.style.display = 'none'; \
|
| 68 |
+
"/>
|
| 69 |
+
</div>
|
| 70 |
+
""", unsafe_allow_html=True)
|
| 71 |
+
|
| 72 |
+
|
| 73 |
+
###################################################################
|
| 74 |
+
|
| 75 |
+
|
| 76 |
+
if "urls_input" not in st.session_state:
|
| 77 |
+
st.session_state["urls_input"] = normalize_text(default_urls_input)
|
| 78 |
+
|
| 79 |
+
tab1, tab2 = st.tabs(["Danh sách các liên kết", "Chỉnh sửa danh sách các liên kết"])
|
| 80 |
+
|
| 81 |
+
|
| 82 |
+
with tab2:
|
| 83 |
+
edited_urls_input = st.text_area(
|
| 84 |
+
"Thêm hoặc bớt các liên kết",
|
| 85 |
+
value=st.session_state["urls_input"],
|
| 86 |
+
height=500,
|
| 87 |
+
key=f"edit_urls_input"
|
| 88 |
+
)
|
| 89 |
+
|
| 90 |
+
if st.button("Cập nhật thay đổi"):
|
| 91 |
+
st.session_state["urls_input"] = normalize_text(edited_urls_input)
|
| 92 |
+
st.session_state['selected_url'] = None
|
| 93 |
+
st.success("Đã lưu thay đổi!")
|
| 94 |
+
st.rerun()
|
| 95 |
+
|
| 96 |
+
|
| 97 |
+
with tab1:
|
| 98 |
+
# Giao diện Streamlit với 2 cột bằng nhau, chiếm trọn màn hình
|
| 99 |
+
col1, col2 = st.columns([3, 6])
|
| 100 |
+
|
| 101 |
+
urls = st.session_state["urls_input"].strip().split()
|
| 102 |
+
|
| 103 |
+
##############
|
| 104 |
+
with col1:
|
| 105 |
+
##############
|
| 106 |
+
# Hiển thị danh sách các URL và cho phép chọn
|
| 107 |
+
for idx, url in enumerate(urls):
|
| 108 |
+
url_preview = url[:50] + "..." # rút gọn
|
| 109 |
+
if st.button(url_preview, key = idx):
|
| 110 |
+
st.session_state['selected_url'] = url
|
| 111 |
+
|
| 112 |
+
|
| 113 |
+
##############
|
| 114 |
+
with col2:
|
| 115 |
+
##############
|
| 116 |
+
if 'selected_url' not in st.session_state or st.session_state["selected_url"] is None:
|
| 117 |
+
"Nhấn vào url để hiển thị nội dung"
|
| 118 |
+
|
| 119 |
+
else:
|
| 120 |
+
selected_url = st.session_state['selected_url']
|
| 121 |
+
fresh_text = url_content(selected_url)
|
| 122 |
+
_, chunks = add_chunk_markers(fresh_text, para = True)
|
| 123 |
+
|
| 124 |
+
# Hiển thị url
|
| 125 |
+
st.markdown(selected_url)#, help="tooltip")
|
| 126 |
+
|
| 127 |
+
# Hiển thị summary và nhận xét
|
| 128 |
+
text = url_content(selected_url) # tuy ko dùng nhưng phải lấy text trước thì mới có cái để summ
|
| 129 |
+
llm_generated = get_llm_gen_contents(selected_url)
|
| 130 |
+
|
| 131 |
+
if "summary" in llm_generated:
|
| 132 |
+
summ = llm_generated["summary"]
|
| 133 |
+
|
| 134 |
+
# Hiển thị phần tóm tắt văn bản
|
| 135 |
+
summ = re.sub(r'<cite>((?:\[\d+\])+)</cite>', r' \1', summ)
|
| 136 |
+
summ_sub = rf'<strong><a href="#" sourceid="chunk_\1" target="" class="cite tooltip">[\1]\
|
| 137 |
+
<span class="tooltiptext">__chunk_\1_text__</span></a></strong>'
|
| 138 |
+
|
| 139 |
+
summ = re.sub(r'\[(\d+)\]', summ_sub, summ)
|
| 140 |
+
for idx, chunk in enumerate(chunks):
|
| 141 |
+
chunk = re.sub(r'https?://', '', chunk, flags = re.IGNORECASE | re.MULTILINE)
|
| 142 |
+
summ = summ.replace(f"__chunk_{idx}_text__", f"{chunk}")
|
| 143 |
+
|
| 144 |
+
st.write(summ, unsafe_allow_html = True)
|
| 145 |
+
|
| 146 |
+
|
| 147 |
+
tab1, tab2, tab3, tab4, tab5 = st.tabs([
|
| 148 |
+
"Xem nội dung",
|
| 149 |
+
"Bản tinh gọn",
|
| 150 |
+
"Chỉnh sửa nội dung",
|
| 151 |
+
"Cào lại nội dung",
|
| 152 |
+
"Xóa nội dung",
|
| 153 |
+
])
|
| 154 |
+
|
| 155 |
+
|
| 156 |
+
######################
|
| 157 |
+
with tab1:
|
| 158 |
+
######################
|
| 159 |
+
marked_chunks = [ f'<span id="chunk_{i}" class="source">{c}</span>' for i, c in enumerate(chunks) ]
|
| 160 |
+
|
| 161 |
+
content = "\n\n".join(marked_chunks)
|
| 162 |
+
|
| 163 |
+
st.write(content, unsafe_allow_html = True)
|
| 164 |
+
|
| 165 |
+
# Nhúng javascript sau cùng để hilite citing
|
| 166 |
+
# https://discuss.streamlit.io/t/injecting-js/22651/6
|
| 167 |
+
javascript("""
|
| 168 |
+
function removeHilite() {
|
| 169 |
+
document.querySelectorAll('.hilite').forEach(element => {
|
| 170 |
+
element.classList.remove('hilite');
|
| 171 |
+
element.setAttribute("style", "background: #FFFFFF;")
|
| 172 |
+
});
|
| 173 |
+
}
|
| 174 |
+
|
| 175 |
+
setTimeout(() => {
|
| 176 |
+
|
| 177 |
+
console.log("BẮT ĐẦU NHÚNG JS");
|
| 178 |
+
removeHilite();
|
| 179 |
+
|
| 180 |
+
document.querySelectorAll('.cite').forEach(element => {
|
| 181 |
+
|
| 182 |
+
console.log('Element:', element);
|
| 183 |
+
element.addEventListener('click', () => {
|
| 184 |
+
|
| 185 |
+
let chunkId = element.getAttribute("sourceid");
|
| 186 |
+
removeHilite();
|
| 187 |
+
|
| 188 |
+
let hiliteElem = document.getElementById(chunkId);
|
| 189 |
+
console.log('Element to hilite:', chunkId, hiliteElem);
|
| 190 |
+
|
| 191 |
+
hiliteElem.classList.add('hilite');
|
| 192 |
+
hiliteElem.setAttribute("style", "background: #eadef9;")
|
| 193 |
+
|
| 194 |
+
// Scroll into view
|
| 195 |
+
hiliteElem.scrollIntoView({ behavior: "smooth", block: "center" });
|
| 196 |
+
});
|
| 197 |
+
});
|
| 198 |
+
|
| 199 |
+
}, 200);
|
| 200 |
+
""".strip())
|
| 201 |
+
|
| 202 |
+
|
| 203 |
+
######################
|
| 204 |
+
with tab2:
|
| 205 |
+
######################
|
| 206 |
+
clean_view = get_clean_view(selected_url)
|
| 207 |
+
if clean_view is None:
|
| 208 |
+
st.write("Hệ thống đang tạo nội dung, vui lòng chờ ...")
|
| 209 |
+
|
| 210 |
+
if st.button("Reload"):
|
| 211 |
+
st.rerun()
|
| 212 |
+
else:
|
| 213 |
+
st.write(clean_view)
|
| 214 |
+
|
| 215 |
+
if st.button("Dùng bản tinh gọn làm nội dung"):
|
| 216 |
+
url_content(selected_url, update_text = clean_view)
|
| 217 |
+
st.success("Đã lưu thay đổi!")
|
| 218 |
+
st.rerun()
|
| 219 |
+
|
| 220 |
+
|
| 221 |
+
######################
|
| 222 |
+
with tab3:
|
| 223 |
+
######################
|
| 224 |
+
fresh_text = url_content(selected_url)
|
| 225 |
+
|
| 226 |
+
edited_text = st.text_area(
|
| 227 |
+
f"*Lưu ý: Văn bản không được dài quá {config.text_max_words} từ, nếu quá sẽ tự động cắt bỏ*",
|
| 228 |
+
value =fresh_text,
|
| 229 |
+
height = 500,
|
| 230 |
+
key = f"edit_{selected_url}"
|
| 231 |
+
)
|
| 232 |
+
|
| 233 |
+
if st.button("Lưu thay đổi"):
|
| 234 |
+
url_content(selected_url, update_text = edited_text)
|
| 235 |
+
st.success("Đã lưu thay đổi!")
|
| 236 |
+
st.rerun()
|
| 237 |
+
|
| 238 |
+
|
| 239 |
+
######################
|
| 240 |
+
with tab4:
|
| 241 |
+
######################
|
| 242 |
+
if st.button("Bạn chắc chắn muốn cào lại nội dung?"):
|
| 243 |
+
reset_content(selected_url)
|
| 244 |
+
st.rerun()
|
| 245 |
+
|
| 246 |
+
|
| 247 |
+
######################
|
| 248 |
+
with tab5:
|
| 249 |
+
######################
|
| 250 |
+
if st.button("Bạn chắc chắn muốn xóa nội dung?"):
|
| 251 |
+
urls_input = st.session_state["urls_input"]
|
| 252 |
+
urls_input = urls_input.replace(selected_url, "")
|
| 253 |
+
|
| 254 |
+
st.session_state["urls_input"] = normalize_text(urls_input)
|
| 255 |
+
st.session_state['selected_url'] = None
|
| 256 |
+
|
| 257 |
+
st.success("Đã xóa nội dung!")
|
| 258 |
+
st.rerun()
|
requirements.txt
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
google-generativeai
|
| 2 |
+
together
|
| 3 |
+
|
| 4 |
+
streamlit
|
| 5 |
+
trafilatura
|
| 6 |
+
markdown
|
| 7 |
+
selenium
|