sync ms

app.py

@@ -1,17 +1,30 @@
 import gradio as gr
-from tiktok import tiktok_parser
-from bili import bili_parser
-from bvid2acid import bv2acid
+from modules.tiktok import tiktok_parser
+from modules.bili import bili_parser
+from modules.bvid2acid import bv2acid
+from utils import LANG
+
+ZH2EN = {
+    "本站不提供任何视频存储服务，仅提供最基本的解析服务，请勿滥用": "This site does not provide any audio storage services, only provide the most basic parsing services, please DO NOT abuse",
+    "抖音": "Tiktok",
+    "B站": "Bilibili",
+}
+
+
+def _L(zh_txt: str):
+    if LANG:
+        return ZH2EN[zh_txt]
+    else:
+        return zh_txt
+
 
 if __name__ == "__main__":
     with gr.Blocks() as demo:
-        gr.Markdown(
-            "This site does not provide any video storage services, only to provide the most basic resolution services, please DO NOT abuse"
-        )
-        with gr.Tab("Tiktok"):
+        gr.Markdown(_L("本站不提供任何视频存储服务，仅提供最基本的解析服务，请勿滥用"))
+        with gr.Tab(_L("抖音")):
             tiktok_parser()
 
-        with gr.Tab("Bilibili"):
+        with gr.Tab(_L("B站")):
             with gr.Column():
                 bv2acid()
                 bili_parser()

config.py

@@ -1,11 +0,0 @@
-import os
-
-TIMEOUT = None
-TMP_DIR = "./__pycache__"
-
-API_BILI = os.getenv("api_bili")
-API_TIKTOK = os.getenv("api_tiktok")
-
-HEADER = {
-    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/53.0.2785.116 Safari/537.36"
-}

bili.py → modules/bili.py

@@ -1,31 +1,28 @@
 import re
 import requests
 import gradio as gr
-from tqdm import tqdm
-from utils import timestamp, clean_dir
-from config import TMP_DIR, HEADER, TIMEOUT, API_BILI
-
-
-def download_file(url, video_id, cache_dir=f"{TMP_DIR}/bili"):
-    clean_dir(cache_dir)
-    local_file = f"{cache_dir}/{video_id}.mp4"
-    response = requests.get(url, stream=True)
-    if response.status_code == 200:
-        total_size = int(response.headers.get("Content-Length", 0)) + 1
-        time_stamp = timestamp()
-        progress_bar = tqdm(
-            total=total_size,
-            unit="B",
-            unit_scale=True,
-            desc=f"[{time_stamp}] {local_file}",
-        )
-        with open(local_file, "wb") as f:
-            for chunk in response.iter_content(chunk_size=8192):
-                if chunk:
-                    f.write(chunk)
-                    progress_bar.update(len(chunk))
-
-    return local_file
+from utils import download_file, HEADER, TIMEOUT, API_BILI_2, API_BILI_1, LANG, TMP_DIR
+
+ZH2EN = {
+    "状态栏": "Status",
+    "请输入B站视频链接": "Please input Bilibili video link",
+    "分P": "Part",
+    "视频标题": "Video title",
+    "视频封面": "Video cover",
+    "视频下载": "Download video",
+    "视频简介": "Video introduction",
+    "视频时长(s)": "Video duration(s)",
+    "UP主头像": "Uploader avatar",
+    "UP主昵称": "Uploader nickname",
+    "B站视频解析": "Bilibili video parser",
+}
+
+
+def _L(zh_txt: str):
+    if LANG:
+        return ZH2EN[zh_txt]
+    else:
+        return zh_txt
 
 
 def extract_fst_url(text):
@@ -51,42 +48,80 @@ def get_real_url(short_url):
     ).url.split("/?")[0]
 
 
+def get_video(video_url: str, p: int, cache=f"{TMP_DIR}/bili"):
+    bvid = video_url.split("bilibili.com/video/")[-1]
+    response = requests.get(
+        API_BILI_2,
+        params={
+            "bv": bvid,
+            "p": p,
+            "otype": "json",
+        },
+        timeout=TIMEOUT,
+    )
+    response_json = response.json()
+    retcode = response_json["code"]
+    if retcode == 0:
+        qualities: list = response_json["accept_quality"]
+        quality = min(qualities[0], 80)
+        url = requests.get(
+            API_BILI_2,
+            params={
+                "bv": bvid,
+                "p": p,
+                "q": quality,
+                "otype": "url",
+            },
+            timeout=TIMEOUT,
+        ).text
+        return download_file(url, bvid, cache)
+
+    else:
+        raise ConnectionError(f"Failed to get video: {retcode}")
+
+
 # outer func
 def infer(video_url: str, p: int):
