revert

app.py

@@ -8,94 +8,58 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 import sqlite3
-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"]
+from datetime import datetime
 
 
-def create_employee_table():
-    conn = sqlite3.connect("employee_database.db")  # Using in-memory database for demonstration
-    cursor = conn.cursor()
+# 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
-    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()
+# 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')
+]
+
+# Insert multiple records efficiently using executemany
+cursor.executemany('''
+INSERT INTO employees (first_name, last_name, job_title)
+VALUES (?, ?, ?)
+''', employees_data)
 
+# Commit the transaction
+conn.commit()
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool