File size: 5,792 Bytes
5f685fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
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
from shortGPT.gpt import reddit_gpt, gpt_voice
import os
class RedditShortEngine(ContentShortEngine):
# Mapping of variable names to database paths
def __init__(self,voiceModule: VoiceModule, background_video_name: str, background_music_name: str,short_id="",
num_images=None, watermark=None, language:Language = Language.ENGLISH):
super().__init__(short_id=short_id, short_type="reddit_shorts", background_video_name=background_video_name, background_music_name=background_music_name,
num_images=num_images, watermark=watermark, language=language, voiceModule=voiceModule)
def __generateRandomStory(self):
question = reddit_gpt.getInterestingRedditQuestion()
script = reddit_gpt.createRedditScript(question)
return script
def __getRealisticStory(self, max_tries=3):
current_realistic_score = 0
current_try = 0
current_generated_script = ""
while (current_realistic_score < 6 and current_try < max_tries) or len(current_generated_script) > 1000:
new_script = self.__generateRandomStory()
new_realistic_score = reddit_gpt.getRealisticness(new_script)
if new_realistic_score >= current_realistic_score:
current_generated_script = new_script
current_realistic_score = new_realistic_score
current_try += 1
return current_generated_script, current_try
def _generateScript(self):
"""
Implements Abstract parent method to generate the script for the reddit short
"""
self.logger("Generating reddit question & entertaining story")
self._db_script, _ = self.__getRealisticStory(max_tries=1)
self._db_reddit_question = reddit_gpt.getQuestionFromThread(
self._db_script)
def _prepareCustomAssets(self):
"""
Override parent method to generate custom reddit image asset
"""
self.logger("Rendering short: (3/4) preparing custom reddit image...")
self.verifyParameters(question=self._db_reddit_question,)
title, header, n_comments, n_upvotes = reddit_gpt.generateRedditPostMetadata(
self._db_reddit_question)
imageEditingEngine = EditingEngine()
imageEditingEngine.ingestFlow(Flow.WHITE_REDDIT_IMAGE_FLOW, {
"username_text": header,
"ncomments_text": n_comments,
"nupvote_text": n_upvotes,
"question_text": title
})
imageEditingEngine.renderImage(
self.dynamicAssetDir+"redditThreadImage.png")
self._db_reddit_thread_image = self.dynamicAssetDir+"redditThreadImage.png"
def _editAndRenderShort(self):
"""
Override parent method to customize video rendering sequence by adding a Reddit image
"""
self.verifyParameters(
voiceover_audio_url=self._db_audio_path,
video_duration=self._db_background_video_duration,
music_url=self._db_background_music_url)
outputPath = self.dynamicAssetDir+"rendered_video.mp4"
if not (os.path.exists(outputPath)):
self.logger("Rendering short: Starting automated editing...")
videoEditor = EditingEngine()
videoEditor.addEditingStep(EditingStep.ADD_VOICEOVER_AUDIO, {
'url': self._db_audio_path})
videoEditor.addEditingStep(EditingStep.ADD_BACKGROUND_MUSIC, {'url': self._db_background_music_url,
'loop_background_music': self._db_voiceover_duration,
"volume_percentage": 0.11})
videoEditor.addEditingStep(EditingStep.CROP_1920x1080, {
'url': self._db_background_trimmed})
videoEditor.addEditingStep(EditingStep.ADD_SUBSCRIBE_ANIMATION, {'url': AssetDatabase.get_asset_link('subscribe animation')})
if self._db_watermark:
videoEditor.addEditingStep(EditingStep.ADD_WATERMARK, {
'text': self._db_watermark})
videoEditor.addEditingStep(EditingStep.ADD_REDDIT_IMAGE, {
'url': self._db_reddit_thread_image})
caption_type = EditingStep.ADD_CAPTION_SHORT_ARABIC if self._db_language == Language.ARABIC.value else EditingStep.ADD_CAPTION_SHORT
for timing, text in self._db_timed_captions:
videoEditor.addEditingStep(caption_type, {'text': text.upper(),
'set_time_start': timing[0],
'set_time_end': timing[1]})
if self._db_num_images:
for timing, image_url in self._db_timed_image_urls:
videoEditor.addEditingStep(EditingStep.SHOW_IMAGE, {'url': image_url,
'set_time_start': timing[0],
'set_time_end': timing[1]})
videoEditor.renderVideo(outputPath, logger= self.logger if self.logger is not self.default_logger else None)
self._db_video_path = outputPath
|