+    status = "Success"
     title = cover = desc = dur = video = author = avatar = None
-    if not video_url:
-        title = "Empty video link!"
-        return title, cover, video, desc, dur, avatar, author
-
     try:
+        if not video_url:
+            raise ValueError("视频链接为空!")
+
         video_url = extract_fst_url(video_url)
         if "b23.tv" in video_url:
             video_url = get_real_url(video_url)
 
-        response = requests.get(API_BILI, params={"url": video_url}, timeout=TIMEOUT)
+        response = requests.get(
+            API_BILI_1,
+            params={
+                "url": video_url,
+                "type": "json",
+            },
+            timeout=TIMEOUT,
+        )
         response_json = response.json()
         retcode = response_json["code"]
-        if retcode == 1:
+        if retcode == 200:
             title = response_json["title"]
             cover = response_json["imgurl"]
             desc = response_json["desc"]
 
             response_data = response_json["data"][int(p) - 1]
             dur = response_data["duration"]
-            video_id = video_url.split("/")[-1]
-            video = download_file(response_data["video_url"], video_id)
+            video = get_video(video_url, p)
 
             author_data = response_json["user"]
             author = author_data["name"]
             avatar = author_data["user_img"]
 
         else:
-            raise ConnectionError(f"Failed to call API, error code: {retcode}")
+            raise ConnectionError(f"接口调用失败, 错误码: {retcode}")
 
     except Exception as e:
-        title = f"Failed to parse video: {e}"
+        status = f"视频解析失败: {e}"
 
-    return title, cover, video, desc, dur, avatar, author
+    return status, title, cover, video, desc, dur, avatar, author
 
 
 def bili_parser():
@@ -94,30 +129,36 @@ def bili_parser():
         fn=infer,
         inputs=[
             gr.Textbox(
-                label="Please input Bilibili video link",
+                label=_L("请输入B站视频链接"),
                 placeholder="https://www.bilibili.com/video/*",
             ),
-            gr.Slider(label="Part", minimum=1, maximum=1000, step=1, value=1),
+            gr.Slider(label=_L("分P"), minimum=1, maximum=1000, step=1, value=1),
         ],
         outputs=[
-            gr.Textbox(label="Video title", show_copy_button=True),
-            gr.Image(label="Video cover", show_share_button=False),
+            gr.Textbox(label=_L("状态栏"), show_copy_button=True),
+            gr.Textbox(label=_L("视频标题"), show_copy_button=True),
+            gr.Image(label=_L("视频封面"), show_share_button=False),
             gr.Video(
-                label="Download video",
+                label=_L("视频下载"),
                 show_download_button=True,
                 show_share_button=False,
+                format="mp4",
             ),
-            gr.TextArea(label="Video introduction", show_copy_button=True),
-            gr.Textbox(label="Video duration(s)", show_copy_button=True),
-            gr.Image(label="Uploader avatar", show_share_button=False),
-            gr.Textbox(label="Uploader nickname", show_copy_button=True),
+            gr.TextArea(label=_L("视频简介"), show_copy_button=True),
+            gr.Textbox(label=_L("视频时长(s)"), show_copy_button=True),
+            gr.Image(label=_L("UP主头像"), show_share_button=False),
+            gr.Textbox(label=_L("UP主昵称"), show_copy_button=True),
         ],
-        title="Bilibili video parser",
+        title=_L("B站视频解析"),
         flagging_mode="never",
         examples=[
-            ["BV1G8iRYBE4f", 1],
-            ["https://b23.tv/LSoJzpW", 1],
-            ["https://www.bilibili.com/video/BV1G8iRYBE4f", 1],
+            ["BV1Dt4y1o7bU", 1],
+            ["https://b23.tv/LuTAbzj", 1],
+            ["https://www.bilibili.com/video/BV1Dt4y1o7bU", 1],
+            [
+                "【『新年おめでとう | 2024』你就如同烟花一般，在我心中绽放！-哔哩哔哩】 https://b23.tv/LuTAbzj",
+                1,
+            ],
         ],
         cache_examples=False,
     )

bvid2acid.py → modules/bvid2acid.py

@@ -1,32 +1,48 @@
 import requests
 import gradio as gr
