Update src/core/services/database_service.py

src/core/services/database_service.py

@@ -78,6 +78,39 @@ class DatabaseService:
                     FOREIGN KEY (parent_account_id) REFERENCES accounts (id),
                     FOREIGN KEY (account_owner_id) REFERENCES users (id)
                 );
+                
+                CREATE TABLE IF NOT EXISTS contacts (
+                    id TEXT PRIMARY KEY,
+                    account_id TEXT,
+                    first_name TEXT,
+                    last_name TEXT,
+                    email TEXT,
+                    phone TEXT,
+                    title TEXT,
+                    department TEXT,
+                    reports_to_id TEXT,
+                    influence_level TEXT,
+                    engagement_score REAL,
+                    preferences TEXT,
+                    created_at TEXT,
+                    last_contacted TEXT,
+                    FOREIGN KEY (account_id) REFERENCES accounts (id),
+                    FOREIGN KEY (reports_to_id) REFERENCES contacts (id)
+                );
+                
+                CREATE TABLE IF NOT EXISTS interactions (
+                    id TEXT PRIMARY KEY,
+                    type TEXT,
+                    account_id TEXT,
+                    owner_id TEXT,
+                    transcript TEXT,
+                    summary TEXT,
+                    sentiment_score REAL,
+                    metadata TEXT,
+                    created_at TEXT,
+                    FOREIGN KEY (account_id) REFERENCES accounts (id),
+                    FOREIGN KEY (owner_id) REFERENCES users (id)
+                );
             ''')
 
     def generate_synthetic_data(self):
@@ -185,6 +218,62 @@ class DatabaseService:
                 )
             ''', accounts)
             
+            # Generate contacts for each account
+            contacts = []
+            for account in accounts:
+                for _ in range(random.randint(3, 8)):  # 3-8 contacts per account
+                    contacts.append({
+                        'id': str(uuid.uuid4()),
+                        'account_id': account['id'],
+                        'first_name': self.fake.first_name(),
+                        'last_name': self.fake.last_name(),
+                        'email': self.fake.email(),
+                        'phone': self.fake.phone_number(),
+                        'title': random.choice(['CEO', 'CTO', 'CFO', 'VP Sales', 'Director']),
+                        'department': random.choice(['Executive', 'Sales', 'IT', 'Finance']),
+                        'reports_to_id': None,
+                        'influence_level': random.choice(['High', 'Medium', 'Low']),
+                        'engagement_score': random.uniform(0, 100),
+                        'preferences': json.dumps({}),
+                        'created_at': datetime.now().isoformat(),
+                        'last_contacted': datetime.now().isoformat()
+                    })
+            
+            c.executemany('''
+                INSERT OR REPLACE INTO contacts VALUES (
+                    :id, :account_id, :first_name, :last_name, :email, :phone,
+                    :title, :department, :reports_to_id, :influence_level,
+                    :engagement_score, :preferences, :created_at, :last_contacted
+                )
+            ''', contacts)
+            
+            # Generate interactions for each account
+            interactions = []
+            for account in accounts:
+                for _ in range(random.randint(5, 12)):  # 5-12 interactions per account
+                    interactions.append({
+                        'id': str(uuid.uuid4()),
+                        'type': random.choice(['call', 'meeting', 'email', 'presentation']),
+                        'account_id': account['id'],
+                        'owner_id': account['account_owner_id'],
+                        'transcript': self.fake.paragraph(),
+                        'summary': self.fake.sentence(),
+                        'sentiment_score': random.uniform(0, 1),
+                        'metadata': json.dumps({
+                            'duration': random.randint(15, 120),
+                            'location': random.choice(['virtual', 'in-person']),
+                            'attendees': random.randint(1, 5)
+                        }),
+                        'created_at': datetime.now().isoformat()
+                    })
+            
+            c.executemany('''
+                INSERT OR REPLACE INTO interactions VALUES (
+                    :id, :type, :account_id, :owner_id, :transcript, :summary,
+                    :sentiment_score, :metadata, :created_at
+                )
+            ''', interactions)
+            
             conn.commit()
 
     def add_account(self, account_data: Dict[str, Any]) -> str: