new db

app.py

@@ -8,58 +8,94 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 import sqlite3
-from datetime import datetime
-
+import random
+
+# Sample data for generating employees
+first_names = [
+    "John",
+    "Emma",
+    "Michael",
+    "Sarah",
+    "William",
+    "Olivia",
+    "James",
+    "Ava",
+    "Robert",
+    "Isabella",
+]
+last_names = [
+    "Smith",
+    "Johnson",
+    "Williams",
+    "Brown",
+    "Jones",
+    "Davis",
+    "Miller",
+    "Wilson",
+    "Moore",
+    "Taylor",
+]
+job_titles = [
+    "Software Engineer",
+    "Data Analyst",
+    "Marketing Manager",
+    "Sales Representative",
+    "Product Designer",
+]
+departments = ["Engineering", "Finance", "Marketing", "Sales", "Operations"]
 
-# Connect to SQLite database. It will be created if it doesn't exist
-conn = sqlite3.connect('employee_database.db')
-cursor = conn.cursor()
 
-# Create employees table with basic columns
-cursor.execute('''
-CREATE TABLE IF NOT EXISTS employees (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    first_name TEXT NOT NULL,
-    last_name TEXT NOT NULL,
-    job_title TEXT NOT NULL
-)
-''')
-
-# Commit the transaction
-conn.commit()
-
-# Sample data for employees
-employees_data = [
-    ('John', 'Smith', 'Software Engineer'),
-    ('Sarah', 'Johnson', 'Marketing Manager'),
-    ('Michael', 'Brown', 'Data Scientist'),
-    ('Emma', 'Wilson', 'Product Designer'),
-    ('William', 'Davis', 'DevOps Engineer'),
-    ('Olivia', 'Miller', 'UX Researcher'),
-    ('James', 'Wilson', 'Backend Developer'),
-    ('Jessica', 'Moore', 'Project Manager'),
-    ('Robert', 'Taylor', 'Frontend Developer'),
-    ('Lisa', 'Anderson', 'Quality Assurance'),
-    ('Richard', 'Thomas', 'Technical Writer'),
-    ('Amy', 'Jackson', 'Business Analyst'),
-    ('Charles', 'White', 'Database Administrator'),
-    ('Elizabeth', 'Harris', 'Customer Support'),
-    ('Thomas', 'Martin', 'Network Engineer'),
-    ('Jennifer', 'Thompson', 'HR Specialist'),
-    ('Donald', 'Walker', 'Security Specialist'),
-    ('Margaret', 'Young', 'Content Manager'),
-    ('Ronald', 'Allen', 'Sales Representative'),
-    ('Susan', 'Scott', 'Financial Analyst')
-]
+def create_employee_table():
+    conn = sqlite3.connect("employee_database.db")  # Using in-memory database for demonstration
+    cursor = conn.cursor()
 
-# Insert multiple records efficiently using executemany
-cursor.executemany('''
-INSERT INTO employees (first_name, last_name, job_title)
-VALUES (?, ?, ?)
-''', employees_data)
+    # Create employees table
+    cursor.execute(
+        """
+        CREATE TABLE IF NOT EXISTS employees (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            first_name TEXT NOT NULL,
+            last_name TEXT NOT NULL,
+            job_title TEXT NOT NULL,
+            department_name TEXT NOT NULL,
+            salary REAL NOT NULL
+        )
+    """
+    )
+
+    # Generate and insert sample data
+    for _ in range(30):
+        employee = {
+            "first_name": random.choice(first_names),
+            "last_name": random.choice(last_names),
+            "job_title": random.choice(job_titles),
+            "department_name": random.choice(departments),
+            "salary": round(random.uniform(40000, 120000), 2),
+        }
+
+        cursor.execute(
+            """
+            INSERT INTO employees 
+            (first_name, last_name, job_title, department_name, salary)
+            VALUES (?, ?, ?, ?, ?)
+        """,
+            (
+                employee["first_name"],
+                employee["last_name"],
+                employee["job_title"],
+                employee["department_name"],
+                employee["salary"],
+            ),
+        )
+
+    conn.commit()
+
+    conn.close()
+
+
+# Create and populate the database
+create_employee_table()
 
-# Commit the transaction
-conn.commit()
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool