Update app.py

app.py

@@ -1,37 +1,32 @@
 import gradio as gr
 import zipfile
 import sqlite3
-import tempfile
-import shutil
-import os
+import io
 
-# 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, '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
+    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
 
-    return sqlite3.connect(tmp_db_path)
-
-# Open connection at startup
 conn = get_connection_from_zip("proteins.db.zip", "proteins.db")
 cursor = conn.cursor()
 
-# 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 "No protein found for the given sequence."
-
+        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))
 
-# Gradio UI
 demo = gr.Interface(
     fn=search_by_sequence,
     inputs=gr.Textbox(label="Enter protein sequence"),
@@ -40,4 +35,4 @@ demo = gr.Interface(
     description="Search Swiss-Prot by full protein sequence (exact match)"
 )
 
-demo.launch()
+demo.launch(show_api=True, ssr=False)