Update app.py

app.py

@@ -2,6 +2,7 @@ import libtorrent as lt
 import time
 import streamlit as st
 import os
+import shutil
 
 st.title('磁力链接 BT 下载器')
 
@@ -9,7 +10,7 @@ magnet_link = st.text_input('请输入磁力链接：')
 
 if magnet_link:
     ses = lt.session()
-    h = lt.add_magnet_uri(ses, magnet_link, {'save_path': './'})
+    h = lt.add_magnet_uri(ses, magnet_link, {'save_path': './downloads'})
     st.write('开始下载...')
 
     while not h.is_seed():
@@ -31,16 +32,17 @@ if magnet_link:
 
     st.write('下载完成')
 
-    # 获取下载的文件名
-    torrent_info = h.get_torrent_info()
-    files = torrent_info.files()
-    filepath = f"./{files.file_path(0)}"
+    # 获取下载的文件路径
+    save_path = './downloads'
 
-    # 提供一个下载按钮给用户下载文件
-    with open(filepath, "rb") as file:
-        btn = st.download_button(
-            label="下载文件",
+    # 压缩文件夹为 zip 文件
+    shutil.make_archive('downloads', 'zip', save_path)
+
+    # 提供一个下载按钮给用户下载 zip 文件
+    with open('downloads.zip', 'rb') as file:
+        st.download_button(
+            label='下载文件',
             data=file,
-            file_name=os.path.basename(filepath),
-            mime="application/octet-stream",
+            file_name='downloads.zip',
+            mime='application/zip'
         )