Update app_dialogue.py

app_dialogue.py

@@ -71,18 +71,21 @@ import tempfile
 def convert_to_rgb_pil(image):
     print(f"***** convert_to_rgb_pil ******")
     print(f"params: image is - {image}")
-    if image.mode == "RGB":
-        return image
-
-    image_rgba = image.convert("RGBA")
-    background = Image.new("RGBA", image_rgba.size, (255, 255, 255))
-    alpha_composite = Image.alpha_composite(background, image_rgba)
-    alpha_composite = alpha_composite.convert("RGB")
-
+    #if image.mode == "RGB":
+    #    return image
     # Save the converted image to a temporary file
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
     temp_file_path = temp_file.name
-    alpha_composite.save(temp_file_path)
+
+    if image.mode != "RGB":
+        image_rgba = image.convert("RGBA")
+        background = Image.new("RGBA", image_rgba.size, (255, 255, 255))
+        alpha_composite = Image.alpha_composite(background, image_rgba)
+        alpha_composite = alpha_composite.convert("RGB")
+        alpha_composite.save(temp_file_path)
+    else:
+        image.save(temp_file_path)
+        
     temp_file.close()
     print(f"# Return the path to the saved image as - {temp_file_path}")
     return temp_file_path  # Return the path to the saved image