Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 # Register HEIC opener
 pillow_heif.register_heif_opener()
 
-def process_image(image, strength):
+def process_image(image, user_strength):
     if image is None:
         return None, None
     
@@ -15,70 +15,98 @@ def process_image(image, strength):
     original_size = img.size
 
     # ============================================================
-    # 🧮 THE "STRENGTH" MATH
-    # Mapping the 1-10 slider to your specific code settings
+    # 🧮 LOGIC MAPPING
+    # User Input: 1 to 10
+    # Old Logic:  5 to 10 (Since <5 wasn't compressed enough)
     # ============================================================
     
-    # STRENGTH 1 (Light)  -> Scale 1.0,  Wax 0, Quality 50
-    # STRENGTH 8 (Sweet)  -> Scale 0.15, Wax 3, Quality 8  <-- Your Settings
-    # STRENGTH 10 (Nuke)  -> Scale 0.05, Wax 9, Quality 1
-    
-    # 1. SCALE: As strength goes up, scale goes DOWN (0.15 at level 8)
-    scale_factor = max(0.05, 1.0 - (strength * 0.1))
-    if strength >= 8: # Fine tune for the "sweet spot"
-        scale_factor = 0.15 - ((strength - 8) * 0.05)
+    # Map 1-10 input to the 5-10 strength range
+    # Input 1 -> 5.0
+    # Input 10 -> 10.0
+    strength = 5 + ((user_strength - 1) * (5/9))
 
-    # 2. WAX: As strength goes up, wax goes UP
-    wax_size = int(strength / 2) # Level 8 = 4, Level 10 = 5
-    if wax_size % 2 == 0: wax_size += 1 # Wax size must be odd number (1, 3, 5...)
+    # 1. SCALE CALCULATION
+    # At old strength 5, scale was ~0.5. At 10, scale was 0.05.
+    scale_factor = max(0.05, 1.0 - (strength * 0.1))
+    
+    # 2. WAX CALCULATION
+    wax_size = int(strength / 2) 
+    if wax_size % 2 == 0: wax_size += 1 
 
-    # 3. QUALITY: As strength goes up, quality goes DOWN
+    # 3. QUALITY CALCULATION
     quality_val = int(max(1, 60 - (strength * 6)))
 
     # ============================================================
-    # ⚙️ THE ENGINE (Your exact code)
+    # ⚙️ ENGINE
     # ============================================================
     
-    # A. Wax Effect
+    # Wax
     if wax_size > 1:
         img = img.filter(ImageFilter.MedianFilter(size=wax_size))
     
-    # B. Resolution Crush (Downscale)
+    # Crush
     new_w = int(original_size[0] * scale_factor)
     new_h = int(original_size[1] * scale_factor)
-    new_w, new_h = max(1, new_w), max(1, new_h) # Safety check
+    new_w, new_h = max(1, new_w), max(1, new_h)
     
     img = img.resize((new_w, new_h), resample=Image.BILINEAR)
 
-    # C. Save as HEIC (Compression Artifacts)
-    unique_name = f"waxy_{uuid.uuid4().hex[:8]}.heic"
+    # Save
+    unique_name = f"compressed_{uuid.uuid4().hex[:8]}.heic"
     img.save(unique_name, format="HEIF", quality=quality_val)
     
-    # D. Reload and Upscale (For Preview)
+    # Preview
     preview_img = Image.open(unique_name)
     preview_img = preview_img.resize(original_size, resample=Image.NEAREST)
     
     return preview_img, unique_name
 
 # ==========================================
-# 📱 THE SIMPLE INTERFACE
+# 🎨 STYLE & INTERFACE
 # ==========================================
 
+# Force white background, bigger text, sans-serif
 css = """
-#col-container {max-width: 700px; margin-left: auto; margin-right: auto;}
+body, .gradio-container {
+    background-color: white !important; 
+    color: black !important; 
+    font-family: 'Helvetica', 'Arial', sans-serif !important;
+}
+h1 {
+    color: black !important; 
+    font-size: 2.5em !important;
+    font-weight: bold !important;
+}
+span, p, label {
+    font-size: 1.2em !important;
+    color: #333 !important;
+}
+/* Hide the footer */
+footer {display: none !important;}
 """
 
-with gr.Blocks(css=css, theme=gr.themes.Monochrome()) as demo:
+# Clean theme with sans-serif font
+theme = gr.themes.Default(
+    font=[gr.themes.GoogleFont("Helvetica"), "Arial", "sans-serif"],
+    text_size=gr.themes.sizes.text_lg
+)
+
+with gr.Blocks(css=css, theme=theme) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("# 🫠 Waxy Crusher")
+        gr.Markdown("# Image Compressor")
         
         with gr.Row():
             input_img = gr.Image(type="pil", label="Upload", sources=["upload", "clipboard"])
         
-        # The One Slider to Rule Them All
-        strength_slider = gr.Slider(1, 10, value=8, step=1, label="Destruction Level (8 = Sweet Spot)")
+        strength_slider = gr.Slider(
+            minimum=1, 
+            maximum=10, 
+            value=5, 
+            step=0.5, 
+            label="Compression Strength"
+        )
         
-        btn = gr.Button("🔥 CRUSH IT", variant="primary", size="lg")
+        btn = gr.Button("COMPRESS", variant="primary", size="lg")
         
         result_img = gr.Image(label="Preview", interactive=False)
         download_file = gr.File(label="Download HEIC")