Update app.py

app.py

@@ -38,12 +38,14 @@ def remove_background_bria(input_path):
 
 # Fungsi untuk memproses gambar menggunakan prompt
 def text_to_image(prompt):
+    os.makedirs("generated_images", exist_ok=True)
     image = sd_model(prompt).images[0]  # Use sd_model instead of pipe
     image_path = f"generated_images/{prompt.replace(' ', '_')}.png"
     image.save(image_path)
     return image, image_path
 
 def text_image_to_image(image, prompt):
+    os.makedirs("generated_images", exist_ok=True)
     # Convert input image to PIL Image for processing
     modified_image = sd_model(prompt, init_image=image, strength=0.75).images[0]  # Use sd_model here too
     image_path = f"generated_images/{prompt.replace(' ', '_')}_modified.png"
@@ -493,8 +495,16 @@ def create_canvas(canvas_size, bg_choice, custom_color):
 
 def apply_watermark(canvas, watermark_path):
     watermark = Image.open(watermark_path).convert("RGBA")
-    canvas.paste(watermark, (0, 0), watermark)
+    
+    # Calculate the position to center the watermark
+    canvas_width, canvas_height = canvas.size
+    watermark_width, watermark_height = watermark.size
+
+    x = (canvas_width - watermark_width) // 2
+    y = (canvas_height - watermark_height) // 2
 
+    canvas.paste(watermark, (x, y), watermark)
+    
 def save_image(canvas, output_folder, filename, output_format):
     output_ext = 'jpg' if output_format == 'JPG' else 'png'
     output_path = os.path.join(output_folder, f"{os.path.splitext(filename)[0]}.{output_ext}")