+from utils import LANG, HEADER
+
+ZH2EN = {
+    "状态栏": "Status",
+    "将 Bvid 转为 aid 或 cid": "Bvid to aid / cid",
+}
+
+
+def _L(zh_txt: str):
+    if LANG:
+        return ZH2EN[zh_txt]
+    else:
+        return zh_txt
 
 
-# outer func
 def infer(bvid: str):
+    status = "Success"
+    aid = cid = None
     try:
         response = requests.get(
             "https://api.bilibili.com/x/web-interface/view",
             params={"bvid": bvid},
-            headers={
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36 Edg/131.0.0.0"
-            },
+            headers=HEADER,
         )
         data = response.json()["data"]
-        return data["aid"], data["cid"]
+        aid = data["aid"]
+        cid = data["cid"]
 
     except Exception as e:
-        return "Failed to parse aid / cid", f"{e}"
+        status = f"{e}"
+
+    return status, aid, cid
 
 
 def bv2acid():
     return gr.Interface(
         fn=infer,
-        inputs=gr.Textbox(label="bvid", show_copy_button=True),
+        inputs=gr.Textbox(label="bvid"),
         outputs=[
+            gr.Textbox(label=_L("状态栏"), show_copy_button=True),
             gr.Textbox(label="aid", show_copy_button=True),
             gr.Textbox(label="cid", show_copy_button=True),
         ],
-        title="Bvid to aid / cid",
+        title=_L("将 Bvid 转为 aid 或 cid"),
         flagging_mode="never",
     )

tiktok.py → modules/tiktok.py

@@ -1,31 +1,26 @@
 import re
 import requests
 import gradio as gr
-from tqdm import tqdm
-from utils import timestamp, clean_dir
-from config import API_TIKTOK, TIMEOUT, TMP_DIR
+from utils import download_file, API_TIKTOK, TIMEOUT, LANG, TMP_DIR
 
+ZH2EN = {
+    "状态栏": "Status",
+    "请输入抖音视频分享短链接": "Please enter TikTok video sharing short link",
+    "视频下载": "Video download",
+    "视频描述": "Video description",
+    "解析耗时": "Parsing time",
+    "作者头像": "Author avatar",
+    "作者昵称": "Author nickname",
+    "作者签名": "Author signature",
+    "抖音无水印视频解析": "Parse TikTok video without watermark",
+}
 
-def download_file(url, video_id, cache_dir=f"{TMP_DIR}/tiktok"):
-    clean_dir(cache_dir)
-    local_file = f"{cache_dir}/{video_id}.mp4"
-    response = requests.get(url, stream=True)
-    if response.status_code == 200:
-        total_size = int(response.headers.get("Content-Length", 0)) + 1
-        time_stamp = timestamp()
-        progress_bar = tqdm(
-            total=total_size,
-            unit="B",
-            unit_scale=True,
-            desc=f"[{time_stamp}] {local_file}",
-        )
-        with open(local_file, "wb") as f:
-            for chunk in response.iter_content(chunk_size=8192):
-                if chunk:
-                    f.write(chunk)
-                    progress_bar.update(len(chunk))
 
-    return local_file
+def _L(zh_txt: str):
+    if LANG:
+        return ZH2EN[zh_txt]
+    else:
+        return zh_txt
 
 
 def extract_fst_url(text):
@@ -38,16 +33,16 @@ def extract_fst_url(text):
 
 
 # outer func
-def infer(video_url):
+def infer(video_url, cache=f"{TMP_DIR}/tiktok"):
+    status = "Success"
     video = parse_time = desc = avatar = author = sign = None
-    if not video_url:
-        desc = "The video sharing link is empty!"
-        return video, desc, parse_time, avatar, author, sign
-
     try:
+        if not video_url:
+            raise ValueError("视频分享链接为空!")
+
         video_url = extract_fst_url(video_url)
         if not video_url:
-            raise ValueError("Please enter a valid video sharing link!")
+            raise ValueError("请输入有效的视频分享链接!")
 
         response = requests.get(API_TIKTOK, params={"url": video_url}, timeout=TIMEOUT)
         response_json = response.json()
@@ -55,7 +50,7 @@ def infer(video_url):
         if retcode == 200:
             response_data = response_json["data"]
             video_id = response_data["play_url"].split("video_id=")[1].split("&")[0]
-            video = download_file(response_data["video_url"], video_id)
+            video = download_file(response_data["video_url"], video_id, cache)
             parse_time = response_data["parse_time"]
 
             additional_data = response_data["additional_data"][0]
