Update app.py

app.py

@@ -46,82 +46,81 @@ def set_image(img):
 
 st.title("AI Remover Objetos")
 
-with st.beta_container():
-    st.image(open("assets/demo.png", "rb").read())
-    st.markdown(
-        """
-        So you want to remove an object in your photo? You don't need to learn photo editing skills.
-        **Just draw over the parts of the image you want to remove, then our AI will remove them.**
-        """
-    )
-
-    uploaded_file = st.file_uploader("Choose image", accept_multiple_files=False, type=["png", "jpg", "jpeg"])
-
-    if uploaded_file is not None:
-
-        if st.session_state.reuse_image is not None:
-            img_input = Image.fromarray(st.session_state.reuse_image)
-        else:
-            bytes_data = uploaded_file.getvalue()
-            img_input = Image.open(BytesIO(bytes_data)).convert("RGBA")
-
-        stroke_width = st.slider("Brush size", 1, 100, 50)
+st.image(open("assets/demo.png", "rb").read())
 
-        st.write("**Now draw (brush) the part of image that you want to remove.**")
+st.markdown(
+    """
+    So you want to remove an object in your photo? You don't need to learn photo editing skills.
+    **Just draw over the parts of the image you want to remove, then our AI will remove them.**
+    """
+)
+uploaded_file = st.file_uploader("Choose image", accept_multiple_files=False, type=["png", "jpg", "jpeg"])
 
-        # Canvas size logic
-        canvas_bg = deepcopy(img_input)
-        aspect_ratio = canvas_bg.width / canvas_bg.height
-        streamlit_width = 720
+if uploaded_file is not None:
+    
+    if st.session_state.reuse_image is not None:
+        img_input = Image.fromarray(st.session_state.reuse_image)
+    else:
+        bytes_data = uploaded_file.getvalue()
+        img_input = Image.open(BytesIO(bytes_data)).convert("RGBA")
 
-        # Max width is 720. Resize the height to maintain its aspectratio.
-        if canvas_bg.width > streamlit_width:
-            canvas_bg = canvas_bg.resize((streamlit_width, int(streamlit_width / aspect_ratio)))
+    stroke_width = st.slider("Brush size", 1, 100, 50)
 
-        canvas_result = st_canvas(
-            stroke_color="rgba(255, 0, 255, 1)",
-            stroke_width=stroke_width,
-            background_image=canvas_bg,
-            width=canvas_bg.width,
-            height=canvas_bg.height,
-            drawing_mode="freedraw",
-            key="compute_arc_length", 
+    st.write("**Now draw (brush) the part of image that you want to remove.**")
+    
+    # Canvas size logic
+    canvas_bg = deepcopy(img_input)
+    aspect_ratio = canvas_bg.width / canvas_bg.height
+    streamlit_width = 720
+    
+    # Max width is 720. Resize the height to maintain its aspectratio.
+    if canvas_bg.width > streamlit_width:
+        canvas_bg = canvas_bg.resize((streamlit_width, int(streamlit_width / aspect_ratio)))
+    
+    canvas_result = st_canvas(
+        stroke_color="rgba(255, 0, 255, 1)",
+        stroke_width=stroke_width,
+        background_image=canvas_bg,
+        width=canvas_bg.width,
+        height=canvas_bg.height,
+        drawing_mode="freedraw",
+        key="compute_arc_length", 
+    )
+    
+    if canvas_result.image_data is not None:
+        im = np.array(Image.fromarray(canvas_result.image_data.astype(np.uint8)).resize(img_input.size))
+        background = np.where(
+            (im[:, :, 0] == 0) & 
+            (im[:, :, 1] == 0) & 
+            (im[:, :, 2] == 0)
         )
-
-        if canvas_result.image_data is not None:
-            im = np.array(Image.fromarray(canvas_result.image_data.astype(np.uint8)).resize(img_input.size))
-            background = np.where(
-                (im[:, :, 0] == 0) & 
-                (im[:, :, 1] == 0) & 
-                (im[:, :, 2] == 0)
-            )
-            drawing = np.where(
-                (im[:, :, 0] == 255) & 
-                (im[:, :, 1] == 0) & 
-                (im[:, :, 2] == 255)
+        drawing = np.where(
+            (im[:, :, 0] == 255) & 
+            (im[:, :, 1] == 0) & 
+            (im[:, :, 2] == 255)
+        )
+        im[background]=[0,0,0,255]
+        im[drawing]=[0,0,0,0] # RGBA
+        
+        reuse = False
+        
+        if st.button('Submit'):
+            
+            with st.spinner("AI is doing the magic!"):
+                output = process_inpaint(np.array(img_input), np.array(im)) #TODO Put button here
+                img_output = Image.fromarray(output).convert("RGB")
+            
+            st.write("AI has finished the job!")
+            st.image(img_output)
+            # reuse = st.button('Edit again (Re-use this image)', on_click=set_image, args=(inpainted_img, ))
+            
+            uploaded_name = os.path.splitext(uploaded_file.name)[0]
+            image_download_button(
+                pil_image=img_output,
+                filename=uploaded_name,
+                fmt="jpg",
+                label="Download Image"
             )
-            im[background]=[0,0,0,255]
-            im[drawing]=[0,0,0,0] # RGBA
-
-            reuse = False
-
-            if st.button('Submit'):
-
-                with st.spinner("AI is doing the magic!"):
-                    output = process_inpaint(np.array(img_input), np.array(im)) #TODO Put button here
-                    img_output = Image.fromarray(output).convert("RGB")
-
-                st.write("AI has finished the job!")
-                st.image(img_output)
-                # reuse = st.button('Edit again (Re-use this image)', on_click=set_image, args=(inpainted_img, ))
-
-                uploaded_name = os.path.splitext(uploaded_file.name)[0]
-                image_download_button(
-                    pil_image=img_output,
-                    filename=uploaded_name,
-                    fmt="jpg",
-                    label="Download Image"
-                )
-
-                st.info("**TIP**: If the result is not perfect, you can download it then "
-                        "upload then remove the artifacts.")
+                
+            st.info("**TIP**: If the result is not perfect, you can download it then "
+                    "upload then remove the artifacts.")