update

app.py

@@ -136,11 +136,11 @@ def handle_pdb_id_input(pdb_id):
     try: 
         pdb_id = pdb_id.strip().upper()
         pdbl = PDBList()
-        pdbl.retrieve_pdb_file(pdb_id, pdir=TEMP_DIR, file_format='pdb')
-        pdb_file_path = os.path.join(TEMP_DIR, f"pdb{pdb_id}.ent")
-        html_out, dd, structure_dict = load_structure(pdb_file_path)
-        # 删除文件
-        os.remove(pdb_file_path)
+        # 使用 TemporaryDirectory 创建临时文件夹
+        with TemporaryDirectory() as temp_dir:
+            pdbl.retrieve_pdb_file(pdb_id, pdir=temp_dir, file_format='pdb')
+            pdb_file_path = os.path.join(temp_dir, f"pdb{pdb_id}.ent")
+            html_out, dd, structure_dict = load_structure(pdb_file_path)
         return html_out, dd, structure_dict
     except Exception as e:
         error_message = f"<b style='color:red'>获取PDB ID {pdb_id} 失败 {e}</b>"