Chrunos commited on
Commit
6cfdeb9
·
verified ·
1 Parent(s): 7abe343

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -277,9 +277,10 @@ async def process_url(video_url: str) -> str:
277
 
278
 
279
 
 
280
 
281
  async def extract_video_info(video_url: str) -> str:
282
- api_url = f'https://yt-dl-web.vercel.app/api/info?url={video_url}'
283
  session = cloudscraper.create_scraper()
284
  try:
285
  response = session.get(api_url, timeout=20)
@@ -312,7 +313,7 @@ async def extract_video_info(video_url: str) -> str:
312
  async def test_download(request: Request):
313
  data = await request.json()
314
  video_url = data.get('url')
315
- response = await process_url(video_url)
316
  return response
317
 
318
 
@@ -410,6 +411,7 @@ api_key_header = APIKeyHeader(name="X-API-Key")
410
  # Store this securely in your environment variables
411
  API_KEY = os.getenv("API_KEY")
412
 
 
413
  async def verify_api_key(api_key: str = Security(api_key_header)):
414
  if api_key != API_KEY:
415
  raise HTTPException(
 
277
 
278
 
279
 
280
+ EXTRACT_API = os.getenv("EXTRACT_API")
281
 
282
  async def extract_video_info(video_url: str) -> str:
283
+ api_url = f'{EXTRACT_API}?url={video_url}'
284
  session = cloudscraper.create_scraper()
285
  try:
286
  response = session.get(api_url, timeout=20)
 
313
  async def test_download(request: Request):
314
  data = await request.json()
315
  video_url = data.get('url')
316
+ response = await extract_video_info(video_url)
317
  return response
318
 
319
 
 
411
  # Store this securely in your environment variables
412
  API_KEY = os.getenv("API_KEY")
413
 
414
+
415
  async def verify_api_key(api_key: str = Security(api_key_header)):
416
  if api_key != API_KEY:
417
  raise HTTPException(