Update app.py

app.py

@@ -86,26 +86,26 @@ if __name__ == "__main__":
 
     colA, colB = st.columns(2)
 
-    with colA:
-        st.write("## Webcam image")
-        st.write("You can draw the mask on the image below.")
-        image = decode_image(webcam.getvalue())
-
-        canvas = make_canvas(image)
+    if webcam:
+        with colA:
+            st.write("## Webcam image")
+            st.write("You can draw the mask on the image below.")
+            image = decode_image(webcam.getvalue())
+    
+            canvas = make_canvas(image)
+        
+        if st.button("Inpaint"):
+            st.write("Start inpainting process")
+            mask_image = get_mask(np.array(canvas.image_data))
+            result = inpainting(image, mask_image, prompt, negative_prompt)
+            st.session_state["result"] = result
+        else:
+            result = None
     
-    if st.button("Inpaint"):
-        st.write("Start inpainting process")
-        mask_image = get_mask(np.array(canvas.image_data))
-        result = inpainting(image, mask_image, prompt, negative_prompt)
-        st.session_state["result"] = result
-    else:
-        result = None
-
-    with colB:
-        st.write("## Generated image")
-        st.write("The generated image will appear here.")
-        if webcam:
+        with colB:
+            st.write("## Generated image")
+            st.write("The generated image will appear here.")
             st.image(webcam)
-        if 'result' in st.session_state:
-            print("Showing result")
-            st.image(st.session_state["result"])
+            if 'result' in st.session_state:
+                print("Showing result")
+                st.image(st.session_state["result"])