Refactor BrandStyleManager and main.py to integrate database functionality for campaign management. Replace JSON file handling with database calls for loading, saving, updating, and deleting campaigns. Update sample campaign loading to retrieve data from the database instead of JSON files.

backend/database.py

@@ -0,0 +1,80 @@
+import sqlite3
+import os
+from typing import List, Dict, Any
+import datetime
+
+class Database:
+    def __init__(self):
+        data_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "data"))
+        self.db_path = os.path.join(data_dir, "campaigns.db")
+        self._init_db()
+
+    def _init_db(self):
+        """Initialize the database with required tables."""
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            cursor.execute('''
+                CREATE TABLE IF NOT EXISTS campaigns (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    prompt TEXT NOT NULL,
+                    content TEXT NOT NULL,
+                    timestamp TEXT NOT NULL
+                )
+            ''')
+            conn.commit()
+
+    def get_all_campaigns(self) -> List[Dict[str, Any]]:
+        """Retrieve all campaigns from the database."""
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            cursor.execute('SELECT prompt, content, timestamp FROM campaigns ORDER BY timestamp DESC')
+            rows = cursor.fetchall()
+            return [
+                {
+                    "prompt": row[0],
+                    "content": row[1],
+                    "timestamp": row[2]
+                }
+                for row in rows
+            ]
+
+    def add_campaign(self, prompt: str, content: str) -> None:
+        """Add a new campaign to the database."""
+        timestamp = datetime.datetime.now().isoformat()
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            cursor.execute(
+                'INSERT INTO campaigns (prompt, content, timestamp) VALUES (?, ?, ?)',
+                (prompt, content, timestamp)
+            )
+            conn.commit()
+
+    def update_campaign(self, index: int, content: str) -> bool:
+        """Update an existing campaign's content."""
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            cursor.execute('SELECT id FROM campaigns ORDER BY timestamp DESC LIMIT 1 OFFSET ?', (index,))
+            result = cursor.fetchone()
+            if result:
+                campaign_id = result[0]
+                timestamp = datetime.datetime.now().isoformat()
+                cursor.execute(
+                    'UPDATE campaigns SET content = ?, timestamp = ? WHERE id = ?',
+                    (content, timestamp, campaign_id)
+                )
+                conn.commit()
+                return True
+            return False
+
+    def delete_campaign(self, index: int) -> bool:
+        """Delete a campaign by its index."""
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            cursor.execute('SELECT id FROM campaigns ORDER BY timestamp DESC LIMIT 1 OFFSET ?', (index,))
+            result = cursor.fetchone()
+            if result:
+                campaign_id = result[0]
+                cursor.execute('DELETE FROM campaigns WHERE id = ?', (campaign_id,))
+                conn.commit()
+                return True
+            return False