File size: 2,728 Bytes
6f98489 ce79011 6f98489 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 |
"""
@author: idoia lerchundi
"""
from io import BytesIO
from PIL import Image,ImageFile
import streamlit as st
from rembg import remove
#from io import BytesIO
def local_css(file_name):
with open(file_name) as f:
st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
st.set_page_config(
page_title="Streamlit iCodeIdoia",
page_icon="images/ilpicon1.png",layout="wide",initial_sidebar_state="expanded"
)
st.image("images/banner.jpg")
# ---- LOAD
local_css("styles/style.css")
#Input and output as bytes DEMO
def remove_bgdemo(input_path):
output_path = 'output/demooutput.png'
with open(input_path, 'rb') as i:
with open(output_path, 'wb') as o:
input = i.read()
output = remove(input)
#saves output
saving = o.write(output)
return output_path
def remove_bg(input_path):
output_path = 'output/output.png'
with open(input_path, 'rb') as i:
with open(output_path, 'wb') as o:
input = i.read()
output = remove(input)
#saves output
#remove the output file
saving = o.write(output)
return output_path
def main():
# ---- TABS
tab1, tab2 = st.tabs(["Demo","Application"])
with tab1:
# Handle first image
url = 'samples/1.jpg'
st.subheader("Remove background from image demo")
img_description = st.text('Image background will be removed.')
if st.button('Remove Background Demo'):
st.text("Selected image.")
sel_image = Image.open(url)
st.image(sel_image)
generated_img = remove_bgdemo(url)
st.text("Image background removed.")
st.image(generated_img)
with tab2:
st.subheader("Remove background from image, place image in samples folder.")
uploaded_file = st.file_uploader("Upload an image. e.g. Select 1_for_super_resolution.jpg from samples folder.", type=['png', 'jpg', 'jpeg', 'gif'])
st.caption("Note: this app uses samples folder input path, please change code accordingly.")
if uploaded_file is not None:
#PIL
lr_image = Image.open(uploaded_file)
st.text("Your Uploaded Image")
st.image(lr_image)
imagepath = "output/output.png"
inputpath="output/input.png"
lr_image.save(inputpath)
if st.button('Remove background'):
st.text("Please wait a few seconds. Monitor top right of your screen.")
imagepath = imagepath + uploaded_file.name
generated_img = remove_bg(inputpath)
st.text("Image background removed.")
st.image(generated_img)
if __name__ == "__main__":
main() |