Merge pull request #699 from GooeyAI/deafult_widget

use bot branding from web integration when available

Author: anish-work

bots/models/bot_integration.py

@@ -518,17 +518,7 @@ def get_web_widget_config(
         config = self.web_config_extras | dict(
             target=target,
             integration_id=self.api_integration_id(),
-            branding=(
-                self.web_config_extras.get("branding", {})
-                | dict(
-                    name=self.name,
-                    byLine=self.by_line,
-                    description=self.descripton,
-                    conversationStarters=self.conversation_starters,
-                    photoUrl=self.photo_url,
-                    websiteUrl=self.website_url,
-                )
-            ),
+            branding=self.get_web_widget_branding(),
         )
 
         google_maps_api_key = None
@@ -551,6 +541,16 @@ def get_web_widget_config(
             config["apiUrl"] = get_api_route_url(stream_create)
         return config
 
+    def get_web_widget_branding(self) -> dict:
+        return self.web_config_extras.get("branding", {}) | dict(
+            name=self.name,
+            byLine=self.by_line,
+            description=self.descripton,
+            conversationStarters=self.conversation_starters,
+            photoUrl=self.photo_url,
+            websiteUrl=self.website_url,
+        )
+
     def translate(self, text: str) -> str:
         from daras_ai_v2.asr import run_google_translate, should_translate_lang
 

recipes/VideoBots.py

@@ -797,6 +797,24 @@ def render_output(self):
                 )
             )
 
+        # fill branding with bot integration data if available
+        bot_integration = (
+            BotIntegration.objects.filter(
+                published_run=self.current_pr,
+                platform=Platform.WEB,
+            )
+            .order_by("-updated_at")
+            .first()
+        )
+        if bot_integration:
+            bot_branding = bot_integration.get_web_widget_branding()
+        else:
+            bot_branding = dict(
+                name=self.current_pr.title,
+                photoUrl=self.current_pr.photo_url,
+                title=self.current_pr.title,
+            )
+        bot_branding["showPoweredByGooey"] = False
         gui.html(
             # language=html
             f"""
@@ -846,12 +864,7 @@ def render_output(self):
                 enableAudioMessage=True,
                 enablePhotoUpload=True,
                 enableConversations=False,
-                branding=dict(
-                    name=self.current_pr.title,
-                    photoUrl=self.current_pr.photo_url,
-                    title="Preview",
-                    showPoweredByGooey=False,
-                ),
+                branding=bot_branding,
                 fillParent=True,
                 secrets=dict(GOOGLE_MAPS_API_KEY=settings.GOOGLE_MAPS_API_KEY),
             ),