Update app.py

app.py

@@ -20,13 +20,13 @@ def resize_image(img, max_width=800):
 
 # --- HEADER ---
 st.title("📄 Smart ID & Document Scanner")
-st.write("---")
+st.markdown("---")
 
 # --- Clear Session State on New Upload ---
 if "uploaded_files_hash" not in st.session_state:
     st.session_state.uploaded_files_hash = None
 
-uploaded_files = st.file_uploader("Upload your Images", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
+uploaded_files = st.file_uploader("📤 Upload your Images", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
 
 if uploaded_files:
     uploaded_files_names = [file.name for file in uploaded_files]
@@ -36,78 +36,80 @@ if uploaded_files:
         st.session_state.clear()
         st.session_state.uploaded_files_hash = current_hash
 
-    app_mode = st.radio("Select Mode:", ["Normal Mode (No Editing)", "Advanced Mode (Edit Each Image)"])
+    app_mode = st.radio("🛠️ Select Mode:", ["Normal Mode (No Editing)", "Advanced Mode (Edit Each Image)"])
 
-    st.write("---")
+    st.markdown("---")
     processed_images = []
 
     if app_mode == "Normal Mode (No Editing)":
         for uploaded_file in uploaded_files:
             img = Image.open(uploaded_file).convert("RGB")
-            img = resize_image(img)  # Always resize
+            img = resize_image(img)
             processed_images.append(img)
             st.image(img, caption=uploaded_file.name, use_container_width=True)
 
     else:  # Advanced Mode
         for idx, uploaded_file in enumerate(uploaded_files):
-            st.subheader(f"Editing: {uploaded_file.name}")
+            st.subheader(f"🖼️ Editing: {uploaded_file.name}")
 
             original_img = Image.open(uploaded_file).convert("RGB")
             original_img = resize_image(original_img)
 
-            # Initialize session states
             if f"edited_{idx}" not in st.session_state:
                 st.session_state[f"edited_{idx}"] = original_img
                 st.session_state[f"cropper_{idx}"] = original_img.copy()
-                st.session_state[f"preview_{idx}"] = None  # Will be set after crop
+                st.session_state[f"preview_{idx}"] = None
 
-            # --- Layout: 2 columns ---
             col_cropper, col_preview = st.columns([1, 1])
 
             with col_cropper:
-                st.write("### Select Crop Area (Original Image):")
+                st.markdown("### ✂️ Select Crop Area (Original Image):")
                 crop_output = st_cropper(
                     st.session_state[f"cropper_{idx}"],
                     realtime_update=False,
                     box_color='#FF9900',
-                    aspect_ratio=None
+                    aspect_ratio=None,
+                    return_type="image",
+                    fit_image=True,
+                    container_height=600
                 )
 
                 if isinstance(crop_output, np.ndarray):
                     crop_output = Image.fromarray(crop_output)
 
-                if st.button(f"Crop Now {idx}", key=f"crop_{idx}"):
+                if st.button("✂️ Crop Now", key=f"crop_{idx}"):
                     st.session_state[f"preview_{idx}"] = crop_output
                     st.success("✅ Cropped! Now Edit Preview.")
 
             with col_preview:
-                st.write("### Preview and Edit:")
+                st.markdown("### 🎯 Preview and Edit:")
                 if st.session_state[f"preview_{idx}"] is not None:
                     st.image(st.session_state[f"preview_{idx}"], caption=f"Preview: {uploaded_file.name}", use_container_width=True)
 
-                    if st.button(f"Enhance Now {idx}", key=f"enhance_{idx}"):
+                    if st.button("✨ Enhance Image", key=f"enhance_{idx}"):
                         enhancer = ImageEnhance.Contrast(st.session_state[f"preview_{idx}"])
                         sharpener = ImageEnhance.Sharpness(enhancer.enhance(1.5))
                         st.session_state[f"preview_{idx}"] = sharpener.enhance(2.0)
 
-                    angle = st.slider(f"Rotate Angle", -180, 180, 0, key=f"rotate_slider_{idx}")
+                    angle = st.slider("🌀 Rotate Angle", -180, 180, 0, key=f"rotate_slider_{idx}")
 
-                    if st.button(f"Rotate Now {idx}", key=f"rotate_{idx}"):
+                    if st.button("🔄 Rotate Image", key=f"rotate_{idx}"):
                         rotated_img = st.session_state[f"preview_{idx}"].rotate(angle, expand=True)
                         st.session_state[f"preview_{idx}"] = rotated_img
 
-                    if st.button(f"Reset Preview {idx}", key=f"reset_preview_{idx}"):
+                    if st.button("🗑️ Reset Preview", key=f"reset_preview_{idx}"):
                         st.session_state[f"preview_{idx}"] = crop_output.copy()
 
-                    if st.button(f"Apply Changes {idx}", key=f"apply_{idx}"):
+                    if st.button("✅ Apply Changes", key=f"apply_{idx}"):
                         st.session_state[f"edited_{idx}"] = st.session_state[f"preview_{idx}"].copy()
                         st.success("✅ Changes Applied!")
+
                 else:
-                    st.info("Please crop an area first to enable editing.")
+                    st.info("✂️ Please crop an area first to enable editing.")
 
             processed_images.append(st.session_state.get(f"edited_{idx}", original_img))
 
-            st.write("---")
+            st.markdown("---")
 
     # --- EXPORT OPTIONS ---
     st.header("📥 Download Options")
@@ -115,7 +117,7 @@ if uploaded_files:
     col1, col2 = st.columns(2)
 
     with col1:
-        if st.button("Download All as ZIP 📷"):
+        if st.button("📸 Download All as ZIP"):
             with st.spinner("Creating ZIP..."):
                 zip_buffer = BytesIO()
                 with zipfile.ZipFile(zip_buffer, "a", zipfile.ZIP_DEFLATED) as zip_file:
@@ -128,7 +130,7 @@ if uploaded_files:
                 st.download_button(label="📥 Download ZIP", data=zip_buffer, file_name="images.zip", mime="application/zip")
 
     with col2:
-        if st.button("Download All as PDF 📄"):
+        if st.button("📄 Download All as PDF"):
             with st.spinner("Generating PDF..."):
                 pdf = FPDF()
                 temp_image_paths = []