Update app.py

app.py

@@ -111,40 +111,44 @@ async def fetch_data_for_url(request: Request):
 app.mount("/file", StaticFiles(directory=global_download_dir), name="downloads")
 
 
+def extract_video_id(url):
+    # 定义正则表达式模式
+    pattern = r'(?:(?:youtube\.com\/(?:watch\?v=|embed\/|shorts\/)|youtu\.be\/)([a-zA-Z0-9_-]{11}))'
+    
+    # 进行匹配
+    match = re.search(pattern, url)
+    
+    if match:
+        return match.group(1)
+    else:
+        return None
+
+
 @app.get("/yt-audio")
 async def yt_audio(url: str):
     try:
         # 设置请求头
+        videoId = extract_video_id(url)
         headers = {
             "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/18.2 Safari/605.1.15",
             "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9",
             "Referer": "https://www.youtube.com/"
         }
 
+        body = f'{'context':{'client':{'clientName':'IOS','clientVersion':'19.14.3','deviceModel':'iPhone15,4'}},'videoId':{videoId}}'
+
         # 发送请求
-        response = requests.get(url, headers=headers)
+        response = requests.post(
+            url,
+            timeout=20,
+            json=body_json,
+            headers=headers
+        )
         response.raise_for_status()
 
-        # 定义正则表达式模式
-        pattern = r'var ytInitialPlayerResponse = (\{.*?(?=.*?streamingData)(?=.*?videoDetails).*?\}\}\}\});'
-
-        # 查找匹配项
-        match = re.search(pattern, response.text)
-        logging.info(response.text)
-
-        if match:
-            # 提取匹配到的JSON数据
-            json_str = match.group(1)
-
-            # 去除末尾的分号
-            json_str = json_str.rstrip(';')
-
-            # 将JSON字符串转换为Python字典
-            data = json.loads(json_str)
+        logging.info(response)
 
-            return data
-        else:
-            return {"error": "No matching data found in the HTML response."}
+        return response
 
     except requests.RequestException as e:
         return {"error": f"Request error: {str(e)}"}