require subscribe animation from youtube

RayVentura · Aug 5, 2023 · d2e1406 · d2e1406
1 parent 79bbf7a
commit d2e1406
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/shortGPT/editing_framework/editing_steps/subscribe_animation.json b/shortGPT/editing_framework/editing_steps/subscribe_animation.json
@@ -2,9 +2,11 @@
 	"subscribe_animation": {
 		"type": "video",
 		"z": 6,
-		"inputs":{},
+		"inputs":{
+			"parameters": ["url"]
+		},
 		"parameters": {
-			"url": "public/subscribe-animation.mp4",
+			"url": null,
 			"audio": false
 		},
 		"actions": [

diff --git a/shortGPT/engine/content_short_engine.py b/shortGPT/engine/content_short_engine.py
@@ -130,7 +130,7 @@ def _editAndRenderShort(self):
                                                                           "volume_percentage": 0.11})
             videoEditor.addEditingStep(EditingStep.CROP_1920x1080, {
                                        'url': self._db_background_trimmed})
-            videoEditor.addEditingStep(EditingStep.ADD_SUBSCRIBE_ANIMATION)
+            videoEditor.addEditingStep(EditingStep.ADD_SUBSCRIBE_ANIMATION, {'url': AssetDatabase.get_asset_link('subscribe animation')})
 
             if self._db_watermark:
                 videoEditor.addEditingStep(EditingStep.ADD_WATERMARK, {

diff --git a/shortGPT/engine/reddit_short_engine.py b/shortGPT/engine/reddit_short_engine.py
@@ -1,4 +1,5 @@
 from shortGPT.audio.voice_module import VoiceModule
+from shortGPT.config.asset_db import AssetDatabase
 from shortGPT.config.languages import Language
 from shortGPT.engine.content_short_engine import ContentShortEngine
 from shortGPT.editing_framework.editing_engine import EditingEngine, EditingStep, Flow
@@ -79,7 +80,7 @@ def _editAndRenderShort(self):
                                                                           "volume_percentage": 0.11})
             videoEditor.addEditingStep(EditingStep.CROP_1920x1080, {
                                        'url': self._db_background_trimmed})
-            videoEditor.addEditingStep(EditingStep.ADD_SUBSCRIBE_ANIMATION)
+            videoEditor.addEditingStep(EditingStep.ADD_SUBSCRIBE_ANIMATION, {'url': AssetDatabase.get_asset_link('subscribe animation')})
 
             if self._db_watermark:
                 videoEditor.addEditingStep(EditingStep.ADD_WATERMARK, {