add timestamp

app.py

@@ -3,6 +3,21 @@ import re
 import shutil
 import requests
 import gradio as gr
+from datetime import datetime
+from zoneinfo import ZoneInfo
+from tzlocal import get_localzone
+
+TIMEOUT = 15
+API = os.getenv("api_tiktok")
+
+
+def timestamp(naive_time: datetime = None, target_tz=ZoneInfo("Asia/Shanghai")):
+    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")
 
 
 def download_file(url, video_id, cache_dir="./__pycache__"):
@@ -17,6 +32,7 @@ def download_file(url, video_id, cache_dir="./__pycache__"):
             for chunk in response.iter_content(chunk_size=8192):
                 file.write(chunk)
 
+    print(f"[{timestamp()}] File was downloaded to {local_file}")
     return local_file
 
 
@@ -41,7 +57,7 @@ def infer(video_url):
         return video, parse_time, desc, avatar, author, sign
 
     try:
-        response = requests.get(os.getenv("api_tiktok"), params={"url": video_url})
+        response = requests.get(API, params={"url": video_url}, timeout=TIMEOUT)
         response_json = response.json()
         retcode = response_json["code"]
         if retcode == 200:

requirements.txt

@@ -0,0 +1 @@
+tzlocal