Update app.py

app.py

@@ -1,19 +1,11 @@
-# App code based on: https://github.com/petergro-hub/ComicInpainting
-# Model based on: https://github.com/saic-mdal/lama
-
 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)")
@@ -32,8 +24,6 @@ def image_download_button(pil_image, filename: str, fmt: str, label="Download"):
         mime=mime,
     )
 
-
-
 if "button_id" not in st.session_state:
     st.session_state["button_id"] = ""
 if "color_to_label" not in st.session_state:
@@ -41,6 +31,7 @@ if "color_to_label" not in st.session_state:
 
 if 'reuse_image' not in st.session_state:
     st.session_state.reuse_image = None
+
 def set_image(img):
     st.session_state.reuse_image = img
 
@@ -113,17 +104,14 @@ if uploaded_file is not None:
         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
+                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)
-            # 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(
@@ -136,87 +124,41 @@ if uploaded_file is not None:
             st.info("**TIP**: If the result is not perfect, you can download it then "
                     "upload then remove the artifacts.")
 
-
+# CSS styles for dark mode and responsiveness
+st.markdown("""
 <style>
 body {
     font-family: 'Arial', sans-serif;
-    margin: 0;
     background-color: #1a1a2e;
     color: #ffffff;
-    overflow-x: hidden;
-}
-
-.stButton button {
-    background-color: #ffffff;
-    color: #1a1a2e;
-}
-
-.stButton button:hover {
-    background-color: #9370db;
-    color: #ffffff;
-}
-
-.stSlider {
-    color: #ffffff;
-}
-
-.stSlider .stSlider-label {
-    color: #ffffff;
-}
-
-.stSlider .stSlider-track {
-    background-color: #4a4a5e;
-}
-
-.stSlider .stSlider-thumb {
-    background-color: #ffffff;
 }
-
-.stMarkdown {
-    color: #ffffff;
-}
-
-.stSpinner {
-    color: #ffffff;
-}
-
-.stFileUploader {
+.stApp {
+    max-width: 800px;
+    margin: 0 auto;
+    padding: 20px;
     background-color: #2a2a3e;
-    color: #ffffff;
-    border-color: #4a4a5e;
-}
-
-.stFileUploader:hover {
-    background-color: #3a3a4e;
+    border-radius: 10px;
 }
-
-.stImage {
+canvas {
+    display: block;
+    margin: 0 auto;
     max-width: 100%;
     height: auto;
 }
-
-.stTextInput input {
-    background-color: #2a2a3e;
-    color: #ffffff;
-    border-color: #4a4a5e;
-}
-
-.stTextInput input:focus {
-    border-color: #6a6a7e;
-}
-
-.stDownloadButton {
-    background-color: #ffffff;
-    color: #1a1a2e;
-}
-
-.stDownloadButton:hover {
+.stButton button {
     background-color: #9370db;
+    border: none;
+    padding: 10px;
+    border-radius: 5px;
     color: #ffffff;
+    font-weight: bold;
+    cursor: pointer;
 }
-
-.stInfo {
-    background-color: #2a2a3e;
+.stButton button:hover {
+    background-color: #7b5fbf;
+}
+.stSlider {
     color: #ffffff;
 }
 </style>
+""", unsafe_allow_html=True)