Update app.py

app.py

@@ -19,13 +19,13 @@ def image_download_button(pil_image, filename: str, fmt: str, label="Download"):
         raise Exception(f"Unknown image format (Available: {fmt} - case sensitive)")
     
     pil_format = "JPEG" if fmt == "jpg" else "PNG"
-    #file_format = "jpg" if fmt == "jpg" else "png"
-    file_format = "png"
-    #mime = "image/jpeg" if fmt == "jpg" else "image/png"
-    mime = "image/png"
+    file_format = "jpg" if fmt == "jpg" else "png"
+    mime = "image/jpeg" if fmt == "jpg" else "image/png"
+    
 
     buf = BytesIO()
-    pil_image.save(buf, format='PNG')
+    pil_image.save(buf, format=pil_format)
+
 
     
     return st.download_button(
@@ -102,7 +102,7 @@ if uploaded_file is not None:
             (im[:, :, 1] == 0) & 
             (im[:, :, 2] == 255)
         )
-        im[background] = [0, 0, 0, 0]
+        im[background] = [0, 0, 0, 255]
         im[drawing]=[0,0,0,0] # RGBA
         
         reuse = False
@@ -111,7 +111,7 @@ if uploaded_file is not None:
             
             with st.spinner("A IA está fazendo a mágica!"):
                 output = process_inpaint(np.array(img_input), np.array(im)) #TODO Put button here
-                img_output = Image.fromarray(output).convert("RGBA")
+                img_output = Image.fromarray(output).convert("RGB")