add emoji to WorkflowMetadata

GooeyAI · Mar 7, 2025 · e12ae88 · e12ae88
1 parent acb1d48
commit e12ae88
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 74 deletions.
diff --git a/bots/admin.py b/bots/admin.py
@@ -883,6 +883,7 @@ class WorkflowMetadataAdmin(admin.ModelAdmin):
         "created_at",
         "updated_at",
         "price_multiplier",
+        "emoji",
     ]
     search_fields = ["workflow", "meta_title", "meta_description"]
     list_filter = ["workflow"]

diff --git a/bots/migrations/0093_workflowmetadata_emoji.py b/bots/migrations/0093_workflowmetadata_emoji.py
@@ -0,0 +1,18 @@
+# Generated by Django 5.1.3 on 2025-03-07 06:51
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('bots', '0092_publishedrun_photo_url'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='workflowmetadata',
+            name='emoji',
+            field=models.TextField(blank=True, default=''),
+        ),
+    ]
diff --git a/bots/models.py b/bots/models.py
@@ -207,6 +207,7 @@ class WorkflowMetadata(models.Model):
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
     price_multiplier = models.FloatField(default=1)
+    emoji = models.TextField(blank=True, default="")
 
     def __str__(self):
         return self.meta_title

diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
@@ -70,7 +70,7 @@
 from routers.root import RecipeTabs
 from workspaces.models import Workspace, WorkspaceMembership
 from workspaces.widgets import get_current_workspace, set_current_workspace
-from daras_ai_v2.utils import get_relative_time, get_workflow_emoji
+from daras_ai_v2.utils import get_relative_time
 
 DEFAULT_META_IMG = (
     # Small
@@ -2527,6 +2527,7 @@ def render_example_preview_media(self, published_run: PublishedRun):
                         },
                     )
                 else:
+                    workflow = Workflow(self.current_pr.workflow)
                     preview_image = self.get_example_preview_image(
                         published_run.saved_run.to_dict()
                     )
@@ -2542,7 +2543,7 @@ def render_example_preview_media(self, published_run: PublishedRun):
                         )
                     else:
                         gui.write(
-                            f"# {get_workflow_emoji(Workflow(self.current_pr.workflow))}",
+                            f"# {workflow.get_or_create_metadata().emoji}",
                             className="m-0 container-margin-reset",
                         )
 

diff --git a/daras_ai_v2/utils.py b/daras_ai_v2/utils.py
@@ -27,75 +27,3 @@ def get_relative_time(timestamp: datetime) -> str:
             )
 
     return "Just now"
-
-
-def get_workflow_emoji(workflow: Workflow) -> str:
-    match workflow:
-        case 1:  # DOC_SEARCH
-            return "🔍"
-        case 2:  # DOC_SUMMARY
-            return "📚"
-        case 3:  # GOOGLE_GPT
-            return "🌐"
-        case 4:  # VIDEO_BOTS
-            return "💬"
-        case 5:  # LIPSYNC_TTS
-            return "👄"
-        case 6:  # TEXT_TO_SPEECH
-            return "🗣️"
-        case 7:  # ASR (Speech Recognition)
-            return "👂🏼"
-        case 8:  # LIPSYNC
-            return "👄"
-        case 9:  # DEFORUM_SD (Deforum Animation)
-            return "🎞️"
-        case 10:  # COMPARE_TEXT2IMG
-            return "🌄"
-        case 11:  # TEXT_2_AUDIO
-            return "🎵"
-        case 12:  # IMG_2_IMG
-            return "🖼️"
-        case 13:  # FACE_INPAINTING
-            return "🖌️"
-        case 14:  # GOOGLE_IMAGE_GEN
-            return "🖼️"
-        case 15:  # COMPARE_UPSCALER
-            return "🌄"
-        case 16:  # SEO_SUMMARY
-            return "🙋🏽‍♀️"
-        case 17:  # EMAIL_FACE_INPAINTING
-            return "📧"
-        case 18:  # SOCIAL_LOOKUP_EMAIL
-            return "✍️"
-        case 19:  # OBJECT_INPAINTING
-            return "🖌️"
-        case 20:  # IMAGE_SEGMENTATION
-            return "🖼️"
-        case 21:  # COMPARE_LLM
-            return "⚖️"
-        case 22:  # CHYRON_PLANT
-            return "🌱"
-        case 23:  # LETTER_WRITER
-            return "✉️"
-        case 24:  # SMART_GPT
-            return "💡"
-        case 25:  # QR_CODE
-            return "🏁"
-        case 26:  # DOC_EXTRACT
-            return "📄"
-        case 27:  # RELATED_QNA_MAKER
-            return "💬"
-        case 28:  # RELATED_QNA_MAKER_DOC
-            return "💬"
-        case 29:  # EMBEDDINGS
-            return "🧠"
-        case 30:  # BULK_RUNNER
-            return "🦾"
-        case 31:  # BULK_EVAL
-            return "⚖️"
-        case 32:  # FUNCTIONS
-            return "🛠️"
-        case 33:  # TRANSLATION
-            return "🌐"
-        case _:
-            return "💬"