Update app.py

app.py

@@ -3,9 +3,8 @@ from PIL import Image, ImageEnhance, ImageFilter, ImageDraw, ImageFont
 import io
 import numpy as np
 import cv2
-from rembg import remove  # For background removal
 
-# ✅ Add custom CSS for black background and consistent layout
+# ✅ Add custom CSS for black background and non-draggable images
 st.markdown(
     """
     <style>
@@ -43,22 +42,10 @@ st.markdown(
         .stDownloadButton>button:hover {
             background-color: #007bb5;
         }
-        /* Flex container for side-by-side images */
-        .image-container {
-            display: flex;
-            justify-content: space-around;
-            align-items: center;
-        }
-        .image-box {
-            max-width: 45%;
-            margin: 10px;
-            text-align: center;
-        }
+        /* ✅ Make images non-draggable */
         img {
-            width: 100%;
-            height: auto;
-            border-radius: 10px;
-            box-shadow: 0 4px 8px rgba(0,0,0,0.2);
+            pointer-events: none;
+            -webkit-user-drag: none;
         }
     </style>
     """,
@@ -73,9 +60,9 @@ st.write("Enhance, edit, and apply AI-powered enhancements to your images!")
 uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
 
 if uploaded_file:
-    # Load original image
+    # Load image
     image = Image.open(uploaded_file)
-    original_image = image.copy()
+    st.image(image, caption="Uploaded Image", use_column_width=True)
 
     # Sidebar Tools
     st.sidebar.title("Editing Tools")
@@ -117,18 +104,20 @@ if uploaded_file:
             sketch = cv2.divide(gray, 255 - blur, scale=256)
             image = Image.fromarray(sketch)
 
-    # Background Removal
-    if st.sidebar.checkbox("Remove Background"):
-        image = Image.open(io.BytesIO(remove(uploaded_file.read())))
-
     # Watermarking
     if st.sidebar.checkbox("Add Watermark"):
         watermark_text = st.sidebar.text_input("Watermark Text", "My Watermark")
         draw = ImageDraw.Draw(image)
-        font = ImageFont.load_default()  # Use default font if arial.ttf doesn't work
+        
+        # Try loading arial font, fallback to default
+        try:
+            font = ImageFont.truetype("arial.ttf", 30)
+        except IOError:
+            font = ImageFont.load_default()
+        
         width, height = image.size
         text_width, text_height = draw.textsize(watermark_text, font)
-        x, y = width - text_width - 10, height - text_height - 10
+        x, y = width - text_width - 20, height - text_height - 20
         draw.text((x, y), watermark_text, font=font, fill=(255, 255, 255, 128))
 
     # Enhance Image
@@ -145,21 +134,17 @@ if uploaded_file:
     except Exception as e:
         st.write(f"Error during enhancement: {e}")
 
-    # Display Images Side by Side
-    st.markdown('<div class="image-container">', unsafe_allow_html=True)
-
-    col1, col2 = st.columns(2)
-
-    with col1:
-        st.image(original_image, caption="Original Image", use_column_width=True)
-
-    with col2:
-        st.image(image, caption="Enhanced Image", use_column_width=True)
-
-    st.markdown('</div>', unsafe_allow_html=True)
+    # Display Enhanced Image
+    st.image(image, caption="Enhanced Image", use_column_width=True)
 
     # Download Button
     buf = io.BytesIO()
     image.save(buf, format="PNG")
     byte_im = buf.getvalue()
-    st.download_button("Download Enhanced Image", data=byte_im, file_name="enhanced_image.png", mime="image/png")
+    
+    st.download_button(
+        "Download Enhanced Image", 
+        data=byte_im, 
+        file_name="enhanced_image.png", 
+        mime="image/png"
+    )