Update app.py

app.py

@@ -46,81 +46,82 @@ def set_image(img):
 
 st.title("AI Remover Objetos")
 
-st.image(open("assets/demo.png", "rb").read())
+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.**
+        """
+    )
 
-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"])
+    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")
+    if uploaded_file is not None:
 
-    stroke_width = st.slider("Brush size", 1, 100, 50)
+        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")
 
-    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)
-        )
-        drawing = np.where(
-            (im[:, :, 0] == 255) & 
-            (im[:, :, 1] == 0) & 
-            (im[:, :, 2] == 255)
+        stroke_width = st.slider("Brush size", 1, 100, 50)
+
+        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", 
         )
-        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"
+
+        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)
             )
-                
-            st.info("**TIP**: If the result is not perfect, you can download it then "
-                    "upload then remove the artifacts.")
+            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"
+                )
+
+                st.info("**TIP**: If the result is not perfect, you can download it then "
+                        "upload then remove the artifacts.")