Update app.py

app.py

@@ -28,40 +28,26 @@ download_db(HF_DB_URL, LOCAL_DB_PATH)
 conn = get_connection(LOCAL_DB_PATH)
 cursor = conn.cursor()
 
-
-def print_schema():
-    cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
-    tables = cursor.fetchall()
-    print("📋 Tables:", tables)
-    
-    for (table_name,) in tables:
-        print(f"\n🔎 Columns in table '{table_name}':")
-        cursor.execute(f"PRAGMA table_info({table_name});")
-        for col in cursor.fetchall():
-            print(col)
-
-print_schema()
-
-# Query function
-def search_by_sequence(seq: str) -> str:
-    seq = seq.strip().upper()
-    cursor.execute("SELECT json FROM proteins")
-    for (json_blob,) in cursor.fetchall():
-        try:
-            data = json.loads(json_blob)
-            if data.get("sequence", "").upper() == seq:
-                return json.dumps(data, indent=2)
-        except json.JSONDecodeError:
-            continue
-    return "No match found."
+# Search by protein ID (exact match)
+def search_by_id(protein_id: str) -> str:
+    protein_id = protein_id.strip().upper()
+    cursor.execute("SELECT json FROM proteins WHERE id = ?", (protein_id,))
+    row = cursor.fetchone()
+    if not row:
+        return "❌ No match found."
+    try:
+        data = json.loads(row[0])
+        return json.dumps(data, indent=2)
+    except json.JSONDecodeError:
+        return "❌ Found row but failed to parse JSON."
 
 # Gradio app
 demo = gr.Interface(
-    fn=search_by_sequence,
-    inputs=gr.Textbox(label="Enter protein sequence"),
-    outputs=gr.Textbox(label="Protein metadata"),
-    title="SwissProt Search",
-    description="Search Swiss-Prot by full protein sequence (exact match)"
+    fn=search_by_id,
+    inputs=gr.Textbox(label="Enter UniProt ID (e.g. P0DTC2)"),
+    outputs=gr.Textbox(label="Protein metadata (JSON)"),
+    title="SwissProt Search by ID",
+    description="Search Swiss-Prot database by UniProt protein ID"
 )
 
 demo.launch(show_api=True)