@@ -65,12 +60,12 @@ def infer(video_url):
             sign = additional_data["signature"]
 
         else:
-            raise ConnectionError(f"Interface call failed, error code: HTTP {retcode}")
+            raise ConnectionError(f"接口调用失败, 错误码: HTTP {retcode}")
 
     except Exception as e:
-        desc = f"Video parsing failed: {e}"
+        status = f"视频解析失败: {e}"
 
-    return video, desc, parse_time, avatar, author, sign
+    return status, video, desc, parse_time, avatar, author, sign
 
 
 def tiktok_parser():
@@ -78,23 +73,24 @@ def tiktok_parser():
         fn=infer,
         inputs=[
             gr.Textbox(
-                label="Please enter TikTok video sharing short link",
+                label=_L("请输入抖音视频分享短链接"),
                 placeholder="https://v.douyin.com/*",
             ),
         ],
         outputs=[
+            gr.Textbox(label=_L("状态栏"), show_copy_button=True),
             gr.Video(
-                label="Video download",
+                label=_L("视频下载"),
                 show_download_button=True,
                 show_share_button=False,
             ),
-            gr.Textbox(label="Video description", show_copy_button=True),
-            gr.Textbox(label="Parsing time", show_copy_button=True),
-            gr.Image(label="Author avatar", show_share_button=False),
-            gr.Textbox(label="Author nickname", show_copy_button=True),
-            gr.TextArea(label="Author signature", show_copy_button=True),
+            gr.Textbox(label=_L("视频描述"), show_copy_button=True),
+            gr.Textbox(label=_L("解析耗时"), show_copy_button=True),
+            gr.Image(label=_L("作者头像"), show_share_button=False),
+            gr.Textbox(label=_L("作者昵称"), show_copy_button=True),
+            gr.TextArea(label=_L("作者签名"), show_copy_button=True),
         ],
-        title="Parse TikTok video without watermark",
+        title=_L("抖音无水印视频解析"),
         flagging_mode="never",
         examples=[
             "https://v.douyin.com/8FVe5DzarE0",

utils.py

@@ -1,17 +1,30 @@
 import os
 import shutil
+import requests
+from tqdm import tqdm
 from datetime import datetime
-from zoneinfo import ZoneInfo
-from tzlocal import get_localzone
 
 
-def timestamp(naive_time: datetime = None, target_tz=ZoneInfo("Asia/Shanghai")):
+LANG = os.getenv("language")
+API_TIKTOK = os.getenv("api_tiktok")
+API_BILI_1 = os.getenv("api_bili_1")
+API_BILI_2 = os.getenv("api_bili_2")
+if not (API_TIKTOK and API_BILI_1 and API_BILI_2):
+    print("请检查环境变量")
+    exit()
+
+TIMEOUT = None
+TMP_DIR = "./__pycache__"
+HEADER = {
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/53.0.2785.116 Safari/537.36"
+}
+
+
+def timestamp(naive_time: datetime = None):
     if not naive_time:
         naive_time = datetime.now()
 
-    local_tz = get_localzone()
-    aware_local = naive_time.replace(tzinfo=local_tz)
-    return aware_local.astimezone(target_tz).strftime("%Y-%m-%d %H:%M:%S")
+    return naive_time.strftime("%Y-%m-%d %H:%M:%S")
 
 
 def mk_dir(dirpath: str):
@@ -27,3 +40,28 @@ def rm_dir(dirpath: str):
 def clean_dir(dirpath: str):
     rm_dir(dirpath)
     os.makedirs(dirpath)
+
+
+def download_file(url, video_id, cache_dir: str):
+    clean_dir(cache_dir)
+    local_file = f"{cache_dir}/{video_id}.mp4"
+    response = requests.get(url, headers=HEADER, stream=True)
+    if response.status_code == 200:
+        total_size = int(response.headers.get("Content-Length", 0)) + 1
+        time_stamp = timestamp()
+        progress_bar = tqdm(
+            total=total_size,
+            unit="B",
+            unit_scale=True,
+            desc=f"[{time_stamp}] {local_file}",
+        )
+        with open(local_file, "wb") as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                if chunk:  # 确保 chunk 不为空
+                    f.write(chunk)  # 更新进度条
+                    progress_bar.update(len(chunk))
+
+    else:
+        raise ConnectionError(f"HTTP: {response.status_code}")
+
+    return local_file