Update app.py

app.py

@@ -4,40 +4,48 @@ import numpy as np
 from PIL import Image, ImageOps
 import subprocess
 
-# --- 1. CNC BRAIN V5 (HD Upscale + Smooth Logic) ---
-def process_vector(image, invert_input, threshold_val, solidify_strength, smooth_factor, remove_noise):
+# --- 1. CNC BRAIN V6 (Micro-Detail Logic) ---
+def process_vector(image, invert_input, detail_sens, thickness, remove_noise):
     if image is None: return None, None
 
-    # --- STEP 0: HD UPSCALE (The Magic Fix) ---
-    # Agar photo choti hai ya sharp hai, to usse 2x bada kar do.
-    # Isse curves "Blocky" nahi honge, bilkul smooth aayenge.
+    # --- STEP 0: HD UPSCALE ---
+    # Detail nikalne ke liye photo bada hona zaroori hai
     w, h = image.size
-    # High Quality Resizing (Lanczos Filter)
     image = image.resize((w * 2, h * 2), Image.LANCZOS)
 
     # --- INVERT LOGIC ---
     if invert_input:
         image = ImageOps.invert(image.convert("RGB"))
 
-    # Step A: Convert to Grayscale & Blur
+    # Step A: Grayscale & Contrast Boost (No Blur!)
     img_np = np.array(image.convert("RGB"))
     gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)
     
-    # Blur thoda badhaya hai taaki pixelate na ho
-    gray_blurred = cv2.GaussianBlur(gray, (9, 9), 0)
-
-    # Step B: Thresholding
-    _, binary = cv2.threshold(gray_blurred, threshold_val, 255, cv2.THRESH_BINARY_INV)
+    # CLAHE (Smart Contrast): Ye Gold color ko background se alag karega
+    clahe = cv2.createCLAHE(clipLimit=3.0, tileGridSize=(8,8))
+    gray = clahe.apply(gray)
+
+    # Step B: Adaptive Thresholding (Detail Hunter)
+    # Block size jitna chota (odd number), utni barik detail pakdega
+    block_size = int(detail_sens)
+    if block_size % 2 == 0: block_size += 1 # Must be odd
+    
+    binary = cv2.adaptiveThreshold(gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, 
+                                 cv2.THRESH_BINARY_INV, block_size, 5)
 
-    # Step C: SOLIDIFY (Double Line Fixer)
+    # Step C: THICKNESS CONTROL (Negative allows opening gaps)
     kernel = np.ones((3,3), np.uint8)
     
-    if solidify_strength > 0:
-        binary = cv2.dilate(binary, kernel, iterations=int(solidify_strength))
-        binary = cv2.erode(binary, kernel, iterations=1)
-
-    # Step D: Noise Removal
+    if thickness > 0:
+        # Positive: Mota karo (Dilate)
+        binary = cv2.dilate(binary, kernel, iterations=int(thickness))
+    elif thickness < 0:
+        # Negative: Patla karo (Erode) - Ye 'chipki hui' lines ko kholega
+        binary = cv2.erode(binary, kernel, iterations=abs(int(thickness)))
+
+    # Step D: Noise Removal (Optional - for very noisy photos)
     if remove_noise:
+        # Bilkul halka sa clean, zyada nahi
         binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)
 
     # Step E: Invert for Potrace
@@ -48,14 +56,13 @@ def process_vector(image, invert_input, threshold_val, solidify_strength, smooth
     cv2.imwrite(temp_bmp, final_binary)
 
     # Step F: Vectorize
-    output_svg = "ts_vector_hd.svg"
+    output_svg = "ts_vector_detail.svg"
     
     cmd = [
         "potrace", temp_bmp,
         "-s", "-o", output_svg,
-        "-t", "10", 
-        "-a", str(smooth_factor), # Curve smoothness controlled here
-        "--alphamax", "1.34",     # Maximize curves
+        "-t", "2", # Turdsize kam kiya taaki chote dots na udein
+        "-a", "1.0",
         "--opaque"
     ]
     
@@ -77,28 +84,32 @@ body, .gradio-container { font-family: 'Inter', sans-serif !important; backgroun
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as app:
     with gr.Column(elem_id="main_card"):
-        gr.Markdown("# 🏆 TS VECTOR HD (V5)", elem_id="logo_text")
-        gr.Markdown("### Auto-Upscale for Smooth Curves")
+        gr.Markdown("# 🔬 TS VECTOR PRO (V6)", elem_id="logo_text")
+        gr.Markdown("### Optimized for High-Detail & Mandala Designs")
         
         with gr.Row():
             with gr.Column():
                 inp_img = gr.Image(type="pil", label="Upload Photo", height=300)
                 
                 gr.Markdown("### 🎨 Image Type")
-                inv_chk = gr.Checkbox(label="Is Design WHITE on BLACK Background?", value=False)
+                inv_chk = gr.Checkbox(label="Is Design WHITE on BLACK?", value=False)
+                
+                gr.Markdown("### 🎛️ Detail Settings")
+                
+                # Naya Slider: Block Size
+                sens_sld = gr.Slider(11, 99, value=45, step=2, label="1. Detail Sensitivity (Kam = Barik, Zyada = Mota Area)")
+                
+                # Naya Slider: Negative value allowed
+                thick_sld = gr.Slider(-3, 3, value=0, step=1, label="2. Line Thickness (Minus me le jao gap kholne 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=0, step=1, label="Solidify (Keep 0 for Graphics)")
-                noise_chk = gr.Checkbox(label="Clean Noise", value=True)
-                smooth_sld = gr.Slider(0, 1.5, value=1.2, label="Smoothing (High for Round shapes)")
+                noise_chk = gr.Checkbox(label="3. Clean Noise (Off rakho agar detail udd rahi hai)", value=False)
                 
-                btn = gr.Button("🔥 CREATE HD VECTOR", variant="primary", elem_classes=["primary-btn"])
+                btn = gr.Button("🔥 CAPTURE DETAILS", 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, inv_chk, thresh_sld, solid_sld, smooth_sld, noise_chk], outputs=[out_file, preview_img])
+    btn.click(process_vector, inputs=[inp_img, inv_chk, sens_sld, thick_sld, noise_chk], outputs=[out_file, preview_img])
 
 app.launch()