Update app.py

app.py

@@ -4,59 +4,45 @@ import numpy as np
 from PIL import Image, ImageOps
 import subprocess
 
-# --- 1. CNC BRAIN V6 (Color Aware Scanner) ---
-def process_vector(image, color_mode, scan_sensitivity, solidify_strength, smooth_factor, remove_noise):
+# --- 1. CNC BRAIN V4.5 (Solid Shape + Invert Logic) ---
+def process_vector(image, invert_input, threshold_val, solidify_strength, smooth_factor, remove_noise):
     if image is None: return None, None
 
-    # A. Grayscale
+    # --- NEW FIX: INVERT LOGIC ---
+    # Agar design White hai aur Background Black, toh hum pehle hi usse palat denge
+    if invert_input:
+        image = ImageOps.invert(image.convert("RGB"))
+
+    # Step A: Convert to Grayscale & Blur
     img_np = np.array(image.convert("RGB"))
     gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)
-
-    # --- INTELLIGENT INVERSION LOGIC ---
-    # Agar user ne kaha "Light Design" (White on Black), toh hum pehle hi photo ko ulta (Invert) kar denge.
-    # Isse White Design -> Black ban jayega aur Black BG -> White ban jayega.
-    # Phir humara purana V5 engine uspe perfect kaam karega!
     
-    if color_mode == "Light Design (White on Black)":
-        gray = cv2.bitwise_not(gray)
+    # Blur to mix highlights
+    gray_blurred = cv2.GaussianBlur(gray, (5, 5), 0)
 
-    # B. Contrast Enhance (CLAHE) - Detail nikalne ke liye
-    clahe = cv2.createCLAHE(clipLimit=3.0, tileGridSize=(8,8))
-    enhanced_gray = clahe.apply(gray)
-
-    # C. Adaptive Scanning (V5 Logic - Jo Golden Jali par perfect tha)
-    block_size = int(scan_sensitivity)
-    if block_size % 2 == 0: block_size += 1
-    
-    # Adaptive Threshold
-    binary = cv2.adaptiveThreshold(enhanced_gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, 
-                                 cv2.THRESH_BINARY_INV, block_size, 5)
+    # Step B: Thresholding (Solid Black logic)
+    _, binary = cv2.threshold(gray_blurred, threshold_val, 255, cv2.THRESH_BINARY_INV)
 
-    # D. POST-PROCESSING
+    # Step C: SOLIDIFY (Double Line Fixer)
     kernel = np.ones((3,3), np.uint8)
-
-    # Clean noisy dots
-    binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)
-
-    # E. SOLIDIFY (Double Line Fixer)
+    
     if solidify_strength > 0:
         binary = cv2.dilate(binary, kernel, iterations=int(solidify_strength))
         binary = cv2.erode(binary, kernel, iterations=1)
 
-    # F. Noise Removal
+    # Step D: Noise Removal
     if remove_noise:
         binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)
 
-    # G. Potrace Preparation
-    # Potrace ko hamesha "Black Design on White BG" chahiye hota hai.
+    # Step E: Invert for Potrace
     final_binary = cv2.bitwise_not(binary)
 
     # Save Temp
     temp_bmp = "temp_trace.bmp"
     cv2.imwrite(temp_bmp, final_binary)
 
-    # H. Vectorize
-    output_svg = "ts_vector_pro.svg"
+    # Step F: Vectorize
+    output_svg = "ts_vector_solid.svg"
     
     cmd = [
         "potrace", temp_bmp,
@@ -79,38 +65,33 @@ custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;600;900&display=swap');
 body, .gradio-container { font-family: 'Inter', sans-serif !important; background: #000000 !important; color: white !important; }
 #main_card { border: 1px solid #333; border-radius: 15px; padding: 20px; background: #111; }
-.primary-btn { background: linear-gradient(45deg, #00ff88, #00b8ff) !important; color: black !important; font-weight: bold !important; }
+.primary-btn { background: #FFD700 !important; color: black !important; font-weight: bold !important; }
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as app:
     with gr.Column(elem_id="main_card"):
-        gr.Markdown("# 🔳 TS VECTOR V6", elem_id="logo_text")
-        gr.Markdown("### Handles Both: Golden Jali & White Vectors")
+        gr.Markdown("# 🏆 TS SOLID TRACER V4.5", elem_id="logo_text")
         
         with gr.Row():
             with gr.Column():
                 inp_img = gr.Image(type="pil", label="Upload Photo", height=300)
                 
-                # --- NEW OPTION HERE ---
-                gr.Markdown("### 🎨 1. Image Color Type (Very Important)")
-                color_mode = gr.Radio(
-                    ["Dark Design (Golden/Sketch)", "Light Design (White on Black)"],
-                    value="Dark Design (Golden/Sketch)",
-                    label="Tumhari photo kaisi hai?"
-                )
+                gr.Markdown("### 🎨 Image Type (Ye naya hai)")
+                # Yahi wo magic button hai jo White Design ko fix karega
+                inv_chk = gr.Checkbox(label="Is Design WHITE on BLACK Background?", value=False)
                 
-                gr.Markdown("### 🎛️ 2. Fine Tuning")
-                scan_sld = gr.Slider(11, 61, value=31, step=2, label="Scanner Sensitivity (Detail)")
-                solid_sld = gr.Slider(0, 5, value=2, step=1, label="Solidify (Mota karne ke liye)")
+                gr.Markdown("### 🎛️ Settings")
+                thresh_sld = gr.Slider(0, 255, value=140, step=1, label="Darkness Threshold")
+                solid_sld = gr.Slider(0, 5, value=2, step=1, label="Solidify (Double Line Fixer)")
                 noise_chk = gr.Checkbox(label="Clean Noise", value=True)
                 smooth_sld = gr.Slider(0, 1.3, value=1.0, label="Smoothing")
                 
-                btn = gr.Button("🚀 CONVERT NOW", variant="primary", elem_classes=["primary-btn"])
+                btn = gr.Button("🔥 CREATE VECTOR", variant="primary", elem_classes=["primary-btn"])
 
             with gr.Column():
                 preview_img = gr.Image(label="Computer Vision Preview", interactive=False)
                 out_file = gr.File(label="Download SVG")
 
-    btn.click(process_vector, inputs=[inp_img, color_mode, scan_sld, solid_sld, smooth_sld, noise_chk], outputs=[out_file, preview_img])
+    btn.click(process_vector, inputs=[inp_img, inv_chk, thresh_sld, solid_sld, smooth_sld, noise_chk], outputs=[out_file, preview_img])
 
 app.launch()