Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import sqlite3
 import os
 import requests
+import json
 
 HF_DB_URL = "https://huggingface.co/datasets/babilonczyk/swiss_prot/resolve/main/proteins.db"
 LOCAL_DB_PATH = "proteins.db"
@@ -44,12 +45,15 @@ print_schema()
 # Query function
 def search_by_sequence(seq: str) -> str:
     seq = seq.strip().upper()
-    cursor.execute("SELECT * FROM proteins WHERE sequence = ?", (seq,))
-    row = cursor.fetchone()
-    if not row:
-        return "No match found."
-    col_names = [desc[0] for desc in cursor.description]
-    return "\n".join(f"{k}: {v}" for k, v in zip(col_names, row))
+    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."
 
 # Gradio app
 demo = gr.Interface(