Update app.py

app.py

@@ -43,7 +43,6 @@ def infer(color_prompt, phone_type_prompt, design_prompt):
         else:
             raise Exception(f"API Error: {response.status_code}")
 
-# Function to save the design with a caption
 def save_design(image, color_prompt, phone_type_prompt, design_prompt):
     # Convert NumPy array to PIL Image
     if isinstance(image, np.ndarray):
@@ -61,7 +60,7 @@ def save_design(image, color_prompt, phone_type_prompt, design_prompt):
         font = ImageFont.load_default()
 
     # Calculate position for text
-    text_width, text_height = draw.multiline_textsize(caption, font=font)
+    text_width, text_height = font.getsize_multiline(caption)  # Updated function
     position = (10, image.height - text_height - 10)  # Bottom-left corner with padding
 
     # Add caption to the image
@@ -72,6 +71,7 @@ def save_design(image, color_prompt, phone_type_prompt, design_prompt):
     image.save(file_path)
     return f"Design saved as {file_path}!"
 
+
 # HTML-based carousel content
 carousel_html = """
 <div class="carousel">