Background prefetch on startup + instant cache response for homepage speed"

ai_runtime_patch_fast.py

@@ -1,14 +1,97 @@
-"""Patch over b9b6ba4 final6: faster home, source thumbnails, rewrite button, in-app source links, topic quality."""
-import re
+"""Patch over b9b6ba4 final6: faster home via background prefetch, source thumbnails, rewrite button, in-app links, topic quality."""
+import re, threading, time
 import ai_runtime_final6 as f6
 from ai_runtime_final6 import app, rt, f5, HTMLResponse, JSONResponse, Request, Query
 import html as html_lib
 
 def clean(s):return re.sub(r"\s+"," ",html_lib.unescape(str(s or ""))).strip()
 
-# Override topic_post to produce cleaner, more focused content.
-app.router.routes=[r for r in app.router.routes if not ((getattr(r,'path',None)=='/api/topic_post' and 'POST' in getattr(r,'methods',set())) or (getattr(r,'path',None)=='/' and 'GET' in getattr(r,'methods',set())))]
+# ===== BACKGROUND PREFETCH ON STARTUP =====
+_bg_home_cache={"t":0,"d":[]}
+_bg_shorts_cache={"t":0,"d":[]}
+_bg_loading=False
 
+def _bg_refresh():
+    global _bg_loading
+    if _bg_loading:return
+    _bg_loading=True
+    try:
+        # Homepage via fast RSS (already defined in f6 as _fast_homepage)
+        if hasattr(f6,'_fast_homepage'):
+            data=f6._fast_homepage()
+            if data:_bg_home_cache.update({"t":time.time(),"d":data})
+        # Shorts
+        if hasattr(f6,'api_shorts_final6'):
+            # Call internal function directly
+            raw=[]
+            for h in f6.YOUTUBE_HANDLES:
+                raw.extend(f6._yt_ytdlp(h,20) or f6._yt_html(h,20))
+            raw.extend(f6._fallback_shorts())
+            seen=set();out=[]
+            for v in raw:
+                vid=v.get('id') or ''
+                if not vid:continue
+                if vid in seen:continue
+                seen.add(vid);out.append(v)
+            if out:_bg_shorts_cache.update({"t":time.time(),"d":out[:40]})
+    except Exception:pass
+    finally:_bg_loading=False
+
+def _start_bg_prefetch():
+    t=threading.Thread(target=_bg_refresh,daemon=True)
+    t.start()
+
+@app.on_event("startup")
+async def startup_prefetch():
+    _start_bg_prefetch()
+
+# Periodic background refresh every 10 minutes.
+def _periodic_refresh():
+    while True:
+        time.sleep(600)
+        try:_bg_refresh()
+        except Exception:pass
+
+_periodic_thread=threading.Thread(target=_periodic_refresh,daemon=True)
+_periodic_thread.start()
+
+# ===== OVERRIDE HOMEPAGE & SHORTS FOR INSTANT RESPONSE =====
+app.router.routes=[r for r in app.router.routes if not (
+    (getattr(r,'path',None)=='/api/homepage' and 'GET' in getattr(r,'methods',set())) or
+    (getattr(r,'path',None)=='/api/shorts' and 'GET' in getattr(r,'methods',set())) or
+    (getattr(r,'path',None)=='/api/topic_post' and 'POST' in getattr(r,'methods',set())) or
+    (getattr(r,'path',None)=='/' and 'GET' in getattr(r,'methods',set()))
+)]
+
+@app.get('/api/homepage')
+def api_homepage_instant():
+    # Return cache immediately; if stale, trigger background refresh.
+    now=time.time()
+    if _bg_home_cache['d']:
+        if now-_bg_home_cache['t']>300:
+            threading.Thread(target=_bg_refresh,daemon=True).start()
+        return JSONResponse(_bg_home_cache['d'])
+    # First call, no cache yet: try fast RSS inline (fast, 4s timeout each).
+    if hasattr(f6,'_fast_homepage'):
+        data=f6._fast_homepage()
+        if data:
+            _bg_home_cache.update({"t":now,"d":data})
+            return JSONResponse(data)
+    return JSONResponse([])
+
+@app.get('/api/shorts')
+def api_shorts_instant(refresh:int=Query(default=0)):
+    now=time.time()
+    if _bg_shorts_cache['d'] and (not refresh or now-_bg_shorts_cache['t']<120):
+        if now-_bg_shorts_cache['t']>600:
+            threading.Thread(target=_bg_refresh,daemon=True).start()
+        return JSONResponse(_bg_shorts_cache['d'])
+    # Inline fallback if no cache.
+    if hasattr(f6,'api_shorts_final6'):
+        return f6.api_shorts_final6(refresh=refresh)
+    return JSONResponse([])
+
+# ===== TOPIC POST (focused, no irrelevant content) =====
 @app.post('/api/topic_post')
 async def topic_post_focused(request:Request):
     body=await request.json();topic=clean(body.get('topic',''))
