Upload ai_runtime_patch_fast.py

ai_runtime_patch_fast.py

@@ -1,4 +1,5 @@
-"""Final patch v2: fix topic rewrite, remove duplicate short slide, full short interaction buttons."""
+"""Final patch v2: fix topic rewrite, remove duplicate short slide, full short interaction buttons.
+FIX: rewrite endpoint now includes slides with images + auto short video trigger."""
 import re, threading, time, json, os, asyncio
 import ai_runtime_final6 as f6
 from ai_runtime_final6 import app, rt, f5, HTMLResponse, JSONResponse, Request, Query
@@ -27,6 +28,35 @@ def _cleanup():
 def _scrape(url,mc=8000):
     try:d=f5.base.scrape_any_url(url);return(d.get('title',''),((d.get('summary','')+'\n'+d.get('text','')).strip())[:mc],d.get('image') or d.get('og_image') or '')
     except:return('','','')
+def _scrape_full(url):
+    """Get title, text, images ALL from scrape_any_url."""
+    try:
+        d = f5.base.scrape_any_url(url)
+        if d:
+            imgs = d.get('images', [])
+            if not imgs and d.get('og_image'):
+                imgs = [d.get('og_image', '')]
+            return d.get('title', ''), d.get('text', ''), imgs, d.get('og_image', '')
+    except:
+        pass
+    return '', '', [], ''
+def _make_slides(text, images, og_img, max_slides=8):
+    """Build slides array from text paragraphs + images."""
+    slides = []
+    paragraphs = [p.strip() for p in text.split('\n') if len(p.strip()) > 60]
+    all_imgs = images[:] if images else []
+    if not all_imgs and og_img:
+        all_imgs = [og_img]
+    if not all_imgs:
+        all_imgs = ['https://image.pollinations.ai/prompt/Vietnamese%20news%20update']
+    for i, pt in enumerate(paragraphs[:max_slides]):
+        slide_img = all_imgs[i] if i < len(all_imgs) else all_imgs[-1]
+        slides.append({'text': pt, 'image': slide_img, 'index': i + 1})
+    if not slides:
+        for i, im in enumerate(all_imgs[:3]):
+            slides.append({'text': '', 'image': im, 'index': i + 1})
+    return slides
+
 _bg_home={"t":0,"d":[]};_bg_shorts={"t":0,"d":[]};_bg_lock=False
 def _bg():
     global _bg_lock
@@ -75,6 +105,8 @@ async def _ask(request:Request):
     if not raw:raw=ctx[:12000]
     ans=await f5.base.qwen_generate(f'Bạn là VNEWS AI. Nội dung: "{title}"\n{raw[:9000]}\n\nHỏi: "{q}"\n\nTrả lời tự nhiên bằng tiếng Việt.',max_tokens=1200)
     return JSONResponse({'answer':ans or 'Chưa trả lời được.','title':title})
+
+# ===== REWRITE with SLIDES + IMAGES + SHORT =====
 @app.post('/api/rewrite_share')
 @app.post('/api/url_wall')
 async def _rw(request:Request):
@@ -83,12 +115,41 @@ async def _rw(request:Request):
     title,raw,img=_scrape(url,14000)
     if len(raw)<50:raw=ctx[:14000]
     if len(raw)<50:return JSONResponse({'error':'Không đọc được bài'},status_code=422)
+    
+    # Get full article data with images
+    full_title, full_text, all_images, og_img = _scrape_full(url)
+    if not all_images and img:
+        all_images = [img]
+    if not all_images and og_img:
+        all_images = [og_img]
+    if not all_images:
+        all_images = ['https://image.pollinations.ai/prompt/Vietnamese%20news%20update']
+    
+    # Generate AI summary
     text=None
     try:text=await asyncio.wait_for(f5.base.qwen_generate(f'Tóm tắt đăng Tường AI:\nTiêu đề: {title}\n{raw[:14000]}\n\n4-6 ý chính. Cuối ghi nguồn.',image_url=img or None,max_tokens=1000),timeout=30)
     except:pass
     if not text or len(text)<80:text=f"Tóm tắt: {title}\n\n{raw[:1200]}\n\nNguồn: {_domain(url)}"
-    post=f5.base.make_post(title or 'Bài viết',text,img,url,'rewrite',sources=[{'title':title,'url':url,'via':_domain(url)}])
-    ps=f5.base._load_ai_wall();ps.insert(0,post);f5.base._save_ai_wall(ps);return JSONResponse({'post':post})
+    
+    # Build slides from full text
+    slides = _make_slides(full_text or raw, all_images, og_img)
+    
+    main_img = all_images[0] if all_images else ''
+    post=f5.base.make_post(title or 'Bài viết',text,main_img,url,'rewrite',sources=[{'title':title,'url':url,'via':_domain(url)}])
+    post['images'] = all_images
+    post['slides'] = slides
+    ps=f5.base._load_ai_wall();ps.insert(0,post);f5.base._save_ai_wall(ps)
+    
+    # Try generate short video in background
+    if len(text) > 100:
+        try:
+            import auto_scheduler
+            threading.Thread(target=lambda: auto_scheduler._try_generate_short(post), daemon=True).start()
+        except:
+            pass
+    
+    return JSONResponse({'post':post})
+
 @app.post('/api/topic/rewrite')
 async def _tr(request:Request):
     b=await request.json();pid=str(b.get('post_id','')).strip()
@@ -185,4 +246,4 @@ async def _index():
     body=getattr(rt.old,'PATCH_INJECT','')+f5.f4.f3.f2.f1.FINAL_INJECT+f5.f4.f3.FINAL3_INJECT+f5.f4.FINAL4_INJECT+f5.FINAL5_INJECT
     body+=getattr(f6,'FINAL6_INJECT','');body+=getattr(f6,'FINAL6_FAST_HOME_INJECT','');body+=getattr(f6,'FINAL6E_INJECT','')
     body+=PATCH_INJECT
-    return HTMLResponse(html.replace('</body>',body+'\n</body>') if '</body>' in html else html+body)
+    return HTMLResponse(html.replace('</body>',body+'\n</body>') if '</body>' in html else html+body)