Return correct value from video creator

textprocessor.py

@@ -41,6 +41,7 @@ class TextProcessor:
     def get_json_scenes(self,
                         prompt: str) -> Dict:
         gpt_prompt = context_prompt.replace("$TRANSCRIPTION", prompt)
+        print("gpt_prompt", gpt_prompt)
         response = openai.Completion.create(
             model=self.model,
             prompt=gpt_prompt,
@@ -51,6 +52,7 @@ class TextProcessor:
             presence_penalty=self.presence_penalty
         )
         scenes = json.loads(response["choices"][0]["text"])
+        print("scenes", scenes)
         if (type(scenes) == list):
             scenes = {i: d for i, d in enumerate(scenes)}
         return scenes

videocreator.py

@@ -11,7 +11,7 @@ class VideoCreator:
         self.tts_pipeline = tts_pipeline
         self.image_pipeline = image_pipeline
         
-    def create_video(self, scenes: Dict) -> Dict:
+    def create_video(self, scenes: Dict) -> str:
         videos_dict = {}
         for index, scene in enumerate(scenes):
             video_scene = self._create_video_from_scene(scenes[scene])
@@ -38,10 +38,11 @@ class VideoCreator:
         image_output.save("img.png")
         return "img.png"
 
-    def _merge_videos(self, videos_dict: Dict):
+    def _merge_videos(self, videos_dict: Dict) -> str:
         videos_to_concatenate = []
         for video in range(len(videos_dict)):
             video_clip = VideoFileClip(videos_dict[video])
             videos_to_concatenate.append(video_clip)
         final_video = concatenate_videoclips(videos_to_concatenate)
-        final_video.write_videofile("final_video.mp4") 
+        final_video.write_videofile("final_video.mp4")
+        return "final_video.mp4"