Update app.py

app.py

@@ -1,21 +1,28 @@
 import os, csv, requests, gradio as gr
 from huggingface_hub import hf_hub_download, upload_file
+import logging
 
-# -- 設定 --
+# 設定日誌等級
+logging.basicConfig(level=logging.INFO)
+
+# -- 環境變數 & 路徑 --
 REPO_ID         = os.getenv("REPO_ID", "YuShen1124/youtube-tags-tool")
 HF_TOKEN        = os.getenv("HF_TOKEN")
 YOUTUBE_API_KEY = os.getenv("AIzaSyCvgpJMhcrDao6Li4qMUbFcifFSobbcIZw")
 DATA_DIR        = "./data"
-os.makedirs(DATA_DIR, exist_ok=True)
 CSV_FILE        = os.path.join(DATA_DIR, "tags_data.csv")
 COUNTER_FILE    = os.path.join(DATA_DIR, "counter.txt")
 
-# 初始化遠端檔案：若不存在則本地建立
+os.makedirs(DATA_DIR, exist_ok=True)
+
+# 初始化遠端檔案，若不存在則本地建立
 def init_file(remote, local):
     try:
         hf_hub_download(repo_id=REPO_ID, filename=remote, local_dir=DATA_DIR, token=HF_TOKEN)
+        logging.info(f"Downloaded {remote} from Hub")
     except Exception:
         open(local, "a", encoding="utf-8").close()
+        logging.info(f"Initialized empty {local}")
 
 init_file("tags_data.csv", CSV_FILE)
 init_file("counter.txt", COUNTER_FILE)
@@ -23,37 +30,55 @@ init_file("counter.txt", COUNTER_FILE)
 # 讀取 CSV
 def read_csv_data():
     data = {}
-    if os.path.exists(CSV_FILE):
+    try:
         with open(CSV_FILE, encoding='utf-8', newline='') as f:
             for tag, count in csv.reader(f):
                 data[tag] = int(count)
+    except Exception as e:
+        logging.error(f"Error reading CSV: {e}")
     return data
 
-# 寫入 CSV 並上傳
+# 寫入並上傳 CSV
 def write_csv_data(data):
-    with open(CSV_FILE, 'w', encoding='utf-8', newline='') as f:
-        csv.writer(f).writerows(data.items())
-    upload_file(
-        path_or_fileobj=CSV_FILE,
-        path_in_repo="tags_data.csv",
-        repo_id=REPO_ID,
-        token=HF_TOKEN
-    )
+    try:
+        with open(CSV_FILE, 'w', encoding='utf-8', newline='') as f:
+            csv.writer(f).writerows(data.items())
+        upload_file(path_or_fileobj=CSV_FILE, path_in_repo="tags_data.csv",
+                    repo_id=REPO_ID, token=HF_TOKEN)
+        logging.info("CSV uploaded to Hub")
+    except Exception as e:
+        logging.error(f"Error writing/uploading CSV: {e}")
 
-# 呼叫 YouTube API 取得 tags
+# 呼叫 YouTube API 取得 tags，附加 timeout 與錯誤處理
 def get_video_tags(keyword):
-    params = {'part':'snippet','q':keyword,'type':'video','maxResults':10,'key':YOUTUBE_API_KEY}
-    resp = requests.get("https://www.googleapis.com/youtube/v3/search", params=params).json()
     tags_count = {}
-    for item in resp.get('items', []):
-        vid = item['id']['videoId']
-        info = requests.get(
-            "https://www.googleapis.com/youtube/v3/videos",
-            params={'part':'snippet','id':vid,'key':YOUTUBE_API_KEY}
-        ).json()
-        for v in info.get('items', []):
-            for tag in v['snippet'].get('tags', []):
-                tags_count[tag] = tags_count.get(tag, 0) + 1
+    try:
+        resp = requests.get(
+            "https://www.googleapis.com/youtube/v3/search",
+            params={'part':'snippet','q':keyword,'type':'video','maxResults':10,'key':YOUTUBE_API_KEY},
+            timeout=10
+        )
+        resp.raise_for_status()
+        items = resp.json().get('items', [])
+    except Exception as e:
+        logging.error(f"Search API error: {e}")
+        return tags_count
+
+    for item in items:
+        vid = item['id'].get('videoId')
+        if not vid: continue
+        try:
+            info = requests.get(
+                "https://www.googleapis.com/youtube/v3/videos",
+                params={'part':'snippet','id':vid,'key':YOUTUBE_API_KEY},
+                timeout=10
+            )
+            info.raise_for_status()
+            for v in info.json().get('items', []):
+                for tag in v['snippet'].get('tags', []):
+                    tags_count[tag] = tags_count.get(tag, 0) + 1
+        except Exception as e:
+            logging.warning(f"Video API error for {vid}: {e}")
     return tags_count
 
 # 更新並分析標籤
@@ -61,20 +86,25 @@ def update_and_analyze_tags(keyword):
     if not keyword.strip():
         return "未輸入關鍵字", "", ""
     new_tags = get_video_tags(keyword)
+    if not new_tags:
+        logging.info("No tags found for keyword")
     agg = read_csv_data()
     for t, c in new_tags.items(): agg[t] = agg.get(t, 0) + c
     write_csv_data(agg)
 
-    # 更新並上傳執行次數
-    cnt = int(open(COUNTER_FILE, encoding='utf-8').read().strip() or 0) + 1
+    # 計數器讀取、更新與上傳
+    try:
+        cnt = int(open(COUNTER_FILE, encoding='utf-8').read().strip() or 0) + 1
+    except:
+        cnt = 1
     with open(COUNTER_FILE, 'w', encoding='utf-8') as f:
         f.write(str(cnt))
-    upload_file(
-        path_or_fileobj=COUNTER_FILE,
-        path_in_repo="counter.txt",
-        repo_id=REPO_ID,
-        token=HF_TOKEN
-    )
+    try:
+        upload_file(path_or_fileobj=COUNTER_FILE, path_in_repo="counter.txt",
+                    repo_id=REPO_ID, token=HF_TOKEN)
+        logging.info("Counter uploaded")
+    except Exception as e:
+        logging.error(f"Error uploading counter: {e}")
 
     sorted_tags = sorted(agg.items(), key=lambda x: x[1], reverse=True)
     full = "\n".join(f"{t}: {c}" for t, c in sorted_tags)
@@ -94,4 +124,4 @@ with gr.Blocks() as demo:
     inp.submit(update_and_analyze_tags, inputs=inp, outputs=[out1, out2, out3])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))