clean up on failure, too

app.py

@@ -12,6 +12,7 @@ api=HfApi()
 FILE_TYPES = ["Embedding", "Stable Diffusion Model", "Lora", "Lycoris"]
 
 def duplicate(source_url, preview_url, model_id, dst_repo, token, repo_type, nsfw, make_pr):
+    dir="/home/user/apps/downloads/"+str(int(time.time()))+str(random.getrandbits(8))
     try:
         _ = whoami(token)
         # ^ this will throw if token is invalid
@@ -24,7 +25,6 @@ def duplicate(source_url, preview_url, model_id, dst_repo, token, repo_type, nsf
         login(token=token)
 
         # keep things separate, partly in case people download different files with same name (`download.zip`). Especially, it also allows saving filename to work
-        dir="/home/user/apps/downloads/"+str(int(time.time()))+str(random.getrandbits(8))+"/"
         subprocess.check_call([r"mkdir","-p",dir])
         subprocess.check_call([r"aria2c","-x16","--split=16",source_url,"--dir="+dir])
 
@@ -54,11 +54,7 @@ def duplicate(source_url, preview_url, model_id, dst_repo, token, repo_type, nsf
         )
 
         # now clean up
-        files=os.listdir(dir)
-        os.remove(dir+files[0])
-        os.remove(dir+files[1])
-        os.remove(dir+files[2])
-        os.rmdir(dir)
+        subprocess.check_call([r"rm","-rf",dir])
 
         return (
             f'Find your commit at the top of <a href=\'https://hf.co/spaces/{dst_repo}/commits/main{repopath}\' target="_blank" style="text-decoration:underline">your commits</a>',
@@ -66,6 +62,7 @@ def duplicate(source_url, preview_url, model_id, dst_repo, token, repo_type, nsf
         )
 
     except Exception as e:
+        subprocess.check_call([r"rm","-rf",dir])
         blames=["grandma","my boss","your boss","God","you","you. It's *all* your fault.","the pope"]
         blameweights=(1,1,1,1,4,2,1)
         excuses=["I blame it all on "+random.choices(blames,weights=blameweights)[0],"It's my fault, sorry.","I did it on purpose.","That file doesn't want to be downloaded.","You nincompoop!"]