Update app.py

app.py

@@ -168,15 +168,52 @@ if model_loading_error:
 # Example queries in a dropdown
 example_categories = {
     "Benign SQL Queries": [
-        "SELECT * FROM users WHERE username='admin'",
-        "SELECT id, name, price FROM products WHERE category_id=5"
+    "SELECT * FROM users WHERE username='admin'",
+    "SELECT id, name, price FROM products WHERE category_id=5",
+    "SELECT COUNT(*) FROM orders WHERE date > '2023-01-01'",
+    "INSERT INTO logs (user_id, action) VALUES (42, 'login')",
+    "UPDATE customers SET last_login='2023-06-15' WHERE id=101",
+    "DELETE FROM sessions WHERE last_activity < '2023-01-01'",
+    "SELECT email FROM subscribers WHERE active=1",
+    "INSERT INTO feedback (user_id, message) VALUES (87, 'Great service!')",
+    "UPDATE inventory SET stock = stock - 1 WHERE product_id = 300",
+    "SELECT name FROM employees WHERE department = 'Sales'",
+    "SELECT AVG(rating) FROM reviews WHERE product_id = 55",
+    "INSERT INTO audit_log (timestamp, event) VALUES (CURRENT_TIMESTAMP, 'update')",
+    "SELECT * FROM appointments WHERE doctor_id = 10 AND status = 'confirmed'",
+    "UPDATE settings SET value='dark' WHERE key='theme'",
+    "SELECT DISTINCT city FROM customers WHERE country='USA'",
+    "DELETE FROM cart_items WHERE user_id=12 AND product_id=78",
+    "SELECT MAX(salary) FROM employees WHERE role='manager'",
+    "INSERT INTO payments (user_id, amount, method) VALUES (33, 99.99, 'credit')",
+    "UPDATE products SET price = price * 1.1 WHERE category_id = 7",
+    "SELECT * FROM messages WHERE sender_id = 5 AND is_read = 0"
     ],
     "Malicious SQL Queries": [
-        "' OR 1=1 --",
-        "admin'; DROP TABLE users; --"
+    "' OR 1=1 --",
+    "admin'; DROP TABLE users; --",
+    "SELECT * FROM users WHERE username='' UNION SELECT username,password FROM admin_users --",
+    "'; WAITFOR DELAY '0:0:10' --",
+    "admin' OR '1'='1",
+    "' OR 'a'='a",
+    "' OR 1=1#",
+    "' OR 1=1/*",
+    "admin'--",
+    "'; EXEC xp_cmdshell('dir'); --",
+    "' OR EXISTS(SELECT * FROM users WHERE username = 'admin') --",
+    "1; DROP TABLE sessions --",
+    "'; SHUTDOWN --",
+    "' OR SLEEP(5) --",
+    "' AND 1=(SELECT COUNT(*) FROM users) --",
+    "admin' AND SUBSTRING(password, 1, 1) = 'a' --",
+    "' UNION ALL SELECT NULL,NULL,NULL --",
+    "0' OR 1=1 ORDER BY 1 --",
+    "1' AND (SELECT COUNT(*) FROM users) > 0 --",
+    "' OR (SELECT ASCII(SUBSTRING(password,1,1)) FROM users WHERE username='admin') > 64 --"
     ]
 }
 
+
 category = st.selectbox("Choose query category:", options=list(example_categories.keys()))
 example = st.selectbox("Select an example:", options=example_categories[category])
 query_source = st.radio("Query source:", ["Use selected example", "Enter my own query"])