Merge pull request #48 from video-db/improve-profanity-agent

ProfanityRemoverAgent: Improve prompt and fix status bug
video-db · Oct 30, 2024 · 31ea3d6 · 31ea3d6
2 parents 8b9b446 + 6636990
commit 31ea3d6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/backend/director/agents/profanity_remover.py b/backend/director/agents/profanity_remover.py
@@ -68,9 +68,14 @@ def run(
         try:
             beep_audio_id = beep_audio_id or BEEP_AUDIO_ID
             if not beep_audio_id:
+                error_message = (
+                    "Please provide the beep_audio_id, or set up BEEP_AUDIO_ID in the .env file located in the backend directory."
+                    "Notify user this verbatum in summary as well since it isn't displayed to them: If you don’t have a beep sound ready, you can use the audio from https://www.youtube.com/watch?v=GvXbEO5Kbgc YouTube link. Upload it as audio, obtain the audio ID, and set it as BEEP_AUDIO_ID in your .env file.'"
+                    "Remember to add above line in summary."
+                )
                 return AgentResponse(
-                    status=AgentStatus.failed,
-                    message="Please provide the beep_audio_id or setup BEEP_AUDIO_ID in .env of backend directory.",
+                    status=AgentStatus.ERROR,
+                    message=error_message,
                 )
             self.output_message.actions.append("Started process to remove profanity..")
             video_content = VideoContent(