Update app.py

app.py

@@ -16,7 +16,7 @@ with open("emojis.json") as f:
 # Directory where emoji PNGs are stored
 emoji_dir = "emoji_pngs"
 
-def preprocess_image(image, emoji_name=None):
+def preprocess_image(image, emoji_name=None, enable_emoji=True):
     # Ensure the image is square by padding it
     size = max(image.size)
     new_image = Image.new("RGBA", (size, size), (255, 255, 255, 0))
@@ -26,21 +26,25 @@ def preprocess_image(image, emoji_name=None):
     new_image_gif = new_image.resize((256, 256), Image.LANCZOS)  # Resize for GIF
     new_image_video = new_image.resize((720, 720), Image.LANCZOS)  # Resize for Video
 
-    # Use the provided emoji or select a random one
-    if not emoji_name:
-        emoji_name = random.choice(list(emojis.keys()))
-    emoji_path = os.path.join(emoji_dir, f"{emoji_name}.png")
-    if os.path.exists(emoji_path):
-        emoji_image = Image.open(emoji_path).convert("RGBA")
-        emoji_image_gif = emoji_image.resize((256, 256), Image.LANCZOS)
-        emoji_image_video = emoji_image.resize((720, 720), Image.LANCZOS)
-
-        new_image_gif.paste(emoji_image_gif, (0, 0), emoji_image_gif)
-        new_image_video.paste(emoji_image_video, (0, 0), emoji_image_video)
+    # Only add emoji if enabled
+    if enable_emoji:
+        # Use the provided emoji or select a random one
+        if not emoji_name:
+            emoji_name = random.choice(list(emojis.keys()))
+        emoji_path = os.path.join(emoji_dir, f"{emoji_name}.png")
+        if os.path.exists(emoji_path):
+            emoji_image = Image.open(emoji_path).convert("RGBA")
+            emoji_image_gif = emoji_image.resize((256, 256), Image.LANCZOS)
+            emoji_image_video = emoji_image.resize((720, 720), Image.LANCZOS)
+
+            new_image_gif.paste(emoji_image_gif, (0, 0), emoji_image_gif)
+            new_image_video.paste(emoji_image_video, (0, 0), emoji_image_video)
+    else:
+        emoji_name = "none"  # Use a placeholder when emojis are disabled
 
     return new_image_gif, new_image_video, emoji_name
 
-def generate_media(editor1_output, editor2_output, transition_type, fps, transition_speed, color, thickness):
+def generate_media(editor1_output, editor2_output, transition_type, fps, transition_speed, color, thickness, enable_emoji):
     frames = []
     gif_duration = 100  # Duration for each frame in the GIF in milliseconds
     total_frames_gif = 18  # Total number of frames for GIF
@@ -52,8 +56,8 @@ def generate_media(editor1_output, editor2_output, transition_type, fps, transit
         img2 = editor2_output["composite"].convert('RGBA')
 
         # Preprocess images to make them square and same size, select emojis once
-        img1_gif, img1_video, random_emoji1 = preprocess_image(img1)
-        img2_gif, img2_video, random_emoji2 = preprocess_image(img2)
+        img1_gif, img1_video, random_emoji1 = preprocess_image(img1, enable_emoji=enable_emoji)
+        img2_gif, img2_video, random_emoji2 = preprocess_image(img2, enable_emoji=enable_emoji)
 
         # Set size for the GIF
         gif_size = (256, 256)
@@ -233,6 +237,7 @@ with gr.Blocks() as iface:
 
     with gr.Row():
         transition_type = gr.Radio(["slide", "rotate"], label="Transition Type", value="slide")
+        enable_emoji = gr.Checkbox(label="Enable Emojis", value=True, info="Toggle emoji overlays on/off")
         generate_button = gr.Button("Generate GIF & MP4")
 
     with gr.Accordion("Advanced Settings", open=False):
@@ -247,9 +252,9 @@ with gr.Blocks() as iface:
         gif_download = gr.File(label="Download GIF", elem_id="gif_download", visible=True)
         mp4_download = gr.File(label="Download MP4", elem_id="mp4_download", visible=True)
 
-    def handle_generation(editor1_output, editor2_output, transition_type, fps, transition_speed, color, thickness):
+    def handle_generation(editor1_output, editor2_output, transition_type, fps, transition_speed, color, thickness, enable_emoji):
         try:
-            gif_path, mp4_path = generate_media(editor1_output, editor2_output, transition_type, fps, transition_speed, color, thickness)
+            gif_path, mp4_path = generate_media(editor1_output, editor2_output, transition_type, fps, transition_speed, color, thickness, enable_emoji)
             return gif_path, gif_path, mp4_path
         except Exception as e:
             print(f"Error in handle_generation: {e}")
@@ -257,9 +262,9 @@ with gr.Blocks() as iface:
 
     generate_button.click(
         handle_generation,
-        inputs=[image_editor1, image_editor2, transition_type, fps, transition_speed, color, thickness],
+        inputs=[image_editor1, image_editor2, transition_type, fps, transition_speed, color, thickness, enable_emoji],
         outputs=[gif_display, gif_download, mp4_download]
     )
 
 # Launch the interface
-iface.launch(share=False)
+iface.launch(share=False)