Update video2.py

video2.py

@@ -267,24 +267,25 @@ from moviepy.editor import VideoFileClip, AudioFileClip
 from moviepy.video.fx.speedx import speedx
 
 # video.py
-import os
-import rust_highlight
-
-CLIPS_DIR = os.path.join(os.environ.get('BASE_DIR', '/app/data'), 'clips')  # Adjust as needed for Hugging Face Spaces
-
 def video_func(id, lines, lang):
-    duration, audio_path = audio_func(id, lines, lang)  # Assuming audio_func is defined elsewhere
+    duration, audio_path = audio_func(id, lines, lang)
     if not duration or not audio_path:
         print("Failed to generate audio.")
         return None
+    
     TEXT = lines[id]
     print("-----------------------------------------------------------------------------")
     print(TEXT)
     
+    # CREATE CLIPS DIRECTORY IF IT DOESN'T EXIST
+    os.makedirs(CLIPS_DIR, exist_ok=True)
+    
+    # Call Rust function
     final_video_path = rust_highlight.generate_video_clip(id, TEXT, audio_path, duration, CLIPS_DIR)
-    if final_video_path is None:
-        print("Failed to generate video.")
-        return None
     
-    print(f"Final video saved at: {final_video_path}")
-    return final_video_path
+    if final_video_path:
+        print(f"Final video saved at: {final_video_path}")
+        return final_video_path
+    else:
+        print("Video generation failed.")
+        return None