Update app.py

app.py

@@ -46,11 +46,17 @@ def crop_face(base64_image):
 
 
 def image_to_base64(image):
-    # Convert PIL Image to Bytes
-    buffered = BytesIO()
-    image.save(buffered, format="JPEG")
-    img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
-    return img_str
+    try:
+        # Prepare the buffer to save the image
+        buffered = BytesIO()
+        # Save the image to the buffer in JPEG format
+        image.save(buffered, format="JPEG")
+        # Encode the buffered image to Base64
+        img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
+        return img_str
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
 
 # Define the Gradio interfaces
 face_crop_interface = gr.Interface(