Update app.py

app.py

@@ -2,46 +2,22 @@ import dash
 from dash import dcc, html, Input, Output, State, callback
 import dash_bootstrap_components as dbc
 import uuid
-import sqlite3
+import json
 import os
 
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 
-DB_FILE = 'links.db'
+DATA_FILE = 'links.json'
 
-def get_db_connection():
-    conn = sqlite3.connect(DB_FILE, check_same_thread=False)
-    conn.row_factory = sqlite3.Row
-    return conn
+def load_links():
+    if os.path.exists(DATA_FILE):
+        with open(DATA_FILE, 'r') as f:
+            return json.load(f)
+    return []
 
-def init_db():
-    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(f"Database initialized successfully at {os.path.abspath(DB_FILE)}")
-    except Exception as e:
-        print(f"Error initializing database: {e}")
-    finally:
-        conn.close()
-
-init_db()
-
-def get_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 from database: {links}")  # Debug print
-        return links
-    except Exception as e:
-        print(f"Error retrieving links: {e}")
-        return []
-    finally:
-        conn.close()
+def save_links(links):
+    with open(DATA_FILE, 'w') as f:
+        json.dump(links, f)
 
 def generate_url_list(links):
     url_list = [
@@ -64,7 +40,51 @@ def generate_url_list(links):
     print(f"Generated URL list: {url_list}")  # Debug print
     return url_list
 
-# ... (rest of the layout code remains the same)
+app.layout = dbc.Container([
+    dcc.Store(id='add-url-status', data=''),
+    dcc.Store(id='trigger-load', data='load'),
+    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")
+        ),
+    ], id="add-url-modal"),
+    dbc.Row([
+        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'),
+        ], width=3, className="bg-light p-3"),
+        dbc.Col([
+            html.Iframe(
+                id='content-iframe',
+                style={'width': '100%', 'height': '800px'},
+                sandbox="allow-scripts allow-same-origin allow-forms allow-popups allow-downloads",
+                allow="fullscreen; geolocation; microphone; camera; midi; encrypted-media; autoplay",
+                referrerPolicy="no-referrer"
+            )
+        ], width=9)
+    ])
+], fluid=True)
+
+@callback(
+    [Output("add-url-modal", "is_open"),
+     Output('new-url-name', 'value'),
+     Output('new-url-link', 'value')],
+    [Input("open-modal-button", "n_clicks"), Input("add-url-button", "n_clicks")],
+    [State("add-url-modal", "is_open")],
+)
+def toggle_modal(n1, n2, is_open):
+    if n1 or n2:
+        if is_open:
+            return False, '', ''
+        return True, dash.no_update, dash.no_update
+    return is_open, dash.no_update, dash.no_update
 
 @callback(
     [Output('url-list', 'children'),
@@ -87,45 +107,26 @@ def update_url_list(trigger_load, add_clicks, delete_clicks, new_name, new_link)
 
     feedback = None
     status = ''
+    links = load_links()
 
     if triggered_id == 'trigger-load':
         print("Initial load triggered")
     elif triggered_id == 'add-url-button' and new_name and new_link:
-        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()
-            print(f"Added new URL to database: {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 to database: {e}")
-            feedback = html.Div("Error adding URL. Please try again.", style={'color': 'red'})
-            status = 'error'
-        finally:
-            conn.close()
+        new_id = str(uuid.uuid4())
+        links.append({'id': new_id, 'name': new_name, 'url': new_link})
+        save_links(links)
+        print(f"Added new URL: {new_name}, {new_link}")  # Debug print
+        feedback = html.Div("URL added successfully!", style={'color': 'green'})
+        status = 'added'
     elif 'delete-button' in triggered_id:
         delete_id = eval(triggered_id)['index']
-        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 from database: {delete_id}")  # Debug print
-            feedback = html.Div("URL deleted successfully!", style={'color': 'green'})
-            status = 'deleted'
-        except Exception as e:
-            print(f"Error deleting URL from database: {e}")
-            feedback = html.Div("Error deleting URL. Please try again.", style={'color': 'red'})
-            status = 'error'
-        finally:
-            conn.close()
+        links = [link for link in links if link['id'] != delete_id]
+        save_links(links)
+        print(f"Deleted URL with id: {delete_id}")  # Debug print
+        feedback = html.Div("URL deleted successfully!", style={'color': 'green'})
+        status = 'deleted'
     
-    links = get_links()
     url_list = generate_url_list(links)
-    
     return url_list, feedback, status
 
 @callback(
@@ -138,21 +139,15 @@ def update_iframe(n_clicks):
         return dash.no_update
     button_id = ctx.triggered[0]['prop_id'].split('.')[0]
     clicked_id = eval(button_id)['index']
-    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()  # This line was likely causing the IndentationError
+    links = load_links()
+    for link in links:
+        if link['id'] == clicked_id:
+            print(f"Loading URL: {link['url']}")  # Debug print
+            return link['url']
     return dash.no_update
 
 if __name__ == '__main__':
     print("Starting the Dash application...")
+    print(f"Data file location: {os.path.abspath(DATA_FILE)}")
     app.run(debug=True, host='0.0.0.0', port=7860)
     print("Dash application has finished running.")