@@ -51,33 +134,33 @@ QUY TẮC BẮT BUỘC:
     posts=f5.base._load_ai_wall();posts.insert(0,post);f5.base._save_ai_wall(posts)
     return JSONResponse({'post':post,'sources_count':len(details or sources)})
 
+# ===== FRONTEND PATCH =====
 PATCH_INJECT=r'''
 <script>
 (function(){
 function esc(s){return String(s||'').replace(/[&<>"']/g,m=>({'&':'&amp;','<':'&lt;','>':'&gt;','"':'&quot;',"'":'&#39;'}[m]));}
-// 1) Faster home: defer hot_topics and shorts refresh to after main content loads.
+// 1) Defer non-critical fetches.
 let _hotLoaded=false;
-function deferHotTopics(){if(_hotLoaded)return;_hotLoaded=true;setTimeout(()=>{if(typeof ensureHotTopics==='function')ensureHotTopics();if(typeof ensureNewsShortsHome==='function')ensureNewsShortsHome();},3000);}
+function deferHotTopics(){if(_hotLoaded)return;_hotLoaded=true;setTimeout(()=>{if(typeof ensureHotTopics==='function')ensureHotTopics();if(typeof ensureNewsShortsHome==='function')ensureNewsShortsHome();},4000);}
 if(document.readyState==='complete')deferHotTopics();else window.addEventListener('load',deferHotTopics);
 
-// 2) Source links in topic posts: open in-app via readArticle instead of external link.
+// 2) Source links open in-app.
 function patchSourceLinksInApp(){document.querySelectorAll('.source-detail-item a[href]').forEach(a=>{if(a.dataset.inapp)return;a.dataset.inapp='1';let url=a.href;a.removeAttribute('target');a.onclick=function(e){e.preventDefault();if(typeof readArticle==='function')readArticle(url);else window.open(url,'_blank');};a.textContent='📖 Đọc trên VNEWS';a.style.cursor='pointer';});}
 
 // 3) Source items show thumbnail.
 function patchTopicSourceImages(){document.querySelectorAll('.source-detail-item').forEach(el=>{if(el.dataset.imgPatched)return;el.dataset.imgPatched='1';let a=el.querySelector('a[href]');if(!a||!a.href)return;let url=a.getAttribute('href')||a.href;fetch('/api/article?url='+encodeURIComponent(url)).then(r=>r.json()).then(d=>{if(d&&(d.og_image||d.img)){let img=document.createElement('img');img.style.cssText='width:100%;aspect-ratio:16/9;object-fit:cover;border-radius:6px;margin-bottom:6px;background:#222';img.loading='lazy';img.src=d.og_image||d.img;img.onerror=function(){this.style.display='none'};el.prepend(img);}}).catch(()=>{});});}
 
-// 4) Add rewrite button to every article view.
+// 4) Add rewrite button.
 function addRewriteButton(){let art=document.querySelector('#view-article .article-view');if(!art||art.querySelector('.rewrite-injected'))return;let actions=art.querySelector('.article-actions');if(!actions){actions=document.createElement('div');actions.className='article-actions';art.appendChild(actions);}if(actions.querySelector('[data-rewrite]'))return;let btn=document.createElement('button');btn.className='primary rewrite-injected';btn.setAttribute('data-rewrite','1');btn.textContent='🤖 AI viết lại & đăng tường';btn.onclick=function(){if(typeof rewriteCurrentArticle==='function')rewriteCurrentArticle();else alert('Chức năng rewrite chưa sẵn sàng.')};actions.insertBefore(btn,actions.firstChild);}
 
-// 5) Patch readLiveTopicWall to add rewrite + in-app source links.
+// 5) Patch readLiveTopicWall.
 let oldReadLive=window.readLiveTopicWall;
 if(oldReadLive){window.readLiveTopicWall=function(){let ret=oldReadLive.apply(this,arguments);setTimeout(()=>{addRewriteButton();patchSourceLinksInApp();patchTopicSourceImages();},600);return ret;}}
 
-// 6) Also patch readArticle to add rewrite after render.
+// 6) Patch readArticle.
 let oldReadArticle=window.readArticle;
 if(oldReadArticle){window.readArticle=async function(){let ret=await oldReadArticle.apply(this,arguments);setTimeout(addRewriteButton,600);return ret;}}
 
-// Run periodically.
 setInterval(()=>{addRewriteButton();patchSourceLinksInApp();patchTopicSourceImages();},1500);
 setTimeout(addRewriteButton,800);
 })();