Update app.py

app.py

@@ -4,44 +4,59 @@ import numpy as np
 from PIL import Image, ImageOps
 import subprocess
 
-# --- 1. CNC BRAIN V4 (Solid Shape Logic) ---
-def process_vector(image, threshold_val, solidify_strength, smooth_factor, remove_noise):
+# --- 1. CNC BRAIN V6 (Color Aware Scanner) ---
+def process_vector(image, color_mode, scan_sensitivity, solidify_strength, smooth_factor, remove_noise):
     if image is None: return None, None
 
-    # Step A: Convert to Grayscale & Blur (Shine hatane ke liye)
+    # A. Grayscale
     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!
     
-    # Strong Blur to mix Gold Highlight and Shadow
-    gray_blurred = cv2.GaussianBlur(gray, (5, 5), 0)
+    if color_mode == "Light Design (White on Black)":
+        gray = cv2.bitwise_not(gray)
 
-    # Step B: Thresholding (Solid Black logic)
-    # Binary Inverse: Jali banegi WHITE, Background banega BLACK
-    _, binary = cv2.threshold(gray_blurred, threshold_val, 255, cv2.THRESH_BINARY_INV)
+    # B. Contrast Enhance (CLAHE) - Detail nikalne ke liye
+    clahe = cv2.createCLAHE(clipLimit=3.0, tileGridSize=(8,8))
+    enhanced_gray = clahe.apply(gray)
 
-    # Step C: SOLIDIFY (Magic Fix for Double Lines)
-    kernel = np.ones((3,3), np.uint8)
+    # 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)
+
+    # D. POST-PROCESSING
+    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:
-        # Dilate: White hissa (Jali) mota hoga aur beech ke holes bharenge
         binary = cv2.dilate(binary, kernel, iterations=int(solidify_strength))
-        # Wapas thoda shape mein laane ke liye halka erode
         binary = cv2.erode(binary, kernel, iterations=1)
 
-    # Step D: Noise Removal (Watermark cleaning)
+    # F. Noise Removal
     if remove_noise:
-        # Morph Open: Chote white dots (noise) gayab karega
         binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)
 
-    # Step E: Invert for Potrace (Potrace needs Black Shape on White)
+    # G. Potrace Preparation
+    # Potrace ko hamesha "Black Design on White BG" chahiye hota hai.
     final_binary = cv2.bitwise_not(binary)
 
     # Save Temp
     temp_bmp = "temp_trace.bmp"
     cv2.imwrite(temp_bmp, final_binary)
 
-    # Step F: Vectorize
-    output_svg = "ts_vector_solid.svg"
+    # H. Vectorize
+    output_svg = "ts_vector_pro.svg"
     
     cmd = [
         "potrace", temp_bmp,
@@ -59,36 +74,43 @@ def process_vector(image, threshold_val, solidify_strength, smooth_factor, remov
 
     return output_svg, Image.fromarray(final_binary)
 
-# --- 2. Interface (FIXED THEME ERROR) ---
+# --- 2. Interface ---
 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: #FFD700 !important; color: black !important; font-weight: bold !important; }
+.primary-btn { background: linear-gradient(45deg, #00ff88, #00b8ff) !important; color: black !important; font-weight: bold !important; }
 """
 
-# Maine yahan se mode='dark' hata diya hai, ab Soft theme use hoga jo safe hai
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as app:
     with gr.Column(elem_id="main_card"):
-        gr.Markdown("# 🏆 TS SOLID TRACER", elem_id="logo_text")
+        gr.Markdown("# 🔳 TS VECTOR V6", elem_id="logo_text")
+        gr.Markdown("### Handles Both: Golden Jali & White Vectors")
         
         with gr.Row():
             with gr.Column():
                 inp_img = gr.Image(type="pil", label="Upload Photo", height=300)
                 
-                gr.Markdown("### 🎛️ Settings")
+                # --- 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?"
+                )
                 
-                thresh_sld = gr.Slider(0, 255, value=140, step=1, label="1. Darkness Threshold")
-                solid_sld = gr.Slider(0, 5, value=2, step=1, label="2. Solidify (Double Line Fixer)")
-                noise_chk = gr.Checkbox(label="3. Clean Noise", value=True)
-                smooth_sld = gr.Slider(0, 1.3, value=1.0, label="4. Smoothing")
+                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)")
+                noise_chk = gr.Checkbox(label="Clean Noise", value=True)
+                smooth_sld = gr.Slider(0, 1.3, value=1.0, label="Smoothing")
                 
-                btn = gr.Button("🔥 CREATE SOLID VECTOR", variant="primary", elem_classes=["primary-btn"])
+                btn = gr.Button("🚀 CONVERT NOW", 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, thresh_sld, solid_sld, smooth_sld, noise_chk], outputs=[out_file, preview_img])
+    btn.click(process_vector, inputs=[inp_img, color_mode, scan_sld, solid_sld, smooth_sld, noise_chk], outputs=[out_file, preview_img])
 
 app.launch()