replaced backed db with sqlite

app_utils/conf.py

@@ -1,10 +1,9 @@
-import shelve
+import sqlite3
 from pathlib import Path
-from typing import Any
+from typing import Any, Dict
 
 from pydantic import BaseModel
 
-
 class PersistentSettings(BaseModel):
     """
     This pydantic model will try to initialize itself from
@@ -15,16 +14,52 @@ class PersistentSettings(BaseModel):
     """
 
     def __init__(self, **data: Any):
-        with shelve.open("config.db", flag="n", protocol=2, writeback=False,) as db:
-            super().__init__(**db.get("settings", default={}), **data)
+        # Connect to the SQLite database
+        self.conn = sqlite3.connect("config.db")
+
+        # Create a table for settings if it doesn't exist
+        self.conn.execute("""
+            CREATE TABLE IF NOT EXISTS settings (
+                key TEXT PRIMARY KEY,
+                value TEXT
+            )
+        """)
+
+        # Fetch settings from the database and initialize
+        super().__init__(**self.fetch_settings(), **data)
+
+    def fetch_settings(self) -> Dict[str, Any]:
+        """
+        Retrieve settings from the database
+        """
+        cursor = self.conn.cursor()
+        cursor.execute("SELECT key, value FROM settings")
+        settings = dict(cursor.fetchall())
+        cursor.close()
+        return settings
 
     def update(self, **data: Any) -> None:
         """
         Persist the pydantic-dict that represents the model
         """
-        with shelve.open("config.db", flag="n", protocol=2, writeback=False,) as db:
-            db["settings"] = {**self.dict(), **data}
+        cursor = self.conn.cursor()
+
+        # Update or insert each key-value pair into the database
+        for key, value in {**self.dict(), **data}.items():
+            cursor.execute(
+                "INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)",
+                (key, value)
+            )
+
+        # Commit the changes to the database
+        self.conn.commit()
+        cursor.close()
 
+    def close(self) -> None:
+        """
+        Close the database connection
+        """
+        self.conn.close()
 
 class TortoiseConfig(PersistentSettings):
     EXTRA_VOICES_DIR: str = ""

app_utils/funcs.py

@@ -52,6 +52,5 @@ def list_voices(extra_voices_dir: Optional[str]):
 
 @st.cache_resource(max_entries=1)
 def load_voice_conditionings(voice, extra_voices_ls):
-    gc.collect()
     voice_samples, conditioning_latents = load_voices(voice, extra_voices_ls)
     return voice_samples, conditioning_latents