From 560dd1b33e6e3c8fb4e136c9b39f3a7723c7d6ef Mon Sep 17 00:00:00 2001
From: John Howe <89397553+timerring@users.noreply.github.com>
Date: Mon, 9 Dec 2024 21:38:40 +0800
Subject: [PATCH] fix: fix the query issue

fix #111
---
 src/upload/extract_video_info.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/upload/extract_video_info.py b/src/upload/extract_video_info.py
index 9b75c89..9ba449c 100644
--- a/src/upload/extract_video_info.py
+++ b/src/upload/extract_video_info.py
@@ -5,7 +5,6 @@
 import json
 import os
 from datetime import datetime
-# from src.upload.query_search_suggestion import get_bilibili_suggestions
 from src.upload.query_search_suggestion import get_bilibili_suggestions
 
 def get_video_info(video_file_path):
@@ -48,7 +47,8 @@ def generate_desc(video_path):
 
 def generate_tag(video_path):
     title, artist, date = get_video_info(video_path)
-    tags = get_bilibili_suggestions(artist)
+    artist_text = re.sub(r'\W+', '', artist)
+    tags = get_bilibili_suggestions(artist_text)
     return tags
 
 def generate_source(video_path):
@@ -63,7 +63,7 @@ def generate_source(video_path):
 
 
 if __name__ == "__main__":
-    video_path = "/home/jh/Downloads/bilive/Videos/31612461/31612461_20241204-19-08-29.mp4"
+    video_path = ""
     video_title = generate_title(video_path)
     print(video_title)
     video_desc = generate_desc(video_path)