Update app.py

app.py

@@ -10,55 +10,6 @@ from safetensors.torch import save_file, load_file
 from huggingface_hub import model_info, create_repo, create_branch, upload_folder
 from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError
 
-def check_link(torrent_url):
-    if torrent_url.endswith('.torrent') or torrent_url.startswith('magnet:?xt=urn:btih:'):
-        return True
-    else:
-        return False
-
-def download_torrent(torrent_url):
-    # Create the file directory if it doesn't exist
-    os.makedirs("torrent", exist_ok=True)
-    if check_link(torrent_url):
-        # command to download torrent file
-        command = os.system(f"aria2c --enable-dht --dir=torrent --summary-interval=10 --seed-ratio=0.1 --allow-overwrite=true {torrent_url}")
-        return "download torrent done!"
-    else:
-        # command to download regular file
-        command = os.system(f"aria2c --dir=torrent --summary-interval=10 -c -x 10 -k 1M -s 10 {torrent_url}")
-        return "download torrent done!"
-    subprocess.run(command, shell=True)
-        
-def push_torrent(model_to, token, branch):
-    try:
-        repo_exists = True
-        r_info = model_info(model_to, token=token)
-    except RepositoryNotFoundError:
-        repo_exists = False
-    finally:
-        if repo_exists:
-            print(r_info)
-        else:
-            create_repo(model_to, private=True, token=token)
-    try:
-        branch_exists = True
-        b_info = model_info(model_to, revision=branch, token=token)
-    except RevisionNotFoundError:
-        branch_exists = False
-    finally:
-        if branch_exists:
-            print(b_info)
-        else:
-            create_branch(model_to, branch=branch, token=token)  
-            # Create the torrent directory if it doesn't exist
-            os.makedirs("torrent", exist_ok=True)
-    upload_folder(folder_path="torrent", path_in_repo="", revision=branch, repo_id=model_to, commit_message=f"torrent", token=token)
-    return "push torrent done!"
-    
-def delete_torrent():
-    os.system(f"rm -rf torrent")
-    return "delete torrent done!"
-
 def download_file(file_url_1, file_name_1):
     # Create the file directory if it doesn't exist
     os.makedirs("file", exist_ok=True)
@@ -370,6 +321,7 @@ with block:
     """
     
     ## 🚨 Please first click all delete buttons 🚨 🎉<br />
+    ##Using aria2c for better downloading
     ## Almost all Download site link works<br />
     ### Special thanks to [@camenduru](https://huggingface.co/camenduru) for creating initial script <br />
     ### Modified By  [@umair007](https://huggingface.co/umair007)<br />
@@ -427,32 +379,7 @@ with block:
                 btn_delete_file = gr.Button("Delete file")               
                 btn_download_file.click(download_file, inputs=[text_file_url_1, text_file_name_1], outputs=out_file)
                 btn_push_file_1.click(push_file_1, inputs=[text_file_model_to, text_file_token, text_file_branch], outputs=out_file)
-                btn_delete_file.click(delete_file, outputs=out_file) 
-
-    gr.Markdown(
-    """
-    ### download and push torrent file <br />
-    ckpt_url = example.com/file.torrent<br />
-    folder_name = openjourney <br />
-    ckpt_model_to = camenduru/openjourney <br />
-    branch = torrent <br />
-    token = get from [https://huggingface.co/settings/tokens](https://huggingface.co/settings/tokens) new token role=write <br />
-    """)
-    with gr.Group():
-        with gr.Box():
-            with gr.Row().style(equal_height=True):
-                text_torrent_url = gr.Textbox(show_label=False, max_lines=1, placeholder="torrent_url")
-                text_torrent_model_to = gr.Textbox(show_label=False, max_lines=1, placeholder="torrent_model_to")
-                text_torrent_branch = gr.Textbox(show_label=False, value="torrent", max_lines=1, placeholder="torrent_branch")
-                text_torrent_token = gr.Textbox(show_label=False, max_lines=1, placeholder="🤗 token")
-                out_torrent = gr.Textbox(show_label=False)
-            with gr.Row().style(equal_height=True):
-                btn_download_torrent = gr.Button("Download torrent")
-                btn_push_torrent = gr.Button("Push torrent to 🤗")
-                btn_delete_torrent = gr.Button("Delete torrent")
-                btn_download_torrent.click(download_torrent, inputs=[text_torrent_url], outputs=out_torrent)
-                btn_push_torrent.click(push_torrent, inputs=[text_torrent_model_to, text_torrent_token, text_torrent_branch], outputs=out_torrent)
-                btn_delete_torrent.click(delete_torrent, outputs=out_torrent)       
+                btn_delete_file.click(delete_file, outputs=out_file)     
                 
     gr.Markdown(
     """