Spaces:
Running
Running
| import numpy as np | |
| import pandas as pd | |
| import streamlit as st | |
| import os | |
| from datetime import datetime | |
| from PIL import Image | |
| from streamlit_drawable_canvas import st_canvas | |
| from io import BytesIO | |
| from copy import deepcopy | |
| from src.core import process_inpaint | |
| def image_download_button(pil_image, filename: str, fmt: str, label="Download"): | |
| if fmt not in ["jpg", "png"]: | |
| 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" | |
| mime = "image/jpeg" if fmt == "jpg" else "image/png" | |
| buf = BytesIO() | |
| pil_image.save(buf, format=pil_format) | |
| return st.download_button( | |
| label=label, | |
| data=buf.getvalue(), | |
| file_name=f'{filename}.{file_format}', | |
| mime=mime, | |
| ) | |
| if "button_id" not in st.session_state: | |
| st.session_state["button_id"] = "" | |
| if "color_to_label" not in st.session_state: | |
| st.session_state["color_to_label"] = {} | |
| if 'reuse_image' not in st.session_state: | |
| st.session_state.reuse_image = None | |
| def set_image(img): | |
| st.session_state.reuse_image = img | |
| st.title("AI Photo Object Removal") | |
| 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") | |
| # Resize the image while maintaining aspect ratio | |
| max_size = 2000 | |
| img_width, img_height = img_input.size | |
| if img_width > max_size or img_height > max_size: | |
| if img_width > img_height: | |
| new_width = max_size | |
| new_height = int((max_size / img_width) * img_height) | |
| else: | |
| new_height = max_size | |
| new_width = int((max_size / img_height) * img_width) | |
| img_input = img_input.resize((new_width, new_height)) | |
| 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", | |
| ) | |
| 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) | |
| ) | |
| 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)) | |
| img_output = Image.fromarray(output).convert("RGB") | |
| st.write("AI has finished the job!") | |
| st.image(img_output) | |
| 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.") | |
| # Markdown | |
| st.markdown(""" | |
| **Best AI Tools** | |
| [Nude AI](https://nudeai.beauty) • [Uncensored AI](https://uncensoredai.cc) • [AI Hentai Generator](https://aihentaigenerator.fun) • [Stable Diffusion Hentai](https://stable-diffusion-hentai.aihentaigenerator.fun) • [Bing Image Creator](https://bingimagecreator.online) • [NSFW AI Art](https://nsfwaiart.art) • [NSFW AI Chatbot](https://nsfw-ai-chatbot.online) • [NSFW AI World](https://nsfwai.world) • [AI Tools Directory](https://aitoolsdirectory.online) • [Viggle AI](https://viggleai.live) • [TDEE Calculator](https://tdeecalculator.online) • [Calculator App](https://calculatorapp.online) • [Compound Interest Calculator](https://compoundinterestcalculator.site) • [AI Story Generator](https://aistorygenerator.fun) • [Llama AI](https://llamaai.online) • [AI Art Free](https://aiartfree.online) • [Cek Khodam](https://cekkhodam.co) • [AI Detector](https://ai-detector.online) • [AI Cover](https://aicover.fun) • [Flux AI](https://flux-ai.online) • [Brat Generator](https://bratgenerator.org) • [AI Text Generator](https://aitextgenerator.live) • [Black Myth Wukong](https://blackmythwukong.buzz) • [Book Summary](https://booksummary.wiki) • [ChatGPT Gratuit](https://chatgptgratuit.chat) • [ChatGPT Online](https://chatgtponline.com) • [Gau Thai](https://gauthai.pro) • [Best Recipe](https://bestrecipe.pro) | |
| **Games** | |
| [Cookie Clicker Unblocked](https://cookie-clicker-unblocked.pro) • [Freaky Font](https://freakyfont.org) • [Chat Gratuito](https://chatgratuito.online) • [Incredibox Sprunki](https://incrediboxsprunki.pro) • [Moto X3M Unblocked](https://motox3munblocked.pro) • [Parkour Civilization](https://parkourcivilization.pro) • [Sled Rider 3D](https://sledrider3d.com) • [Snow Rider 3D](https://snow-rider-3d.pro) • [Sprunki Game](https://sprunkigame.pro) • [Stickman Hook Unblocked](https://stickmanhookunblocked.pro) • [Escape Road](https://escaperoad.fun) • [Sprunked](https://sprunked.pro) • [Abgerny](https://abgerny.buzz) • [Sprunki Phase](https://sprunki-phase.online) • [Block Blast](https://block-blast.pro) • [Traffic Racer](https://trafficracer.online) • [Football Bros](https://footballbros.online) • [Sprunki Parasite](https://sprunkiparasite.com) • [Sprunki Retake](https://sprunkiretake.online) • [Sprunki Mustard](https://sprunki-mustard.online) • [PVZ Fusion Download](https://pvz-fusion.online) • [My Femboy Roommate](https://myfemboy-roommate.com) • [Homicipher](https://homicipher.fun) • [Anime Reborn Codes](https://animereborncodes.com) • [The Rise of the Golden Idol](https://theriseofthegoldenidol.com/) • [Five Hearts Under One Roof](https://fiveheartsunderoneroof.com) • [Googlies](https://googlies.pro/) • [Discordservers Wiki](https://discordservers.wiki) • [Laundry Store Simulator](https://laundrystoresimulator.com) • [Chill Guy Clicker](https://chillguyclicker.fun) • [Aurelia Game](https://aureliagame.com) • [Kiwi Mod](https://kiwimod.com) • [Doodle Empires](https://doodleempires.org) • [Codes Rob](https://codesrob.com) • [Em Manga](https://emmanga.com) • [Chocolate Factory Sim](https://chocolatefactorysim.com) • [Desktop Mate](https://desktop-mate.com) • [Yellow Mod](https://yellowmod.com) • [PP Novel](https://ppnovel.org) • [Phi4 Chat](https://phi4.chat) • [Streamer Life Simulator 2](https://streamerlifesimulator2.org) • [Ninja Gaiden 2 Black](https://ninjagaiden2black.xyz) • [Dynasty Warriors Origins](https://dynastywarriorsorigins.org) • [Block Breaker](https://block-breaker.online) • [Game Repacks](https://gamerepacks.com) • [Em Manga](https://emmanga.com) • [Amenti Game](https://amentigame.com) | |
| """) |