update id_to_seq process

app.py

@@ -1,5 +1,5 @@
 from transformers import pipeline
-from rcsbsearchapi import TextQuery, AttributeQuery, Query
+from rcsbsearchapi import TextQuery, AttributeQuery
 from rcsbsearchapi.search import Sort, SequenceQuery
 import os
 from dotenv import load_dotenv
@@ -375,10 +375,34 @@ class PDBSearchAssistant:
                 pdir=self.pdb_dir, 
                 file_format="pdb"
             )
-            
+            print(pdb_path)
             if not pdb_path or not os.path.exists(pdb_path):
                 print(f"Failed to download PDB file for {pdb_id}")
-                return []
+
+                structure_url = f"https://data.rcsb.org/rest/v1/core/entry/{pdb_id}"
+                response = requests.get(structure_url)
+                structure_data = response.json() if response.status_code == 200 else {}
+
+
+                sequence_url = f"https://data.rcsb.org/rest/v1/core/polymer_entity/{pdb_id}/1"
+                seq_response = requests.get(sequence_url)
+                seq_data = seq_response.json() if response.status_code == 200 else {}
+                sequence = seq_data.get('entity_poly', {}).get('pdbx_seq_one_letter_code', 'N/A')
+                
+                sequences = []
+
+                chain_info = {
+                            'chain_id': "A", # chain.id, 임의 설정 api 3개써서 가져오기는 가능
+                            'entity_id': '1',  # Default entity ID
+                            'description': structure_data.get('struct', {}).get('title', 'N/A'),
+                            'sequence': sequence,
+                            'length': len(sequence),
+                            'resolution': structure_data.get('rcsb_entry_info', {}).get('resolution_combined', [0.0])[0],
+                            'method': structure_data.get('exptl', [{}])[0].get('method', 'Unknown'),
+                            'release_date': structure_data.get('rcsb_accession_info', {}).get('initial_release_date', 'N/A')
+                        }
+                sequences.append(chain_info)
+                return sequences
             
             # Parse structure
             parser = PDB.PDBParser(QUIET=True)
@@ -455,6 +479,7 @@ class PDBSearchAssistant:
             
             if is_sequence_query and pdb_id:
                 # Get sequences for the PDB ID
+
                 sequences = self.get_sequences_by_pdb_id(pdb_id)
                 return {
                     "type": "sequence",
@@ -471,26 +496,6 @@ class PDBSearchAssistant:
             print(f"Error processing query: {str(e)}")
             return {"type": "structure", "results": []}
 
-def pdbsummary(name):
-
-    search_engine = ProteinSearchEngine()
-
-    query = ProteinQuery(
-        name,
-        max_resolution= 5.0
-    )
-
-    results = search_engine.search(query)
-
-    answer = ""
-    for i, structure in enumerate(results, 1):
-        answer += f"\n{i}. PDB ID : {structure.pdb_id}\n"
-        answer += f"\nResolution : {structure.resolution:.2f} A \n"
-        answer += f"Method : {structure.method}\n Title : {structure.title}\n"
-        answer += f"Release Date : {structure.release_date}\n Sequence length: {len(structure.sequence)} aa\n"
-        answer += f"    Sequence:\n {structure.sequence}\n"
-
-    return answer
 
 def render_html(pdb_id):
     if pdb_id is None: