Update app.py

app.py

@@ -4,13 +4,20 @@ import base64
 import io
 
 def convert_base64_to_jpg(image_base64):
-    # Decodifica o base64
-    img_data = base64.b64decode(image_base64)
-    # Cria a imagem
-    image = Image.open(io.BytesIO(img_data))
-    # Salva a imagem como JPG
-    image.save('converted_image.jpg', 'JPEG')
-    return 'converted_image.jpg'
+    # Se o base64 contiver o prefixo 'data:image/png;base64,', remova-o
+    if image_base64.startswith("data:image"):
+        image_base64 = image_base64.split(",")[1]
+
+    try:
+        # Decodifica o base64
+        img_data = base64.b64decode(image_base64)
+        # Cria a imagem
+        image = Image.open(io.BytesIO(img_data))
+        # Salva a imagem como JPG
+        image.save('converted_image.jpg', 'JPEG')
+        return 'converted_image.jpg'
+    except Exception as e:
+        return f"Erro ao processar a imagem: {str(e)}"
 
 # Cria a interface Gradio
 iface = gr.Interface(fn=convert_base64_to_jpg,