File size: 2,233 Bytes
d94bdc5 |
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 |
# Based on https://huggingface.co/spaces/keras-io/Enhance_Low_Light_Image
import streamlit as st
import os
from datetime import datetime
from PIL import Image, ImageOps
from io import BytesIO
from src.st_style import apply_prod_style
# apply_prod_style(st) # NOTE: Uncomment this for production!
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,
)
st.title("Enhance Low Light Photo with AI")
st.image(Image.open("assets/demo.jpg"))
st.write("Image Source: https://www.deviantart.com/dimelotu/art/Emily-s-Room-Lights-Off-561386750")
st.write(
"""
You have a low light photo, and you want to make it clearer and brighter?
Let's enhance them using AI. **Upload the photo then click "Enhance".**
"""
)
uploaded_file = st.file_uploader(
label="Upload your photo here",
accept_multiple_files=False,
type=["png", "jpg", "jpeg"],
)
if uploaded_file is not None:
with st.expander("Original photo", expanded=True):
if uploaded_file is not None:
st.image(uploaded_file)
else:
st.warning("You haven't uploaded any photo yet")
if st.button("Enhance") and uploaded_file is not None:
img_input = Image.open(uploaded_file).convert("RGB")
with st.spinner("AI is doing the magic"):
img_output = ImageOps.autocontrast(img_input, 3)
with st.expander("Success!", expanded=True):
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",
) |