support p

app.py

@@ -24,7 +24,7 @@ def extract_fst_url(text):
     url_pattern = r'(https?://[^\s"]+)'
     match = re.search(url_pattern, text)
     if match:
-        out_url = match.group(1)
+        out_url = match.group(1).split("?")[0]
         if out_url[-1] == "/":
             out_url = out_url[:-1]
 
@@ -45,7 +45,7 @@ def get_real_url(short_url):
     ).url.split("/?")[0]
 
 
-def infer(video_url: str):
+def infer(video_url: str, p: int):
     title = cover = desc = dur = video = author = avatar = None
     if not video_url:
         title = "Empty video link!"
@@ -64,7 +64,7 @@ def infer(video_url: str):
             cover = response_json["imgurl"]
             desc = response_json["desc"]
 
-            response_data = response_json["data"][0]
+            response_data = response_json["data"][int(p) - 1]
             dur = response_data["duration"]
             video_id = video_url.split("/")[-1]
             video = download_file(response_data["video_url"], video_id)
@@ -91,6 +91,7 @@ if __name__ == "__main__":
                 placeholder="https://www.bilibili.com/video/*",
                 show_copy_button=True,
             ),
+            gr.Number(label="Part", minimum=1, maximum=1000, step=1, value=1),
         ],
         outputs=[
             gr.Textbox(label="Video title", show_copy_button=True),
@@ -105,9 +106,9 @@ if __name__ == "__main__":
         description="This site does not provide any video storage services, only to provide the most basic resolution services",
         flagging_mode="never",
         examples=[
-            "BV1Dt4y1o7bU",
-            "https://b23.tv/LuTAbzj",
-            "https://www.bilibili.com/video/BV1Dt4y1o7bU",
+            ["BV1Dt4y1o7bU", 1],
+            ["https://b23.tv/LuTAbzj", 1],
+            ["https://www.bilibili.com/video/BV1Dt4y1o7bU", 1],
         ],
         cache_examples=False,
     ).launch()