diff --git a/app.py b/app.py index 80937fe..933f186 100644 --- a/app.py +++ b/app.py @@ -71,6 +71,7 @@ def init_db(): name TEXT UNIQUE, amount REAL DEFAULT 0, notes TEXT, + image_url TEXT, last_updated TEXT DEFAULT CURRENT_TIMESTAMP)''') # Default user logic remains same... diff --git a/migrate.py b/migrate.py index aae69a8..aedae21 100644 --- a/migrate.py +++ b/migrate.py @@ -12,7 +12,10 @@ def upgrade_db(): # App.py also expects unit_type, adding it to prevent future headaches conn.execute("ALTER TABLE products ADD COLUMN unit_type TEXT DEFAULT 'unit'") print("Successfully added 'unit_type' column.") - + + conn.execute("ALTER TABLE dicom ADD COLUMN image_url TEXT;") + print("Successfully added 'image_url' column.") + conn.commit() print("Migration complete. Your data is intact.")