Update app.py

app.py

@@ -11,6 +11,7 @@ import tempfile
 import os
 import uuid
 import gc  # Import the garbage collector module
+import json
 
 
 app = FastAPI()
@@ -109,6 +110,44 @@ async def fetch_data_for_url(request: Request):
 # Mount StaticFiles to serve files from the global download directory
 app.mount("/file", StaticFiles(directory=global_download_dir), name="downloads")
 
+
+@app.get("/yt-audio")
+async def yt_audio(url: str):
+    try:
+        # 设置请求头
+        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"
+        }
+
+        # 发送请求
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+
+        # 定义正则表达式模式
+        pattern = r'var ytInitialPlayerResponse = (\{.*?(?=.*?streamingData)(?=.*?videoDetails).*?\}\}\}\});'
+
+        # 查找匹配项
+        match = re.search(pattern, response.text)
+
+        if match:
+            # 提取匹配到的JSON数据
+            json_str = match.group(1)
+
+            # 去除末尾的分号
+            json_str = json_str.rstrip(';')
+
+            # 将JSON字符串转换为Python字典
+            data = json.loads(json_str)
+
+            return data
+        else:
+            return {"error": "No matching data found in the HTML response."}
+
+    except requests.RequestException as e:
+        return {"error": f"Request error: {str(e)}"}
+    except json.JSONDecodeError as e:
+        return {"error": f"JSON decoding error: {str(e)}"}
+
 # Middleware to set the correct MIME type
 @app.middleware("http")
 async def set_mime_type_middleware(request: Request, call_next):