Update app.py

app.py

@@ -6,19 +6,6 @@ import requests
 HF_DB_URL = "https://huggingface.co/datasets/babilonczyk/swiss_prot/resolve/main/proteins.db"
 LOCAL_DB_PATH = "proteins.db"
 
-def print_schema():
-    cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
-    tables = cursor.fetchall()
-    print("📋 Tables:", tables)
-    
-    for (table_name,) in tables:
-        print(f"\n🔎 Columns in table '{table_name}':")
-        cursor.execute(f"PRAGMA table_info({table_name});")
-        for col in cursor.fetchall():
-            print(col)
-
-print_schema()
-
 # Download the DB only if not already cached
 def download_db(url: str, local_path: str):
     if not os.path.exists(local_path):
@@ -40,6 +27,20 @@ download_db(HF_DB_URL, LOCAL_DB_PATH)
 conn = get_connection(LOCAL_DB_PATH)
 cursor = conn.cursor()
 
+
+def print_schema():
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
+    tables = cursor.fetchall()
+    print("📋 Tables:", tables)
+    
+    for (table_name,) in tables:
+        print(f"\n🔎 Columns in table '{table_name}':")
+        cursor.execute(f"PRAGMA table_info({table_name});")
+        for col in cursor.fetchall():
+            print(col)
+
+print_schema()
+
 # Query function
 def search_by_sequence(seq: str) -> str:
     seq = seq.strip().upper()