Spaces:
Sleeping
Sleeping
| from moviepy.editor import VideoFileClip, concatenate_videoclips | |
| files = { | |
| 1: "1.mp4", 2: "2.mp4", 3: "3.mp4", 4: "4.mp4", 5: "5.mp4", | |
| 6: "6.mp4", 7: "7.mp4", 8: "8.mp4", 9: "9.mp4", | |
| 10: "10.mp4", 11: "11.mp4", 12: "12.mp4", 13: "13.mp4", 14: "14.mp4", | |
| 15: "15.mp4", 16: "16.mp4", 17: "17.mp4", 18: "18.mp4", 19: "19.mp4", | |
| 20: "20.mp4", 30: "30.mp4", 40: "40.mp4", 50: "50.mp4", | |
| 60: "60.mp4", 70: "70.mp4", 80: "80.mp4", 90: "90.mp4", | |
| 100: "100.mp4", 1000: "1000.mp4", | |
| 100000: "100000.mp4" | |
| } | |
| def number_to_clips(n): | |
| components = [] | |
| if n >= 100000: | |
| lakh = n // 100000 | |
| if lakh > 0: | |
| components.append(lakh) | |
| components.append(100000) | |
| n %= 100000 | |
| if n >= 1000: | |
| thousand = n // 1000 | |
| if thousand > 0: | |
| components.append(thousand) | |
| components.append(1000) | |
| n %= 1000 | |
| if n >= 100: | |
| hundred = n // 100 | |
| if hundred > 0: | |
| components.append(hundred) | |
| components.append(100) | |
| n %= 100 | |
| if n > 20: | |
| tens = (n // 10) * 10 | |
| if tens > 0: | |
| components.append(tens) | |
| n %= 10 | |
| if n > 0: | |
| components.append(n) | |
| return components | |
| def generate_clip_sequence(n): | |
| components = number_to_clips(n) | |
| clip_sequence = [] | |
| for component in components: | |
| if component in files: | |
| clip_sequence.append(files[component]) | |
| else: | |
| # Break down further if component is not in files | |
| sub_components = number_to_clips(component) | |
| clip_sequence.extend([files[sub] for sub in sub_components]) | |
| return clip_sequence | |
| def trimmed_clip_path(clip, avatar_path="custom_clips"): | |
| print(f"{avatar_path}/{clip.split('.mp4')[0]}_cut.mp4") | |
| return f"{avatar_path}/{clip.split('.mp4')[0]}_cut.mp4" | |
| def get_clip_duration(clip, avatar_path="custom_clips"): | |
| if avatar_path == "custom_clips": | |
| avatar_path = "custom_clips" | |
| else: | |
| avatar_path = "template_clips" | |
| try: | |
| clip = VideoFileClip(trimmed_clip_path(clip, avatar_path=avatar_path)) | |
| return clip.duration | |
| except Exception as e: | |
| print(f"Error loading clip {trimmed_clip_path(clip)}: {e}") | |
| return 0 | |
| def cut_clip(clip, st, en): | |
| """ | |
| Cuts the given clip from start time `st` to end time `en`. | |
| """ | |
| return clip.subclip(st, en) | |
| def create_video_old(number,st, en, output_file="output.mp4"): | |
| """ | |
| Creates a video for the given number using pre-defined video clips. | |
| """ | |
| # Special files | |
| intro_file = "trimmed_clips/Intro_cut.mp4" | |
| conclusion_file = "trimmed_clips/Conclusion_cut.mp4" | |
| clips_to_combine = [intro_file] | |
| clips = generate_clip_sequence(number) | |
| for clip in clips: | |
| clips_to_combine.append(trimmed_clip_path(clip)) | |
| clips_to_combine.append(conclusion_file) | |
| # Load and combine video clips | |
| video_clips = [] | |
| for clip_path in clips_to_combine: | |
| try: | |
| clip = VideoFileClip(clip_path) | |
| clip = cut_clip(clip, st, clip.duration - en) | |
| video_clips.append(clip) | |
| except Exception as e: | |
| print(f"Error loading clip {clip_path}: {e}") | |
| # Concatenate the video clips | |
| if video_clips: | |
| final_video = concatenate_videoclips(video_clips) | |
| final_video.write_videofile(output_file, codec="libx264", audio_codec="aac") | |
| else: | |
| print("No clips to combine. Video creation failed.") | |
| def create_advanced_video(number, trim_settings, avatar, output_file="output.mp4"): | |
| if avatar == "custom": | |
| path = "custom_clips" | |
| intro_file = f"{path}/Intro_cut.mp4" | |
| conclusion_file = f"{path}/Conclusion_cut.mp4" | |
| else: | |
| path = "template_clips" | |
| intro_file = f"{path}/Intro_cut.mp4" | |
| conclusion_file = f"{path}/Conclusion_cut.mp4" | |
| print(f"Using avatar: {avatar}") | |
| clips_to_combine = [VideoFileClip(intro_file)] | |
| clips = generate_clip_sequence(number) | |
| video_clips = [] | |
| for i, clip_path in enumerate(clips): | |
| try: | |
| clip = VideoFileClip(trimmed_clip_path(clip_path, avatar_path=path)) | |
| trim_start, trim_end = trim_settings[i] | |
| trimmed_clip = cut_clip(clip, trim_start, clip.duration - trim_end) | |
| video_clips.append(trimmed_clip) | |
| except Exception as e: | |
| print(f"Error loading clip {clip_path}: {e}") | |
| clips_to_combine.extend(video_clips) | |
| clips_to_combine.append(VideoFileClip(conclusion_file)) | |
| if video_clips: | |
| final_video = concatenate_videoclips(clips_to_combine) | |
| final_video.write_videofile(output_file, codec="libx264", audio_codec="aac") | |
| else: | |
| print("No clips to combine. Video creation failed.") | |
| def combine_video(number, video_id, output_file="output.mp4"): | |
| path = f"{video_id}_clips" | |
| intro_file = f"{path}/Intro_cut.mp4" | |
| conclusion_file = f"{path}/Conclusion_cut.mp4" | |
| clips_to_combine = [intro_file] | |
| clips = generate_clip_sequence(number) | |
| for clip in clips: | |
| clips_to_combine.append(trimmed_clip_path(clip, avatar_path=path)) | |
| clips_to_combine.append(conclusion_file) | |
| # Load and combine video clips | |
| video_clips = [] | |
| for clip_path in clips_to_combine: | |
| try: | |
| clip = VideoFileClip(clip_path) | |
| video_clips.append(clip) | |
| except Exception as e: | |
| print(f"Error loading clip {clip_path}: {e}") | |
| # Concatenate the video clips | |
| if video_clips: | |
| final_video = concatenate_videoclips(video_clips) | |
| final_video.write_videofile(output_file, codec="libx264", audio_codec="aac") | |
| else: | |
| print("No clips to combine. Video creation failed.") |