Update app.py

app.py

@@ -1,27 +1,32 @@
 import gradio as gr
 import zipfile
 import sqlite3
-import io
+import tempfile
+import shutil
+import os
 
-# Load DB from ZIP in memory
+# Extract DB from ZIP to a temporary file
 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)
+    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+        with zip_ref.open(db_filename) as db_file:
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".db") as tmp_file:
+                shutil.copyfileobj(db_file, tmp_file)
+                tmp_db_path = tmp_file.name
 
-# Open connection once at startup
+    return sqlite3.connect(tmp_db_path)
+
+# Open connection at startup
 conn = get_connection_from_zip("proteins.db.zip", "proteins.db")
 cursor = conn.cursor()
 
-# Query function
+# Query by protein sequence
 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
+        return "No protein found for the given sequence."
 
     col_names = [desc[0] for desc in cursor.description]
     return "\n".join(f"{k}: {v}" for k, v in zip(col_names, row))