make copilot workflow image to rounded-circle border

Author: anish-work

daras_ai_v2/base.py

@@ -374,6 +374,8 @@ def render(self):
             self._render_header()
 
     def _render_header(self):
+        from widgets.workflow_image import CIRCLE_IMAGE_WORKFLOWS
+
         sr, pr = self.current_sr_pr
         is_example = pr.saved_run == sr
         is_root_example = is_example and pr.is_root()
@@ -416,16 +418,20 @@ def _render_header(self):
                     gui.div(
                         className="d-flex mt-4 mt-md-2 flex-column flex-md-row align-items-center gap-4 container-margin-reset"
                     ),
-                ):
+                ):  
+                    imageStyles = dict(
+                        maxWidth="150px",
+                        height="150px",
+                        margin=0,
+                        minHeight="150px",
+                        objectFit="cover",
+                    )
+                    if self.workflow in CIRCLE_IMAGE_WORKFLOWS:
+                        imageStyles["borderRadius"] = "50%"
+
                     gui.image(
                         src=pr.photo_url,
-                        style=dict(
-                            width="150px",
-                            height="150px",
-                            margin=0,
-                            minHeight="150px",
-                            objectFit="cover",
-                        ),
+                        style=imageStyles,
                     )
                     with gui.div(className="d-flex gap-2 w-100"):
                         with gui.div(className="flex-grow-1"):

widgets/workflow_image.py

@@ -9,7 +9,9 @@
     render_ai_generated_image_widget,
 )
 from workspaces.models import Workspace
+from bots.models import Workflow
 
+CIRCLE_IMAGE_WORKFLOWS = [Workflow.VIDEO_BOTS]
 
 def render_workflow_photo_uploader(
     workspace: Workspace,
@@ -30,6 +32,7 @@ def render_workflow_photo_uploader(
             is_generating=is_generating,
             error_msg=error_msg,
             icon=icons.photo,
+            is_circle_image=pr.workflow in CIRCLE_IMAGE_WORKFLOWS,
         )
     if pressed_generate:
         gui.session_state[key + ":bot-channel"] = run_icon_bot(
@@ -52,19 +55,6 @@ def render_change_notes_input(key: str):
         )
 
 
-def workflow_photo_div(url: str | None):
-    return gui.div(
-        style=dict(
-            aspectRatio="1",
-            width="80%",
-            backgroundImage=f"url({url})" if url else "none",
-            backgroundSize="cover",
-            backgroundPosition="center",
-        ),
-        className="d-flex align-items-center justify-content-center p-5 bg-light b-1 border rounded",
-    )
-
-
 PROMPT_FORMAT = '''\
 Title: %s 
 Description: """

widgets/workflow_metadata_gen.py

@@ -12,6 +12,7 @@ def render_ai_generated_image_widget(
     is_generating: bool,
     error_msg: Optional[str],
     icon: str,
+    is_circle_image: bool = False,
 ) -> bool:
     """
     Renders a reusable AI-generated image widget with preview, error, upload, and action buttons.
@@ -39,9 +40,15 @@ def render_ai_generated_image_widget(
     else:
         img_style = dict()
 
+    img_classes = "d-flex align-items-center justify-content-center bg-light b-1 border"
+    if is_circle_image:
+        img_classes += " rounded-circle"
+    else:
+        img_classes += " rounded"
+
     with gui.div(
         style=dict(height="200px", width="200px") | img_style,
-        className="d-flex align-items-center justify-content-center bg-light b-1 border rounded",
+        className=img_classes,
     ):
         if is_generating:
             with gui.styled(