Update app.py

app.py

@@ -169,20 +169,12 @@ class NameChronicles:
         )
 
     # Database Methods
-
-    def _connect_database(self):
-        """
-        Connect to the database.
-        """
-        return duckdb.connect(database=str(self.db_path))
-
+    
     def _initialize_database(self, refresh):
         """
         Initialize database with data from the Social Security Administration.
         """
-        if not refresh and self.db_path.exists():
-            return
-
+        self.conn = duckdb.connect(":memory:")
         df = pd.concat(
             [
                 pd.read_csv(
@@ -201,9 +193,8 @@ class NameChronicles:
             .rename(columns={"F": "female", "M": "male"})
             .fillna(0)
         )
-        with self._connect_database() as conn:
-            conn.execute("DROP TABLE IF EXISTS names")
-            conn.execute("CREATE TABLE names AS SELECT * FROM df_processed")
+        self.conn.execute("DROP TABLE IF EXISTS names")
+        self.conn.execute("CREATE TABLE names AS SELECT * FROM df_processed")
 
     def _query_names(self, names):
         """
@@ -216,19 +207,18 @@ class NameChronicles:
                 top_names_query = TOP_NAMES_WILDCARD_QUERY
             else:
                 top_names_query = TOP_NAMES_SELECT_QUERY
-            with self._connect_database() as conn:
-                top_names = (
-                    conn.execute(top_names_query, [name.lower()])
-                    .fetch_df()["name"]
-                    .tolist()
-                )
-                if len(top_names) == 0:
-                    pn.state.notifications.info(f"No names found matching {name!r}")
-                    continue
-                data_query = DATA_QUERY.format(
-                    placeholders=", ".join(["?"] * len(top_names))
-                )
-                df = conn.execute(data_query, top_names).fetch_df()
+            top_names = (
+                self.conn.execute(top_names_query, [name.lower()])
+                .fetch_df()["name"]
+                .tolist()
+            )
+            if len(top_names) == 0:
+                pn.state.notifications.info(f"No names found matching {name!r}")
+                continue
+            data_query = DATA_QUERY.format(
+                placeholders=", ".join(["?"] * len(top_names))
+            )
+            df = self.conn.execute(data_query, top_names).fetch_df()
             dfs.append(df)
 
         if len(dfs) > 0:
@@ -241,21 +231,20 @@ class NameChronicles:
     # Widget Methods
 
     def _randomize_name(self, event):
-        with self._connect_database() as conn:
-            name_pattern = self.name_pattern.value.lower()
-            if not name_pattern:
-                name_pattern = "%"
-            else:
-                name_pattern = name_pattern.replace("*", "%")
-            count_range = self.count_range.value
-            gender_select = self.gender_select.value.lower()
-            random_names = (
-                conn.execute(
-                    RANDOM_NAME_QUERY, [name_pattern, *count_range, gender_select]
-                )
-                .fetch_df()["name"]
-                .tolist()
+        name_pattern = self.name_pattern.value.lower()
+        if not name_pattern:
+            name_pattern = "%"
+        else:
+            name_pattern = name_pattern.replace("*", "%")
+        count_range = self.count_range.value
+        gender_select = self.gender_select.value.lower()
+        random_names = (
+            self.conn.execute(
+                RANDOM_NAME_QUERY, [name_pattern, *count_range, gender_select]
             )
+            .fetch_df()["name"]
+            .tolist()
+        )
         if random_names:
             for i in range(len(random_names)):
                 random_name = random_names[i]
@@ -466,4 +455,4 @@ class NameChronicles:
         return template
 
 
-NameChronicles().view().servable()
+NameChronicles().view().servable()