Update app.py

app.py

@@ -14,26 +14,36 @@ def get_db_connection():
     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()
-    conn.close()
+    try:
+        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()
+        print("Database initialized successfully")
+    except Exception as e:
+        print(f"Error initializing database: {e}")
+    finally:
+        conn.close()
 
 init_db()
 
 def get_links():
-    conn = get_db_connection()
-    c = conn.cursor()
-    c.execute("SELECT * FROM links")
-    links = [dict(row) for row in c.fetchall()]
-    conn.close()
-    print(f"Retrieved links: {links}")  # Debug print
-    return links
+    try:
+        conn = get_db_connection()
+        c = conn.cursor()
+        c.execute("SELECT * FROM links")
+        links = [dict(row) for row in c.fetchall()]
+        print(f"Retrieved links: {links}")  # Debug print
+        return links
+    except Exception as e:
+        print(f"Error retrieving links: {e}")
+        return []
+    finally:
+        conn.close()
 
 def generate_url_list(links):
-    return [
+    url_list = [
         dbc.ListGroupItem([
             dbc.Button(
                 link['name'], 
@@ -50,6 +60,8 @@ def generate_url_list(links):
         ], className="d-flex justify-content-between align-items-center")
         for link in links
     ]
+    print(f"Generated URL list: {url_list}")  # Debug print
+    return url_list
 
 app.layout = dbc.Container([
     dcc.Store(id='add-url-status', data=''),
@@ -68,7 +80,7 @@ app.layout = dbc.Container([
         dbc.Col([
             html.H2("My URL App", className="mt-3 mb-4"),
             dbc.Button("Add URL", id="open-modal-button", color="primary", className="mb-3"),
-            html.Div(id='url-list'),  # We'll populate this with a callback
+            html.Div(id='url-list'),
         ], width=3, className="bg-light p-3"),
         dbc.Col([
             html.Iframe(
@@ -97,6 +109,7 @@ def toggle_modal(n1, n2, is_open):
     Input('url-list', 'id')
 )
 def initialize_url_list(id):
+    print("Initializing URL list")  # Debug print
     links = get_links()
     return generate_url_list(links)
 
@@ -117,6 +130,7 @@ def update_url_list(add_clicks, status, delete_clicks, new_name, new_link):
         raise dash.exceptions.PreventUpdate
 
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
+    print(f"Triggered by: {triggered_id}")  # Debug print
 
     if triggered_id == 'add-url-button' and new_name and new_link:
         try:
@@ -125,22 +139,31 @@ def update_url_list(add_clicks, status, delete_clicks, new_name, new_link):
             new_id = str(uuid.uuid4())
             c.execute("INSERT INTO links VALUES (?, ?, ?)", (new_id, new_name, new_link))
             conn.commit()
-            conn.close()
+            print(f"Added new URL: {new_name}, {new_link}")  # Debug print
             feedback = html.Div("URL added successfully!", style={'color': 'green'})
             status = 'added'
         except Exception as e:
             print(f"Error adding URL: {e}")
             feedback = html.Div("Error adding URL. Please try again.", style={'color': 'red'})
             status = 'error'
+        finally:
+            conn.close()
     elif 'delete-button' in triggered_id:
         delete_id = eval(triggered_id)['index']
-        conn = get_db_connection()
-        c = conn.cursor()
-        c.execute("DELETE FROM links WHERE id = ?", (delete_id,))
-        conn.commit()
-        conn.close()
-        feedback = html.Div("URL deleted successfully!", style={'color': 'green'})
-        status = 'deleted'
+        try:
+            conn = get_db_connection()
+            c = conn.cursor()
+            c.execute("DELETE FROM links WHERE id = ?", (delete_id,))
+            conn.commit()
+            print(f"Deleted URL with id: {delete_id}")  # Debug print
+            feedback = html.Div("URL deleted successfully!", style={'color': 'green'})
+            status = 'deleted'
+        except Exception as e:
+            print(f"Error deleting URL: {e}")
+            feedback = html.Div("Error deleting URL. Please try again.", style={'color': 'red'})
+            status = 'error'
+        finally:
+            conn.close()
     else:
         feedback = None
         status = ''
@@ -170,13 +193,18 @@ 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['url']
+    try:
+        conn = get_db_connection()
+        c = conn.cursor()
+        c.execute("SELECT url FROM links WHERE id = ?", (clicked_id,))
+        result = c.fetchone()
+        if result:
+            print(f"Loading URL: {result['url']}")  # Debug print
+            return result['url']
+    except Exception as e:
+        print(f"Error loading URL: {e}")
+    finally:
+        conn.close()
     return dash.no_update
 
 if __name__ == '__main__':