update

.gitignore

@@ -1,7 +1,7 @@
 static/dna.html
 static/dna.pdb
 static/test.cif
-static/tmp
+static/tmp/*
 test.py
 *.ipynb
 __pycache__

app.py

@@ -133,18 +133,18 @@ def handle_file_upload(file):
         return gr.update(), gr.update(), gr.update()
 
 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)
-    return html_out, dd, structure_dict
-    # except Exception as e:
-    #     error_message = f"<b style='color:red'>获取PDB ID {pdb_id} 失败 {e}</b>"
-    #     return error_message, gr.update(), gr.update()
+    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)
+        return html_out, dd, structure_dict
+    except Exception as e:
+        error_message = f"<b style='color:red'>获取PDB ID {pdb_id} 失败 {e}</b>"
+        return error_message, gr.update(), gr.update()
 
 def render_cache(structure_cache):
     summary = structure_cache["summary"]