From 0385bacebf5c42ba773e79a8a368e33bdd5957b2 Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Fri, 7 Mar 2025 09:56:49 +0100
Subject: [PATCH] fix

---
 lollms/services/tts/open_ai_tts/lollms_openai_tts.py |  3 ++-
 lollms/services/tts/xtts/lollms_xtts.py              |  2 +-
 lollms/services/ttv/novita_ai/lollms_novita_ai.py    | 12 +++++-------
 3 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/lollms/services/tts/open_ai_tts/lollms_openai_tts.py b/lollms/services/tts/open_ai_tts/lollms_openai_tts.py
index 0edc4fe..08514c7 100644
--- a/lollms/services/tts/open_ai_tts/lollms_openai_tts.py
+++ b/lollms/services/tts/open_ai_tts/lollms_openai_tts.py
@@ -30,6 +30,7 @@ from lollms.utilities import PackageManager, find_next_available_filename
 from lollms.tts import LollmsTTS
 import subprocess
 import shutil
+from pathlib import Path
 from tqdm import tqdm
 import threading
 from io import BytesIO
@@ -53,7 +54,7 @@ class LollmsOpenAITTS(LollmsTTS):
     def __init__(
                     self, 
                     app:LollmsApplication,
-                    output_folder=None
+                    output_folder:str|Path=None
                     ):
         """
         Initializes the LollmsDalle binding.
diff --git a/lollms/services/tts/xtts/lollms_xtts.py b/lollms/services/tts/xtts/lollms_xtts.py
index 952df24..27658fd 100644
--- a/lollms/services/tts/xtts/lollms_xtts.py
+++ b/lollms/services/tts/xtts/lollms_xtts.py
@@ -64,7 +64,7 @@ class LollmsXTTS(LollmsTTS):
     def __init__(
                     self, 
                     app:LollmsApplication,
-                    output_folder=None
+                    output_folder:str|Path=None
                     ):
         """
         Initializes the LollmsDalle binding.
diff --git a/lollms/services/ttv/novita_ai/lollms_novita_ai.py b/lollms/services/ttv/novita_ai/lollms_novita_ai.py
index 0fa4e08..0295b6d 100644
--- a/lollms/services/ttv/novita_ai/lollms_novita_ai.py
+++ b/lollms/services/ttv/novita_ai/lollms_novita_ai.py
@@ -17,7 +17,7 @@ class LollmsNovitaAITextToVideo(LollmsTTV):
     def __init__(
                     self,
                     app:LollmsApplication,
-                    output_folder:str|Path
+                    output_folder:str|Path=None
                 ):
         """
         Initializes the NovitaAITextToVideo binding.
@@ -94,6 +94,8 @@ class LollmsNovitaAITextToVideo(LollmsTTV):
         """
         if model_name=="":
             model_name = self.model_name
+        if output_dir is None:
+            output_dir = self.output_folder
         url = f"{self.base_url}/txt2video"
         headers = {
             "Authorization": f"Bearer {self.service_config.api_key}",
@@ -148,12 +150,8 @@ class LollmsNovitaAITextToVideo(LollmsTTV):
                 output_dir = Path(output_dir)
                 file_name = output_dir/find_next_available_filename(output_dir, "vid_novita_")  # You can change the filename if needed
                 self.download_video(infos["videos"][0]["video_url"], file_name )
-            else:
-                
-                file_name = self.app.personality.personality_output_folder/find_next_available_filename(output_dir, "vid_novita_")  # You can change the filename if needed
-                self.download_video(infos["videos"][0]["video_url"], self.app.personality.personality_output_folder/"video_{}.mp4" )
-        return infos
-
+                return file_name
+        return None
 
     def settings_updated(self):
         pass