Update app.py

app.py

@@ -7,44 +7,44 @@ import zipfile
 from fpdf import FPDF
 from streamlit_cropper import st_cropper
 
-# --- CONFIG ---
+# --- PAGE CONFIG ---
 st.set_page_config(page_title="Smart Doc Scanner", page_icon="🖼️", layout="centered")
 
-# --- BACKGROUND STYLE ---
+# --- BACKGROUND with OVERLAY ---
 def set_background():
-    page_bg_img = f"""
+    page_bg_img = """
     <style>
-    [data-testid="stAppViewContainer"] {{
-        background-image: url("https://images.unsplash.com/photo-1524995997946-a1c2e315a42f");
+    [data-testid="stAppViewContainer"] {
+        background-image: url('https://images.unsplash.com/photo-1524995997946-a1c2e315a42f');
         background-size: cover;
-    }}
-    [data-testid="stHeader"] {{
-        background: rgba(0, 0, 0, 0);
-    }}
+        background-position: center;
+    }
+    [data-testid="stAppViewContainer"]::before {
+        content: "";
+        position: absolute;
+        top: 0; left: 0;
+        width: 100%; height: 100%;
+        background: rgba(255, 255, 255, 0.7); /* White transparent overlay */
+        z-index: 0;
+    }
+    .stApp {
+        position: relative;
+        z-index: 1;
+    }
     </style>
     """
     st.markdown(page_bg_img, unsafe_allow_html=True)
 
 set_background()
 
-# --- LIGHT/DARK MODE TOGGLE ---
-mode = st.selectbox("Select Theme:", ["Light Mode", "Dark Mode"])
-if mode == "Dark Mode":
-    dark_style = """
-    <style>
-    body { background-color: #222; color: white; }
-    </style>
-    """
-    st.markdown(dark_style, unsafe_allow_html=True)
-
-# --- APP TITLE ---
+# --- TITLE ---
 st.markdown("<h1 style='text-align: center; color: #2E86C1;'>📄 Smart ID & Document Scanner</h1>", unsafe_allow_html=True)
 st.write("---")
 
 # --- MODE SELECTION ---
 app_mode = st.radio("Select Mode:", ["Normal Mode (No Editing)", "Advanced Mode (Edit Each Image)"])
 
-# --- UPLOAD IMAGES ---
+# --- FILE UPLOAD ---
 uploaded_files = st.file_uploader("Upload your Images", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
 
 if uploaded_files:
@@ -67,19 +67,26 @@ if uploaded_files:
             st.subheader(f"Editing: {uploaded_file.name}")
 
             # Cropper
-            cropped_img = st_cropper(
+            crop_output = st_cropper(
                 st.session_state[f"edited_{idx}"],
                 realtime_update=True,
                 box_color='#FF9900',
                 aspect_ratio=None
             )
 
+            # If output is numpy array, convert to PIL Image
+            if isinstance(crop_output, np.ndarray):
+                crop_output = Image.fromarray(crop_output)
+
+            # Save cropped image
+            st.session_state[f"edited_{idx}"] = crop_output
+
             # Actions
             col1, col2, col3, col4 = st.columns(4)
 
             with col1:
                 if st.button(f"Enhance {idx}", key=f"enhance_{idx}"):
-                    enhancer = ImageEnhance.Contrast(cropped_img)
+                    enhancer = ImageEnhance.Contrast(st.session_state[f"edited_{idx}"])
                     enhanced_img = enhancer.enhance(1.5)
                     sharpener = ImageEnhance.Sharpness(enhanced_img)
                     st.session_state[f"edited_{idx}"] = sharpener.enhance(2.0)
@@ -87,7 +94,7 @@ if uploaded_files:
             with col2:
                 angle = st.slider(f"Rotate {uploaded_file.name}", -180, 180, 0, key=f"rotate_{idx}")
                 if angle != 0:
-                    st.session_state[f"edited_{idx}"] = cropped_img.rotate(angle, expand=True)
+                    st.session_state[f"edited_{idx}"] = st.session_state[f"edited_{idx}"].rotate(angle, expand=True)
 
             with col3:
                 if st.button(f"Reset {idx}", key=f"reset_{idx}"):
@@ -97,7 +104,7 @@ if uploaded_files:
             with col4:
                 st.write("")
 
-            # Show Image
+            # Show Edited Image
             st.image(st.session_state[f"edited_{idx}"], caption=f"Edited {uploaded_file.name}", use_container_width=True)
             processed_images.append(st.session_state[f"edited_{idx}"])