Upload vtv_api.py with huggingface_hub

vtv_api.py

@@ -69,7 +69,7 @@ FPTPLAY_URLS = {
     "vtv3":  "https://live.fptplay53.net/fnxch2/vtv3hd_abr.smil/chunklist.m3u8",
     "vtv4":  "https://live.fptplay53.net/fnxch2/vtv4hd_abr.smil/chunklist.m3u8",
     "vtv5":  "https://live-a.fptplay53.net/live/media/VTV5HD/live_hls_avc/index.m3u8",
-    "vtv6":  "https://live.fptplay53.net/fnxhd1/vtv6hd_vhls.smil/chunklist.m3u8",
+    "vtv6":  "https://live.fptplay53.net/fnxch2/vtv6hd_abr.smil/chunklist.m3u8",
     "vtv7":  "https://live.fptplay53.net/fnxhd1/vtv7hd_vhls.smil/chunklist_b5000000.m3u8",
     "vtv8":  "https://live.fptplay53.net/epzhd1/vtv8hd_vhls.smil/chunklist.m3u8",
     "vtv9":  "https://live.fptplay53.net/fnxhd1/vtv9hd_vhls.smil/chunklist.m3u8",
@@ -132,6 +132,46 @@ def fetch_fptplay_stream(channel_id):
         pass
     return None
 
+def fetch_vtv_streams(channel_id):
+    """Fetch all available stream URLs for a channel (ordered by priority)."""
+    urls = []
+    channel_id = channel_id.lower().strip()
+    name_map = {
+        'vtvct': 'vtv10', 'vtv-can-tho': 'vtv10', 'vtv can tho': 'vtv10',
+        'vtv_can_tho': 'vtv10', 'cantho': 'vtv10',
+        'vietnam_vtv1': 'vtv1', 'vietnam_vtv2': 'vtv2', 'vietnam_vtv3': 'vtv3',
+        'vietnam_vtv4': 'vtv4', 'vietnam_vtv5': 'vtv5', 'vietnam_vtv6': 'vtv6',
+        'vietnam_vtv7': 'vtv7', 'vietnam_vtv8': 'vtv8', 'vietnam_vtv9': 'vtv9',
+    }
+    channel_id = name_map.get(channel_id, channel_id)
+
+    # Collect all available sources
+    sources = [
+        ('vtvgo', lambda: VTVGO_FAILOVER.get(channel_id)),
+        ('fptplay', lambda: fetch_fptplay_stream(channel_id)),
+        ('xemtv', lambda: fetch_xemtv_stream(channel_id)),
+    ]
+
+    # For VTV6: skip VTVGo CDN first (may be misrouted), prefer fptplay, then VTVGo as last resort
+    if channel_id == 'vtv6':
+        sources = [
+            ('fptplay', lambda: fetch_fptplay_stream(channel_id)),
+            ('xemtv', lambda: fetch_xemtv_stream(channel_id)),
+            ('vtvgo', lambda: VTVGO_FAILOVER.get(channel_id)),
+        ]
+
+    seen = set()
+    for name, fetch_fn in sources:
+        try:
+            url = fetch_fn()
+            if url and url not in seen:
+                seen.add(url)
+                urls.append(url)
+        except:
+            pass
+
+    return urls
+
 def fetch_vtv_stream(channel_id):
     """Fetch VTV stream with multi-source fallback chain:
     1. VTVGo CDN (primary, works for most channels)
@@ -161,18 +201,26 @@ def fetch_vtv_stream(channel_id):
         _set_cache(channel_id, None)
         return None
 
-    # Source 1: VTVGo CDN (primary for channels that have it)
-    vtvgourl = VTVGO_FAILOVER.get(channel_id)
-    if vtvgourl:
-        _set_cache(channel_id, vtvgourl)
-        return vtvgourl
+    # Source 1: VTVGo CDN (primary for most channels)
+    if channel_id != 'vtv6':  # VTV6 skip VTVGo CDN due to CDN misrouting
+        vtvgourl = VTVGO_FAILOVER.get(channel_id)
+        if vtvgourl:
+            _set_cache(channel_id, vtvgourl)
+            return vtvgourl
 
-    # Source 2: fptplay CDN (primary for VTV10, fallback for others)
+    # Source 2: fptplay CDN (primary for VTV6/VTV10, fallback for others)
     fpt_url = fetch_fptplay_stream(channel_id)
     if fpt_url:
         _set_cache(channel_id, fpt_url)
         return fpt_url
 
+    # Source 3: VTVGo CDN for VTV6 as last resort
+    if channel_id == 'vtv6':
+        vtvgourl = VTVGO_FAILOVER.get(channel_id)
+        if vtvgourl:
+            _set_cache(channel_id, vtvgourl)
+            return vtvgourl
+
     # Source 3: xemtv.net PHP scraper (last resort, often unreachable)
     xemtv_url = fetch_xemtv_stream(channel_id)
     if xemtv_url:
@@ -186,15 +234,15 @@ def fetch_vtv_stream(channel_id):
 def api_vtv_streams():
     result = {}
     for ch_id in CHANNEL_NAMES:
-        stream_url = fetch_vtv_stream(ch_id)
-        result[ch_id] = {"name": CHANNEL_NAMES[ch_id], "stream_url": stream_url, "status": "ok" if stream_url else "offline"}
+        stream_urls = fetch_vtv_streams(ch_id)
+        result[ch_id] = {"name": CHANNEL_NAMES[ch_id], "stream_urls": stream_urls, "stream_url": stream_urls[0] if stream_urls else None, "status": "ok" if stream_urls else "offline"}
     return JSONResponse(result)
 
 @router.get("/api/vtv/stream/{channel_id}")
 def api_vtv_stream(channel_id: str):
-    stream_url = fetch_vtv_stream(channel_id)
-    if stream_url:
-        return JSONResponse({"stream_url": stream_url, "status": "ok"})
+    stream_urls = fetch_vtv_streams(channel_id)
+    if stream_urls:
+        return JSONResponse({"stream_urls": stream_urls, "stream_url": stream_urls[0], "status": "ok"})
     return JSONResponse({"error": "stream not found", "status": "offline"}, status_code=404)
 
 @router.get("/api/proxy/page")