From 1eab6c85902a8c287150f8f8194b14021467bd78 Mon Sep 17 00:00:00 2001
From: sabaimran <narmiabas@gmail.com>
Date: Mon, 5 Aug 2024 17:23:29 +0530
Subject: [PATCH] Add additional icons for agents, pencil line and chalkboard

---
 src/interface/web/app/common/iconUtils.tsx    |  8 ++++
 .../migrations/0055_alter_agent_style_icon.py | 37 +++++++++++++++++++
 src/khoj/database/models/__init__.py          |  2 +
 3 files changed, 47 insertions(+)
 create mode 100644 src/khoj/database/migrations/0055_alter_agent_style_icon.py

diff --git a/src/interface/web/app/common/iconUtils.tsx b/src/interface/web/app/common/iconUtils.tsx
index 484e4b4f..d2c0ba63 100644
--- a/src/interface/web/app/common/iconUtils.tsx
+++ b/src/interface/web/app/common/iconUtils.tsx
@@ -24,6 +24,8 @@ import {
     BowlFood,
     Lectern,
     Wallet,
+    PencilLine,
+    Chalkboard,
 } from "@phosphor-icons/react";
 import { Markdown, OrgMode, Pdf, Word } from "@/app/components/logo/fileLogo";
 
@@ -95,6 +97,12 @@ const iconMap: IconMap = {
     Wallet: (color: string, width: string, height: string) => (
         <Wallet className={`${width} ${height} ${color} mr-2`} />
     ),
+    PencilLine: (color: string, width: string, height: string) => (
+        <PencilLine className={`${width} ${height} ${color} mr-2`} />
+    ),
+    Chalkboard: (color: string, width: string, height: string) => (
+        <Chalkboard className={`${width} ${height} ${color} mr-2`} />
+    ),
 };
 
 function getIconFromIconName(
diff --git a/src/khoj/database/migrations/0055_alter_agent_style_icon.py b/src/khoj/database/migrations/0055_alter_agent_style_icon.py
new file mode 100644
index 00000000..4cc5c857
--- /dev/null
+++ b/src/khoj/database/migrations/0055_alter_agent_style_icon.py
@@ -0,0 +1,37 @@
+# Generated by Django 5.0.7 on 2024-08-05 11:51
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0054_alter_agent_style_color"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="agent",
+            name="style_icon",
+            field=models.CharField(
+                choices=[
+                    ("Lightbulb", "Lighbulb"),
+                    ("Health", "Health"),
+                    ("Robot", "Robot"),
+                    ("Aperture", "Aperture"),
+                    ("GraduationCap", "Graduation Cap"),
+                    ("Jeep", "Jeep"),
+                    ("Island", "Island"),
+                    ("MathOperations", "Math Operations"),
+                    ("Asclepius", "Asclepius"),
+                    ("Couch", "Couch"),
+                    ("Code", "Code"),
+                    ("Atom", "Atom"),
+                    ("ClockCounterClockwise", "Clock Counter Clockwise"),
+                    ("PencilLine", "Pencil Line"),
+                    ("Chalkboard", "Chalkboard"),
+                ],
+                default="Lightbulb",
+                max_length=200,
+            ),
+        ),
+    ]
diff --git a/src/khoj/database/models/__init__.py b/src/khoj/database/models/__init__.py
index 76ef5ab0..13340649 100644
--- a/src/khoj/database/models/__init__.py
+++ b/src/khoj/database/models/__init__.py
@@ -135,6 +135,8 @@ class Agent(BaseModel):
         CODE = "Code"
         ATOM = "Atom"
         CLOCK_COUNTER_CLOCKWISE = "ClockCounterClockwise"
+        PENCIL_LINE = "PencilLine"
+        CHALKBOARD = "Chalkboard"
 
     creator = models.ForeignKey(
         KhojUser, on_delete=models.CASCADE, default=None, null=True, blank=True