add ignore

.gitignore

@@ -0,0 +1,3 @@
+*__pycache__*
+test.*
+rename.sh

app.py

@@ -1,9 +1,29 @@
 import os
 import re
+import shutil
 import requests
 import gradio as gr
 
 
+def download_file(url, video_id, cache_dir="./__pycache__"):
+    if os.path.exists(cache_dir):
+        shutil.rmtree(cache_dir)
+
+    os.makedirs(cache_dir)
+    local_file = f"{cache_dir}/{video_id}.mp4"
+    try:
+        response = requests.get(url, stream=True)
+        if response.status_code == 200:
+            with open(local_file, "wb") as file:
+                for chunk in response.iter_content(chunk_size=8192):
+                    file.write(chunk)
+
+    except requests.exceptions.RequestException as e:
+        print(f"Failed to download: {e}")
+
+    return local_file
+
+
 def extract_fst_url(text):
     url_pattern = r'(https?://[^\s"]+)'
     match = re.search(url_pattern, text)
@@ -25,12 +45,13 @@ def infer(video_url):
         return video, parse_time, desc, avatar, author, sign
 
     try:
-        response = requests.get(os.getenv("api"), params={"url": video_url})
+        response = requests.get("https://api.xinyew.cn/api/douyinjx", params={"url": video_url})
         response_json = response.json()
         retcode = response_json["code"]
         if retcode == 200:
             response_data = response_json["data"]
-            video = response_data["play_url"]
+            video_id = response_data["play_url"].split("video_id=")[1].split("&")[0]
+            video = download_file(response_data["video_url"], video_id)
             parse_time = response_data["parse_time"]
             additional_data = response_data["additional_data"][0]
             desc = additional_data["desc"]
@@ -58,7 +79,7 @@ if __name__ == "__main__":
             ),
         ],
         outputs=[
-            gr.Video(label="Video download", format="mp4", show_download_button=True),
+            gr.Video(label="Video download", show_download_button=True),
             gr.Textbox(label="Parsing time", show_copy_button=True),
             gr.Textbox(label="Video description", show_copy_button=True),
             gr.Image(label="Author avatar", show_share_button=False),