Update app.py

app.py

@@ -4,18 +4,58 @@ import os
 import threading
 import time
 
-# Session状态初始化
-if 'download' not in st.session_state:
+# 线程安全的下载状态管理
+class DownloadState:
+    _instance = None
+    _lock = threading.Lock()
+    
+    def __new__(cls):
+        with cls._lock:
+            if cls._instance is None:
+                cls._instance = super().__new__(cls)
+                cls._instance.progress = 0
+                cls._instance.is_downloading = False
+                cls._instance.complete = False
+                cls._instance.file_path = None
+                cls._instance.status = ""
+            return cls._instance
+
+def init_session_state():
+    """初始化session_state"""
+    if 'download_initialized' not in st.session_state:
+        state = DownloadState()
+        st.session_state.download = {
+            'progress': state.progress,
+            'is_downloading': state.is_downloading,
+            'complete': state.complete,
+            'file_path': state.file_path,
+            'status': state.status
+        }
+        st.session_state.download_initialized = True
+
+def sync_session_state():
+    """同步状态到session_state"""
+    state = DownloadState()
     st.session_state.download = {
-        'progress': 0,
-        'is_downloading': False,
-        'complete': False,
-        'file_path': None,
-        'status': ""
+        'progress': state.progress,
+        'is_downloading': state.is_downloading,
+        'complete': state.complete,
+        'file_path': state.file_path,
+        'status': state.status
     }
 
 def download_torrent(magnet_link, save_path):
     try:
+        state = DownloadState()
+        
+        # 初始化下载状态
+        with threading.Lock():
+            state.progress = 0
+            state.is_downloading = True
+            state.complete = False
+            state.file_path = None
+            state.status = "正在初始化..."
+        
         ses = lt.session()
         params = {
             'save_path': save_path,
@@ -25,57 +65,64 @@ def download_torrent(magnet_link, save_path):
         handle = lt.add_magnet_uri(ses, magnet_link, params)
         
         # 获取元数据
-        st.session_state.download['status'] = "正在获取元数据..."
+        with threading.Lock():
+            state.status = "正在获取元数据..."
         while not handle.has_metadata():
             time.sleep(1)
         
         # 获取文件信息
         torrent_info = handle.get_torrent_info()
-        files = []
-        for i in range(torrent_info.num_files()):
-            files.append(torrent_info.file_at(i).path)
+        files = [torrent_info.file_at(i).path for i in range(torrent_info.num_files())]
         
         # 开始下载
-        st.session_state.download['status'] = "开始下载..."
-        st.session_state.download['is_downloading'] = True
+        with threading.Lock():
+            state.status = "开始下载..."
         
         while handle.status().state != lt.torrent_status.seeding:
             status = handle.status()
-            progress = status.progress * 100
-            st.session_state.download['progress'] = progress
+            with threading.Lock():
+                state.progress = status.progress * 100
             time.sleep(1)
         
         # 下载完成
-        st.session_state.download['is_downloading'] = False
-        st.session_state.download['complete'] = True
-        st.session_state.download['file_path'] = save_path
-        st.session_state.download['status'] = f"下载完成！保存路径：{save_path}"
+        with threading.Lock():
+            state.is_downloading = False
+            state.complete = True
+            state.file_path = save_path
+            state.status = f"下载完成！保存路径：{save_path}"
         
     except Exception as e:
-        st.session_state.download['status'] = f"错误：{str(e)}"
-        st.session_state.download['is_downloading'] = False
+        with threading.Lock():
+            state.status = f"错误：{str(e)}"
+            state.is_downloading = False
+    finally:
+        sync_session_state()
+
+# 初始化session状态
+init_session_state()
 
 # 界面布局
 st.title("🎯 磁力链接下载器")
 st.write("输入磁力链接开始下载（仅支持单个文件）")
 
 magnet_link = st.text_input("磁力链接：", placeholder="magnet:?xt=urn:btih:...")
-col1, col2 = st.columns([1, 3])
 
-with col1:
-    if st.button("🚀 开始下载", disabled=st.session_state.download['is_downloading']):
-        if not magnet_link.startswith("magnet:"):
-            st.error("请输入有效的磁力链接！")
-        else:
-            save_dir = "./downloads"
-            os.makedirs(save_dir, exist_ok=True)
-            threading.Thread(
-                target=download_torrent,
-                args=(magnet_link, save_dir),
-                daemon=True
-            ).start()
+if st.button("🚀 开始下载", disabled=st.session_state.download['is_downloading']):
+    if not magnet_link.startswith("magnet:"):
+        st.error("请输入有效的磁力链接！")
+    else:
+        save_dir = "./downloads"
+        os.makedirs(save_dir, exist_ok=True)
+        threading.Thread(
+            target=download_torrent,
+            args=(magnet_link, save_dir),
+            daemon=True
+        ).start()
+
+# 实时更新状态
+sync_session_state()
 
-# 下载状态显示
+# 显示下载状态
 if st.session_state.download['is_downloading']:
     st.progress(int(st.session_state.download['progress']))
     st.write(f"🔍 进度：{st.session_state.download['progress']:.1f}%")
@@ -84,7 +131,7 @@ if st.session_state.download['is_downloading']:
 if st.session_state.download['complete']:
     st.success("✅ 下载完成！")
     
-    # 显示下载文件
+    # 文件下载部分
     download_dir = st.session_state.download['file_path']
     files = [f for f in os.listdir(download_dir) if os.path.isfile(os.path.join(download_dir, f))]