Update app.py

app.py

@@ -21,16 +21,12 @@ def create_typing_video(code_text, format_choice, line_spacing, width_choice, he
     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])
@@ -40,16 +36,14 @@ def create_typing_video(code_text, format_choice, line_spacing, width_choice, he
         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
+        else:
             wrapped_lines = current_text.splitlines()
 
-        # Copy the background image for each frame
         image = background.copy()
         draw = ImageDraw.Draw(image)
         
@@ -59,12 +53,10 @@ def create_typing_video(code_text, format_choice, line_spacing, width_choice, he
             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))
     
@@ -72,193 +64,66 @@ def create_typing_video(code_text, format_choice, line_spacing, width_choice, he
         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
+    }.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"
 
-    # 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 = 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)
 
-# 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.5,
-    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="Currently, it is recommended to use the default 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
+        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,
-    theme="bethecloud/storj_theme",
 )
 
 if __name__ == "__main__":