Add progress tracking and background generation

src/chore/GenerationContext.py

@@ -17,7 +17,11 @@ class GenerationContext:
         captioningengine,
         assetsengine,
         settingsengine,
+        backgroundengine,
+        progress
     ) -> None:
+        self.progress = progress
+
         self.powerfulllmengine: engines.LLMEngine.BaseLLMEngine = powerfulllmengine[0]
         self.powerfulllmengine.ctx = self
 
@@ -44,6 +48,11 @@ class GenerationContext:
         self.settingsengine: engines.SettingsEngine.SettingsEngine = settingsengine[0]
         self.settingsengine.ctx = self
 
+        self.backgroundengine: engines.BackgroundEngine.BaseBackgroundEngine = (
+            backgroundengine[0]
+        )
+        self.backgroundengine.ctx = self
+
     def setup_dir(self):
         self.dir = f"output/{time.time()}"
         os.makedirs(self.dir)
@@ -55,27 +64,37 @@ class GenerationContext:
         # ⚠️ IMPORTANT NOTE: All methods called here are expected to be defined as abstract methods in the base classes, if not there is an issue with the engine implementation.
 
         # we start by loading the settings
-        progress = gr.Progress()
-        self.settingsengine.load()
         
+        self.progress(0.1,"Loading settings...")
+        self.settingsengine.load()
+
         self.setup_dir()
 
+        self.progress(0.2, "Generating script...")
         self.script = self.scriptengine.generate()
 
+        self.progress(0.3, "Generating synthtetizing voice...")
         self.timed_script = self.ttsengine.synthesize(
             self.script, self.get_file_path("tts.wav")
         )
 
+        self.assets = []
+
+        if not isinstance(self.backgroundengine, engines.NoneEngine):
+            self.progress(0.4, "Generating background...")
+            self.background = self.backgroundengine.get_background()
+            self.assets.append(self.background)
         self.assetsengine = [
-            engine for engine in self.assetsengine if not isinstance(engine, engines.NoneEngine)
+            engine
+            for engine in self.assetsengine
+            if not isinstance(engine, engines.NoneEngine)
         ]
         if len(self.assetsengine) > 0:
-            self.assets = self.assetsengineselector.get_assets()
-        else:
-            self.assets = []
-
+            self.progress(0.5, "Generating assets...")
+            self.assets.extend(self.assetsengineselector.get_assets())
 
         if not isinstance(self.captioningengine, engines.NoneEngine):
+            self.progress(0.6, "Generating captions...")
             self.captions = self.captioningengine.get_captions()
         else:
             self.captions = []
@@ -84,9 +103,9 @@ class GenerationContext:
 
         # we render to a file called final.mp4
         # using moviepy CompositeVideoClip
+        self.progress(0.7, "Rendering video...")
         clips = [*self.assets, *self.captions]
         clip = mp.CompositeVideoClip(clips, size=(self.width, self.height))
         audio = mp.AudioFileClip(self.get_file_path("tts.wav"))
         clip = clip.set_audio(audio)
         clip.write_videofile(self.get_file_path("final.mp4"), fps=60)
-