Update app.py

app.py

@@ -1,6 +1,7 @@
 import libtorrent as lt
 import time
 import streamlit as st
+import os
 
 st.title('磁力链接 BT 下载器')
 
@@ -29,3 +30,17 @@ if magnet_link:
         time.sleep(1)
 
     st.write('下载完成')
+
+    # 获取下载的文件名
+    torrent_info = h.get_torrent_info()
+    files = torrent_info.files()
+    filepath = f"./{files.file_path(0)}"
+
+    # 提供一个下载按钮给用户下载文件
+    with open(filepath, "rb") as file:
+        btn = st.download_button(
+            label="下载文件",
+            data=file,
+            file_name=os.path.basename(filepath),
+            mime="application/octet-stream",
+        )