Update app.py

app.py

@@ -10,13 +10,14 @@ import json
 
 
 def dlvid(vid):
+ s = requests.Session()
  url = f"https://www.nicovideo.jp/watch/{vid}"
  headers = {
  'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36 Edg/124.0.0.0',
  }
  
  # 发送请求
- response = requests.get(url, headers=headers)
+ response = s.get(url, headers=headers)
  if response.status_code == 200:
  # 解析 HTML
  soup = BeautifulSoup(response.text, 'html.parser')
@@ -50,7 +51,7 @@ def dlvid(vid):
  'x-request-with': 'https://www.nicovideo.jp'
  }
  
- response = requests.request("POST", url, headers=headers, data=payload)
+ response = s.request("POST", url, headers=headers, data=payload)
  
  hlsjson=response.json()
  hlsuri=hlsjson['data']['contentUrl']
@@ -60,7 +61,7 @@ def dlvid(vid):
  
  # 使用 ffmpeg 将 m3u8 转换为 mp4
  mp3link = hlsuri # 假设你的 m3u8 地址存储在 mp3link 中
- ffmpeg_command = f'ffmpeg -protocol_whitelist "concat,file,http,https,tcp,tls,crypto" -i "{mp3link}" -c copy "test.mp4"'
+ ffmpeg_command = f'ffmpeg -protocol_whitelist "concat,file,http,https,tcp,tls,crypto" -i "{mp3link}" -c copy -cookies "{s.cookies.get_dict()}" "test.mp4"'
  os.system(ffmpeg_command)
  return 'test.mp4'
  else: