fix vip ret

README.md

@@ -9,6 +9,4 @@ app_file: app.py
 pinned: false
 license: apache-2.0
 short_description: QQ Music Parser
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---

app.py

@@ -31,13 +31,12 @@ def insert_metadata(file_path: str, title, artist, album):
     print(f"Metadata was successfully inserted into {file_path}")
 
 
-def download_file(id, url: str, title, artist, album, cache="./__pycache__"):
+def download_file(mid: str, url: str, title, artist, album, cache="./__pycache__"):
     if os.path.exists(cache):
         shutil.rmtree(cache)
 
     os.makedirs(cache)
-    format = "mp3"
-    local_filename = f"{cache}/{id}.{format}"
+    local_filename = f"{cache}/{mid}.mp3"
     response = requests.get(url, stream=True)
     if response.status_code == 200:
         with open(local_filename, "wb") as file:
@@ -106,10 +105,10 @@ def infer(url: str):
             params={"mid": song_mid},
         )
 
-        if response.status_code == 200:
+        if response.status_code == 200 and response.json()["code"] == 200:
             data = response.json()["data"]
             song = download_file(
-                data["id"],
+                data["mid"],
                 data["urls"][0]["url"],
                 data["title"],
                 data["author"],
@@ -122,6 +121,9 @@ def infer(url: str):
             quality = data["urls"][0]["type"]
             size = data["urls"][0]["size"]
 
+        else:
+            raise Exception(response.json()["msg"])
+
     except Exception as e:
         title = f"{e}"
 
@@ -144,7 +146,7 @@ if __name__ == "__main__":
             gr.Textbox(label="Artist", show_copy_button=True),
             gr.Textbox(label="Album", show_copy_button=True),
             gr.Textbox(label="Quality", show_copy_button=True),
-            gr.Textbox(label="Size", show_copy_button=True),
+            gr.Textbox(label="Size(B)", show_copy_button=True),
         ],
         title="QQ Music Parser",
         description="This site does not provide any audio storage services, only provide the most basic parsing services",