Update app.py

app.py

@@ -1,265 +1,272 @@
-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 = []
-    
-    # Setup initial parameters
-    image_width, image_height = int(width_choice), int(height_choice)
-    max_width = image_width - 40  # Margin of 20 pixels on each side
-    current_text = ""
-    
-    # Create the background
-    background = Image.new("RGB", (image_width, image_height), color=background_color)
-    
-    # Calculate the maximum width and adjust font size if necessary
-    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)
-    
-    # Generate frames for the typing effect
-    for char in code_text:
-        current_text += char
-
-        if format_choice == "Paragraph":
-            wrapped_lines = textwrap.wrap(current_text, width=max_width // font.getlength(' '))
-        else:  # Programming
-            wrapped_lines = current_text.splitlines()
-
-        # Copy the background image for each frame
-        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
-
-        # Convert to numpy array for OpenCV
-        frame = np.array(image)
-        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
-        video_frames.append(frame)
-    
-    # Create a video writer
-    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()
-
-    # Adjust video speed
-    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)  # Default to 1.0 if video_speed is not found in the dictionary
-    
-    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"
-
-    # Add sound if a sound choice is selected
-    if sound_choice and sound_choice != "No Sound":
-        video = mp.VideoFileClip(video_filename)
-        audio = mp.AudioFileClip(f"type-sounds/{sound_choice}")
-        
-        # Loop the audio to match the duration of the video
-        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"
-    
-    # Add custom audio if provided
-    if custom_audio:
-        video = mp.VideoFileClip(video_filename)
-        audio = mp.AudioFileClip(custom_audio)
-        
-        # Loop the custom audio to match the duration of the video
-        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"
-    
-    # Apply video quality enhancement if enabled
-    if enhance_quality:
-        video = mp.VideoFileClip(video_filename)
-        video = video.fx(vfx.resize, height=720)  # Resize video to enhance quality
-        video = video.fx(vfx.colorx, 1.2)  # Increase contrast
-        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)
-
-# Create Gradio interface
-format_choice = gr.Dropdown(
-    choices=["Paragraph", "Programming"],
-    value="Paragraph",
-    label="Text Format"
-)
-
-line_spacing = gr.Dropdown(
-    choices=[1.0, 1.15, 1.5, 2.0, 2.5, 3.0],
-    value=1.15,
-    label="Line Spacing"
-)
-
-font_choice = gr.Dropdown(
-    choices=[
-        "DejaVuMathTeXGyre.ttf", 
-        "FiraCode-Medium.ttf", 
-        "InputMono-Light.ttf",
-        "JetBrainsMono-Thin.ttf", 
-        "ProggyCrossed Regular Mac.ttf", 
-        "SourceCodePro-Black.ttf", 
-        "arial.ttf", 
-        "calibri.ttf", 
-        "mukta-malar-extralight.ttf", 
-        "noto-sans-arabic-medium.ttf", 
-        "times new roman.ttf",
-        "ANGSA.ttf",
-        "Book-Antiqua.ttf",
-        "CONSOLA.TTF",
-        "COOPBL.TTF",
-        "Rockwell-Bold.ttf",
-        "Candara Light.TTF",
-        "Carlito-Regular.ttf Carlito-Regular.ttf",
-        "Castellar.ttf",
-        "Courier New.ttf",
-        "LSANS.TTF",
-        "Lucida Bright Regular.ttf",
-        "TRTempusSansITC.ttf",
-        "Verdana.ttf",
-        "bell-mt.ttf",
-        "eras-itc-light.ttf",
-        "fonnts.com-aptos-light.ttf",
-        "georgia.ttf",
-        "segoeuithis.ttf",
-        "youyuan.TTF",
-        "TfPonetoneExpanded-7BJZA.ttf",
-    ],
-    value="SourceCodePro-Black.ttf",
-    label="Select Font"
-)
-
-font_size = gr.Dropdown(
-    choices=["16", "18", "20", "22", "24"],
-    value="18",
-    label="Font Size"
-)
-
-width_choice = gr.Dropdown(
-    choices=["400","800", "1024", "1280", "1920"],
-    value="800",
-    label="Video Width"
-)
-
-height_choice = gr.Dropdown(
-    choices=["400", "720", "1080", "1440", "2160"],
-    value="400",
-    label="Video Height"
-)
-
-sound_choice = gr.Dropdown(
-    choices=["No Sound",
-             "Mediumspeed Typing.mp3", 
-             "Speed Typing.mp3",
-             "Bass Typing.mp3",
-             "Bay Typing.mp3",
-             "Crack Typing.mp3",
-             "Deep Sence Typing.mp3",
-             "Flacking Typing.mp3",
-             "Flaw Typing.mp3",
-             "Focused Typing.mp3",
-             "K55 Typing.mp3",
-             "Laptop Typing.mp3",
-             "NDC Typing.mp3",
-             "RedMECH Typing.mp3",
-             "Smooth Typing.mp3",
-             "Stop Tpying.mp3",
-            ],
-    value="No Sound",
-    label="Typing Sound"
-)
-custom_audio = gr.File(
-    label="Upload Custom Audio SFX🔊",
-    type="filepath"
-)
-
-background_color = gr.Dropdown(
-    choices=["black", "white", "darkblue", "orange", "green"],
-    value="black",
-    label="Background Color"
-)
-
-text_color = gr.Dropdown(
-    choices=["black", "white", "darkblue", "orange", "green"],
-    value="white",
-    label="Text Color"
-)
-
-enhance_quality = gr.Checkbox(
-    label="Enhance Video Quality"
-)
-
-video_speed = gr.Dropdown(
-    choices=["1x", "1.25x", "1.5x", "1.75x", "2x"],
-    value="1x",
-    label="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..."),
-        format_choice,
-        line_spacing,
-        width_choice,
-        height_choice,
-        font_choice,
-        font_size,
-        sound_choice,
-        custom_audio,
-        background_color,
-        text_color,
-        enhance_quality,
-        video_speed
-    ],
-    
-    outputs=gr.Video(label="Typing Video"),
-    title="Type Bytes🐧",
-    css=css,
-    theme="bethecloud/storj_theme",
-)
-
-if __name__ == "__main__": 
-    iface.launch(share=True)
+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)
+    
+    video_frames = []
+    
+    # Setup initial parameters
+    image_width, image_height = int(width_choice), int(height_choice)
+    max_width = image_width - 40  # Margin of 20 pixels on each side
+    max_height = image_height - 40  # Margin of 20 pixels on top and bottom
+    current_text = ""
+    
+    # Create the background
+    background = Image.new("RGB", (image_width, image_height), color=background_color)
+    
+    # Load and adjust the font
+    font = ImageFont.truetype(font_path, font_size)
+    
+    # Adjust font size if necessary to fit text within image height
+    def adjust_font_size(text, font, max_width, max_height):
+        while True:
+            wrapped_lines = textwrap.wrap(text, width=max_width // font.getsize(' ')[0])
+            text_height = sum(font.getsize(line)[1] for line in wrapped_lines)
+            
+            if text_height <= max_height:
+                return font
+            
+            # Reduce font size
+            font_size -= 1
+            font = ImageFont.truetype(font_path, font_size)
+    
+    font = adjust_font_size(code_text, font, max_width, max_height)
+    
+    # Generate frames for the typing effect
+    for char in code_text:
+        current_text += char
+
+        if format_choice == "Paragraph":
+            wrapped_lines = textwrap.wrap(current_text, width=max_width // font.getsize(' ')[0])
+        else:  # Programming
+            wrapped_lines = current_text.splitlines()
+
+        # Copy the background image for each frame
+        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.getsize(line)[1]
+            y_position += line_height * line_spacing
+
+        # Convert to numpy array for OpenCV
+        frame = np.array(image)
+        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
+        video_frames.append(frame)
+    
+    # Create a video writer
+    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()
+
+    # Adjust video speed
+    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)  # Default to 1.0 if video_speed is not found in the dictionary
+    
+    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"
+
+    # Add sound if a sound choice is selected
+    if sound_choice and sound_choice != "No Sound":
+        video = mp.VideoFileClip(video_filename)
+        audio = mp.AudioFileClip(f"type-sounds/{sound_choice}")
+        
+        # Loop the audio to match the duration of the video
+        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"
+    
+    # Add custom audio if provided
+    if custom_audio:
+        video = mp.VideoFileClip(video_filename)
+        audio = mp.AudioFileClip(custom_audio)
+        
+        # Loop the custom audio to match the duration of the video
+        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"
+    
+    # Apply video quality enhancement if enabled
+    if enhance_quality:
+        video = mp.VideoFileClip(video_filename)
+        video = video.fx(vfx.resize, height=720)  # Resize video to enhance quality
+        video = video.fx(vfx.colorx, 1.2)  # Increase contrast
+        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)
+
+# Create Gradio interface
+format_choice = gr.Dropdown(
+    choices=["Paragraph", "Programming"],
+    value="Paragraph",
+    label="Text Format"
+)
+
+line_spacing = gr.Dropdown(
+    choices=[1.0, 1.15, 1.5, 2.0, 2.5, 3.0],
+    value=1.15,
+    label="Line Spacing"
+)
+
+font_choice = gr.Dropdown(
+    choices=[
+        "DejaVuMathTeXGyre.ttf", 
+        "FiraCode-Medium.ttf", 
+        "InputMono-Light.ttf",
+        "JetBrainsMono-Thin.ttf", 
+        "ProggyCrossed Regular Mac.ttf", 
+        "SourceCodePro-Black.ttf", 
+        "arial.ttf", 
+        "calibri.ttf", 
+        "mukta-malar-extralight.ttf", 
+        "noto-sans-arabic-medium.ttf", 
+        "times new roman.ttf",
+        "ANGSA.ttf",
+        "Book-Antiqua.ttf",
+        "CONSOLA.TTF",
+        "COOPBL.TTF",
+        "Rockwell-Bold.ttf",
+        "Candara Light.TTF",
+        "Carlito-Regular.ttf",
+        "Castellar.ttf",
+        "Courier New.ttf",
+        "LSANS.TTF",
+        "Lucida Bright Regular.ttf",
+        "TRTempusSansITC.ttf",
+        "Verdana.ttf",
+        "bell-mt.ttf",
+        "eras-itc-light.ttf",
+        "fonnts.com-aptos-light.ttf",
+        "georgia.ttf",
+        "segoeuithis.ttf",
+        "youyuan.TTF",
+        "TfPonetoneExpanded-7BJZA.ttf",
+    ],
+    value="SourceCodePro-Black.ttf",
+    label="Select Font"
+)
+
+font_size = gr.Dropdown(
+    choices=["16", "18", "20", "22", "24"],
+    value="18",
+    label="Font Size"
+)
+
+width_choice = gr.Dropdown(
+    choices=["400","800", "1024", "1280", "1920"],
+    value="800",
+    label="Video Width"
+)
+
+height_choice = gr.Dropdown(
+    choices=["400", "720", "1080", "1440", "2160"],
+    value="400",
+    label="Video Height"
+)
+
+sound_choice = gr.Dropdown(
+    choices=["No Sound",
+             "Mediumspeed Typing.mp3", 
+             "Speed Typing.mp3",
+             "Bass Typing.mp3",
+             "Bay Typing.mp3",
+             "Crack Typing.mp3",
+             "Deep Sence Typing.mp3",
+             "Flacking Typing.mp3",
+             "Flaw Typing.mp3",
+             "Focused Typing.mp3",
+             "K55 Typing.mp3",
+             "Laptop Typing.mp3",
+             "NDC Typing.mp3",
+             "RedMECH Typing.mp3",
+             "Smooth Typing.mp3",
+             "Stop Tpying.mp3",
+            ],
+    value="No Sound",
+    label="Typing Sound"
+)
+custom_audio = gr.File(
+    label="Upload Custom Audio SFX🔊",
+    type="filepath"
+)
+
+background_color = gr.Dropdown(
+    choices=["black", "white", "darkblue", "orange", "green"],
+    value="black",
+    label="Background Color"
+)
+
+text_color = gr.Dropdown(
+    choices=["white", "black", "blue", "orange", "red"],
+    value="white",
+    label="Text Color"
+)
+
+enhance_quality = gr.Checkbox(
+    value=False,
+    label="Enhance Video Quality"
+)
+
+video_speed = gr.Dropdown(
+    choices=["1x", "1.25x", "1.5x", "1.75x", "2x"],
+    value="1x",
+    label="Video Speed"
+)
+
+interface = gr.Interface(
+    fn=generate_video,
+    inputs=[gr.Textbox(lines=10, placeholder="Enter code here...", label="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],
+    
+    outputs=gr.Video(label="Typing Video"),
+    title="Type Bytes🐧",
+    css=css,
+    theme="bethecloud/storj_theme",
+)
+
+if __name__ == "__main__": 
+    interface.launch(share=True)