Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import cv2 | |
| import numpy as np | |
| from PIL import Image, ImageDraw, ImageFont | |
| import textwrap | |
| import moviepy.editor as mp | |
| import moviepy.video.fx.all as vfx | |
| css = """ | |
| #col-container { | |
| margin: 0 auto; | |
| max-width: 290px; | |
| } | |
| """ | |
| def create_typing_video(code_text, format_choice, line_spacing, width_choice, height_choice, font_name="arial.ttf", font_size=18, frame_rate=10, sound_choice=None, custom_audio=None, background_color="black", text_color="white", enhance_quality=False, video_speed="1.0"): | |
| font_path = f"font/{font_name}" | |
| # Convert font_size to integer | |
| font_size = int(font_size) | |
| font = ImageFont.truetype(font_path, font_size) | |
| video_frames = [] | |
| image_width, image_height = int(width_choice), int(height_choice) | |
| max_width = image_width - 40 # Margin of 20 pixels on each side | |
| current_text = "" | |
| background = Image.new("RGB", (image_width, image_height), color=background_color) | |
| while True: | |
| wrapped_lines = textwrap.wrap(code_text, width=max_width // font.getlength(' ')) | |
| text_height = sum([font.getbbox(line)[3] - font.getbbox(line)[1] for line in wrapped_lines]) | |
| if text_height <= image_height - 40: | |
| break | |
| font_size -= 1 | |
| font = ImageFont.truetype(font_path, font_size) | |
| for char in code_text: | |
| current_text += char | |
| if format_choice == "Paragraph": | |
| wrapped_lines = textwrap.wrap(current_text, width=max_width // font.getlength(' ')) | |
| else: | |
| wrapped_lines = current_text.splitlines() | |
| image = background.copy() | |
| draw = ImageDraw.Draw(image) | |
| y_position = 20 | |
| for line in wrapped_lines: | |
| draw.text((20, y_position), line, font=font, fill=text_color) | |
| line_height = font.getbbox(line)[3] - font.getbbox(line)[1] | |
| y_position += line_height * line_spacing | |
| frame = np.array(image) | |
| frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR) | |
| video_frames.append(frame) | |
| video_filename = "typed_code_video.mp4" | |
| out = cv2.VideoWriter(video_filename, cv2.VideoWriter_fourcc(*"mp4v"), frame_rate, (image_width, image_height)) | |
| for frame in video_frames: | |
| out.write(frame) | |
| out.release() | |
| speed_factor = { | |
| "1x": 1.0, | |
| "1.25x": 1.25, | |
| "1.5x": 1.5, | |
| "1.75x": 1.75, | |
| "2x": 2.0 | |
| }.get(video_speed, 1.0) | |
| video = mp.VideoFileClip(video_filename).fx(vfx.speedx, factor=speed_factor) | |
| video.write_videofile("speed_adjusted_video.mp4", codec="libx264") | |
| video_filename = "speed_adjusted_video.mp4" | |
| if sound_choice and sound_choice != "No Sound": | |
| video = mp.VideoFileClip(video_filename) | |
| audio = mp.AudioFileClip(f"type-sounds/{sound_choice}") | |
| audio = audio.fx(mp.afx.audio_loop, duration=video.duration) | |
| video = video.set_audio(audio) | |
| video.write_videofile("typed_code_video_with_sound.mp4", codec="libx264") | |
| video_filename = "typed_code_video_with_sound.mp4" | |
| if custom_audio: | |
| video = mp.VideoFileClip(video_filename) | |
| audio = mp.AudioFileClip(custom_audio) | |
| audio = audio.fx(mp.afx.audio_loop, duration=video.duration) | |
| video = video.set_audio(audio) | |
| video.write_videofile("typed_code_video_with_custom_audio.mp4", codec="libx264") | |
| video_filename = "typed_code_video_with_custom_audio.mp4" | |
| if enhance_quality: | |
| video = mp.VideoFileClip(video_filename) | |
| video = video.fx(vfx.resize, height=720) | |
| video = video.fx(vfx.colorx, 1.2) | |
| video.write_videofile("enhanced_" + video_filename, codec="libx264") | |
| video_filename = "enhanced_" + video_filename | |
| return video_filename | |
| def generate_video(code_text, format_choice, line_spacing, width_choice, height_choice, font_choice, font_size, sound_choice, custom_audio, background_color, text_color, enhance_quality, video_speed): | |
| return create_typing_video(code_text, format_choice, line_spacing, width_choice, height_choice, font_name=font_choice, font_size=font_size, sound_choice=sound_choice, custom_audio=custom_audio, background_color=background_color, text_color=text_color, enhance_quality=enhance_quality, video_speed=video_speed) | |
| iface = gr.Interface( | |
| fn=generate_video, | |
| inputs=[ | |
| gr.Textbox(label="Enter Content", lines=10, placeholder="Enter the text to be displayed in the video..."), | |
| gr.Dropdown(choices=["Paragraph", "Programming"], value="Paragraph", label="Text Format"), | |
| gr.Dropdown(choices=[1.0, 1.15, 1.5, 2.0, 2.5, 3.0], value=1.5, label="Line Spacing"), | |
| gr.Dropdown(choices=["400", "800", "1024", "1280", "1920"], value="800", label="Video Width"), | |
| gr.Dropdown(choices=["400", "720", "1080", "1440", "2160"], value="400", label="Video Height"), | |
| gr.Dropdown(choices=["arial.ttf", "times new roman.ttf"], value="arial.ttf", label="Font Choice"), | |
| gr.Dropdown(choices=["16", "18", "20", "22", "24"], value="18", label="Font Size"), | |
| gr.Dropdown(choices=["No Sound", "Typing.mp3"], value="No Sound", label="Sound Choice"), | |
| gr.File(label="Upload Custom Audio SFX🔊", type="filepath"), | |
| gr.Dropdown(choices=["black", "white"], value="black", label="Background Color"), | |
| gr.Dropdown(choices=["black", "white"], value="white", label="Text Color"), | |
| gr.Checkbox(label="Enhance Video Quality"), | |
| gr.Dropdown(choices=["1x", "1.25x", "1.5x"], value="1x", label="Video Speed"), | |
| ], | |
| outputs=gr.Video(label="Typing Video"), | |
| title="Type Bytes🐧", | |
| css=css, | |
| ) | |
| if __name__ == "__main__": | |
| iface.launch(share=True) |