Update app.py

app.py

@@ -1,7 +1,38 @@
 import gradio as gr
+import zipfile
+import sqlite3
+import io
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load DB from ZIP in memory
+def get_connection_from_zip(zip_path: str, db_filename: str) -> sqlite3.Connection:
+    with zipfile.ZipFile(zip_path) as z:
+        with z.open(db_filename) as f:
+            db_bytes = io.BytesIO(f.read())
+            return sqlite3.connect(f"file:{db_filename}?mode=ro", uri=True, factory=sqlite3.Connection, database=db_bytes)
+
+# Open connection once at startup
+conn = get_connection_from_zip("proteins.db.zip", "proteins.db")
+cursor = conn.cursor()
+
+# 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
+
+    col_names = [desc[0] for desc in cursor.description]
+    return "\n".join(f"{k}: {v}" for k, v in zip(col_names, row))
+
+# Gradio UI
+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)"
+)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()