Update app.py

app.py

@@ -29,14 +29,17 @@ def get_links():
     c.execute("SELECT * FROM links")
     links = [dict(row) for row in c.fetchall()]
     conn.close()
+    print(f"Retrieved links: {links}")  # Debug print
     return links
 
 app.layout = dbc.Container([
+    dcc.Store(id='add-url-status', data=''),
     dbc.Modal([
         dbc.ModalHeader("Add New URL"),
         dbc.ModalBody([
             dbc.Input(id='new-url-name', placeholder="URL Name", className="mb-2"),
             dbc.Input(id='new-url-link', placeholder="URL", className="mb-2"),
+            html.Div(id='add-url-feedback')
         ]),
         dbc.ModalFooter(
             dbc.Button("Add", id="add-url-button", className="ml-auto")
@@ -65,23 +68,36 @@ def toggle_modal(n1, n2, is_open):
     return is_open
 
 @callback(
-    Output('url-list', 'children'),
-    Input('add-url-button', 'n_clicks'),
-    State('new-url-name', 'value'),
-    State('new-url-link', 'value')
+    [Output('url-list', 'children'),
+     Output('add-url-feedback', 'children'),
+     Output('add-url-status', 'data')],
+    [Input('add-url-button', 'n_clicks'),
+     Input('add-url-status', 'data')],
+    [State('new-url-name', 'value'),
+     State('new-url-link', 'value')]
 )
-def update_url_list(n_clicks, new_name, new_link):
+def update_url_list(n_clicks, status, 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()
+        try:
+            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()
+            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'
+    else:
+        feedback = None
+        status = ''
     
     links = get_links()
-    return [
+    url_list = [
         dbc.ListGroupItem([
             dbc.Button(
                 link['name'], 
@@ -98,6 +114,18 @@ def update_url_list(n_clicks, new_name, new_link):
         ], className="d-flex justify-content-between align-items-center")
         for link in links
     ]
+    
+    return url_list, feedback, status
+
+@callback(
+    [Output('new-url-name', 'value'),
+     Output('new-url-link', 'value')],
+    Input('add-url-status', 'data')
+)
+def clear_inputs(status):
+    if status == 'added':
+        return '', ''
+    return dash.no_update, dash.no_update
 
 @callback(
     Output('content-iframe', 'src'),