Update app.py

app.py

@@ -7,27 +7,41 @@ import sqlite3
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 
 # Initialize SQLite database
-conn = sqlite3.connect('links.db', check_same_thread=False)
-c = conn.cursor()
+DB_FILE = 'links.db'
 
-# Create table if it doesn't exist
-c.execute('''CREATE TABLE IF NOT EXISTS links
-             (id TEXT PRIMARY KEY, name TEXT, url TEXT)''')
-conn.commit()
+def get_db_connection():
+    conn = sqlite3.connect(DB_FILE, check_same_thread=False)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+def init_db():
+    conn = get_db_connection()
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS links
+                 (id TEXT PRIMARY KEY, name TEXT, url TEXT)''')
+    conn.commit()
+    
+    # Initialize with some example URLs if the table is empty
+    c.execute("SELECT COUNT(*) FROM links")
+    if c.fetchone()[0] == 0:
+        initial_urls = [
+            {"id": str(uuid.uuid4()), "name": "Google", "url": "https://www.google.com"},
+            {"id": str(uuid.uuid4()), "name": "Bing", "url": "https://www.bing.com"},
+        ]
+        c.executemany("INSERT INTO links VALUES (:id, :name, :url)", initial_urls)
+        conn.commit()
+    conn.close()
+
+init_db()
 
 # Function to get all links from the database
 def get_links():
+    conn = get_db_connection()
+    c = conn.cursor()
     c.execute("SELECT * FROM links")
-    return [{"id": row[0], "name": row[1], "url": row[2]} for row in c.fetchall()]
-
-# Initialize with some example URLs if the table is empty
-if not get_links():
-    initial_urls = [
-        {"id": str(uuid.uuid4()), "name": "Google", "url": "https://www.google.com"},
-        {"id": str(uuid.uuid4()), "name": "Bing", "url": "https://www.bing.com"},
-    ]
-    c.executemany("INSERT INTO links VALUES (:id, :name, :url)", initial_urls)
-    conn.commit()
+    links = [dict(row) for row in c.fetchall()]
+    conn.close()
+    return links
 
 app.layout = dbc.Container([
     dbc.Row([
@@ -56,9 +70,12 @@ app.layout = dbc.Container([
 def update_url_list(n_clicks, new_name, new_link):
     ctx = dash.callback_context
     if ctx.triggered_id == 'add-url-button' and new_name and new_link:
+        conn = get_db_connection()
+        c = conn.cursor()
         new_id = str(uuid.uuid4())
         c.execute("INSERT INTO links VALUES (?, ?, ?)", (new_id, new_name, new_link))
         conn.commit()
+        conn.close()
     
     links = get_links()
     return [
@@ -88,10 +105,13 @@ def update_iframe(n_clicks):
         return dash.no_update
     button_id = ctx.triggered[0]['prop_id'].split('.')[0]
     clicked_id = eval(button_id)['index']
+    conn = get_db_connection()
+    c = conn.cursor()
     c.execute("SELECT url FROM links WHERE id = ?", (clicked_id,))
     result = c.fetchone()
+    conn.close()
     if result:
-        return result[0]
+        return result['url']
     return dash.no_update
 
 @callback(
@@ -105,8 +125,11 @@ def delete_url(n_clicks):
         return dash.no_update
     button_id = ctx.triggered[0]['prop_id'].split('.')[0]
     delete_id = eval(button_id)['index']
+    conn = get_db_connection()
+    c = conn.cursor()
     c.execute("DELETE FROM links WHERE id = ?", (delete_id,))
     conn.commit()
+    conn.close()
     return update_url_list(None, None, None)
 
 if __name__ == '__main__':