Update app.py

app.py

@@ -1,23 +1,18 @@
-import gradio as gr
-import zipfile
-import sqlite3
-import io
-
 def get_connection_from_zip(zip_path: str, db_filename: str) -> sqlite3.Connection:
+    # Extract DB file from ZIP to disk as "temp.db"
     with zipfile.ZipFile(zip_path) as z:
         with z.open(db_filename) as f:
-            db_bytes = io.BytesIO(f.read())
-            conn = sqlite3.connect("file:memdb1?mode=memory&cache=shared", uri=True, check_same_thread=False)
-            disk_conn = sqlite3.connect(f"file:temp.db?mode=ro", uri=True)
-            with open("temp.db", "wb") as temp_file:
-                temp_file.write(db_bytes.getvalue())
-            disk_conn = sqlite3.connect("temp.db", check_same_thread=False)
-            disk_conn.backup(conn)
-            return conn
+            with open("temp.db", "wb") as out_file:
+                out_file.write(f.read())
+
+    # Open SQLite connection (read-only)
+    return sqlite3.connect("temp.db", check_same_thread=False)
 
+# Set up DB
 conn = get_connection_from_zip("proteins.db.zip", "proteins.db")
 cursor = conn.cursor()
 
+# Query logic
 def search_by_sequence(seq: str) -> str:
     seq = seq.strip().upper()
     cursor.execute("SELECT * FROM proteins WHERE sequence = ?", (seq,))
@@ -27,6 +22,7 @@ def search_by_sequence(seq: str) -> str:
     col_names = [desc[0] for desc in cursor.description]
     return "\n".join(f"{k}: {v}" for k, v in zip(col_names, row))
 
+# Gradio app
 demo = gr.Interface(
     fn=search_by_sequence,
     inputs=gr.Textbox(label="Enter protein sequence"),