Update app.py

app.py

@@ -4,12 +4,13 @@ import numpy as np
 from PIL import Image, ImageOps
 import subprocess
 
-# --- 1. CNC BRAIN V7 (Solid Shape Master) ---
-def process_vector(image, invert_input, threshold_val, solidify_strength, smooth_factor, remove_noise):
+# --- 1. CNC BRAIN V5.5 (Solid + Detail Preserved) ---
+def process_vector(image, invert_input, detail_level, solidify_strength, smooth_factor, remove_noise):
     if image is None: return None, None
 
-    # --- STEP 1: HD UPSCALE (Quality ke liye) ---
+    # --- STEP 1: HD UPSCALE (V5 ka magic - Quality ke liye) ---
     w, h = image.size
+    # Photo ko 2 guna bada karo taaki curves smooth aayein
     image = image.resize((w * 2, h * 2), Image.LANCZOS)
 
     # --- INVERT LOGIC ---
@@ -20,42 +21,47 @@ def process_vector(image, invert_input, threshold_val, solidify_strength, smooth
     img_np = np.array(image.convert("RGB"))
     gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)
     
-    # --- STEP 2: MEDIAN BLUR (The Line-Art Killer) ---
-    # Ye texture/lines ko gayab karke shape ko 'Flat' banata hai.
-    # Value 5 ya 7 best hai 3D renders ke liye.
-    gray_flat = cv2.medianBlur(gray, 7)
-
-    # Step B: GLOBAL THRESHOLD (Wapas purana reliable logic)
-    # Adaptive hata diya kyunki wo sketch bana raha tha.
-    # Ab hum user ke hath mein power denge ki wo decide kare kya black hai kya white.
-    _, binary = cv2.threshold(gray_flat, threshold_val, 255, cv2.THRESH_BINARY_INV)
+    # --- STEP 2: SMART CONTRAST (Blur ki jagah ye use karenge) ---
+    # Ye barik details ko ubhaar dega bina dhundla kiye
+    clahe = cv2.createCLAHE(clipLimit=3.0, tileGridSize=(8,8))
+    gray = clahe.apply(gray)
+
+    # --- STEP 3: ADAPTIVE DETAIL THRESHOLD ---
+    # Ye slider decide karega ki kitni barik cheez pakadni hai
+    # Slider value (11-99) ko hum Block Size banayenge
+    block_size = int(detail_level)
+    if block_size % 2 == 0: block_size += 1 # Odd number zaroori hai
+    
+    # Adaptive Threshold: Ye local area dekhta hai (Barik details ke liye best)
+    binary = cv2.adaptiveThreshold(gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, 
+                                 cv2.THRESH_BINARY_INV, block_size, 5)
 
-    # Step C: SOLIDIFY (Filling the gaps)
+    # --- STEP 4: SOLIDIFY (Tera favorite feature) ---
+    # Ye lines ko jodne aur solid banane ke liye hai
     kernel = np.ones((3,3), np.uint8)
     
-    # Morph Close: Ye chote chote gaps ko forcefully band karta hai
-    # Taki wo 'Line Art' na rahe, 'Solid Block' ban jaye.
-    binary = cv2.morphologyEx(binary, cv2.MORPH_CLOSE, kernel, iterations=2)
-
-    # Extra Thickness Control
     if solidify_strength > 0:
+        # Mota karo (Dilate)
         binary = cv2.dilate(binary, kernel, iterations=int(solidify_strength))
+        # Halka sa Erode taaki shape wapas sharp ho jaye
+        binary = cv2.erode(binary, kernel, iterations=1)
     elif solidify_strength < 0:
+        # Agar bahut chipak raha hai to alag karo
         binary = cv2.erode(binary, kernel, iterations=abs(int(solidify_strength)))
 
-    # Step D: Noise Removal
+    # Step 5: Noise Removal
     if remove_noise:
         binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)
 
-    # Step E: Invert for Potrace
+    # Step 6: Invert for Potrace
     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"
+    # Step 7: Vectorize
+    output_svg = "ts_vector_solid_detail.svg"
     
     cmd = [
         "potrace", temp_bmp,
@@ -78,13 +84,13 @@ 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: #00ff88 !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 SOLID TRACER V7", elem_id="logo_text")
-        gr.Markdown("### Converts 3D Images into SOLID Cut Files")
+        gr.Markdown("# 🏆 TS VECTOR V5.5", elem_id="logo_text")
+        gr.Markdown("### Solid Shapes + Fine Detail Control")
         
         with gr.Row():
             with gr.Column():
@@ -93,13 +99,13 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as app:
                 gr.Markdown("### 🎨 Image Type")
                 inv_chk = gr.Checkbox(label="Is Design WHITE on BLACK?", value=False)
                 
-                gr.Markdown("### 🎛️ Solid Settings")
+                gr.Markdown("### 🎛️ Settings (Adjust for Details)")
                 
-                # Global Threshold wapas aa gaya hai - Sabse reliable
-                thresh_sld = gr.Slider(0, 255, value=128, step=1, label="1. Solid Threshold (Adjust until design is full Black)")
+                # Ye hai naya slider - Isse barik detail aayegi
+                detail_sld = gr.Slider(11, 151, value=45, step=2, label="1. Detail Sensitivity (Kam = Barik Detail, Zyada = Mota Design)")
                 
-                # Solidify
-                solid_sld = gr.Slider(-2, 5, value=1, step=1, label="2. Solidify (Mota/Patla)")
+                # Tera favorite Solidify slider wapas aa gaya
+                solid_sld = gr.Slider(-2, 5, value=1, step=1, label="2. Solidify (Gap Bharne ke liye)")
                 
                 noise_chk = gr.Checkbox(label="3. Clean Noise", value=True)
                 smooth_sld = gr.Slider(0, 1.3, value=1.0, label="4. Smoothing")
@@ -110,6 +116,6 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as app:
                 preview_img = gr.Image(label="Computer Vision Preview (Black = Solid Material)", 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, detail_sld, solid_sld, smooth_sld, noise_chk], outputs=[out_file, preview_img])
 
 app.launch()