Update app.py

app.py

@@ -1,142 +1,125 @@
 import gradio as gr
-from rembg import remove
 import cv2
 import numpy as np
 from PIL import Image
-import os
 import subprocess
 
-# --- 1. Processing Logic ---
-def convert_to_vector(image, detail_threshold, smoothness):
-    if image is None:
-        return None, None
+# --- 1. The Core Logic (CNC Brain) ---
+def process_vector(image, mode, thickness, noise_level, smooth_factor):
+    if image is None: return None, None
 
-    # Step A: AI Background Removal (Cleaning)
-    # Photo mein se background uda do taaki sirf Jali bache
-    print("Removing Background...")
-    try:
-        # Convert PIL to simple RGB if needed
-        image = image.convert("RGB")
-        # Rembg magic
-        output_image = remove(image) # Returns RGBA
-    except Exception as e:
-        print(f"AI Error: {e}")
-        output_image = image # Fallback
+    # A. Pre-processing (Convert to Grayscale)
+    # Color se hamein matlab nahi, hamein sirf Light/Dark difference chahiye
+    img_np = np.array(image.convert("RGB"))
+    gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)
 
-    # Step B: Prepare for Tracing (Black & White conversion)
-    # Alpha channel (transparency) ko mask banao
-    img_np = np.array(output_image)
+    # B. TRACING MODES
+    if mode == "Standard (High Contrast)":
+        # Simple Logic: Agar photo saaf hai (Black & White)
+        _, binary = cv2.threshold(gray, 128, 255, cv2.THRESH_BINARY_INV)
+        
+    elif mode == "Scanner Mode (Colored/Light Lights)":
+        # ADVANCED: Adaptive Thresholding
+        # Ye local areas dekhta hai. Colored lights ya shadows ko ignore karta hai.
+        # Block Size 11, C 2 (Standard scanner settings)
+        binary = cv2.adaptiveThreshold(gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, 
+                                     cv2.THRESH_BINARY_INV, 15, 4)
     
-    # Agar transparent hai (PNG), to Alpha use karo, nahi to Grayscale
-    if img_np.shape[2] == 4:
-        alpha = img_np[:, :, 3]
-        # Thresholding based on slider
-        _, binary = cv2.threshold(alpha, detail_threshold, 255, cv2.THRESH_BINARY)
-    else:
-        gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)
-        _, binary = cv2.threshold(gray, detail_threshold, 255, cv2.THRESH_BINARY_INV)
+    elif mode == "Silhouette (Dark Object)":
+        # Agar Jali kaali hai aur peeche light hai
+        _, binary = cv2.threshold(gray, 100, 255, cv2.THRESH_BINARY)
 
-    # Save temp BMP (Potrace needs BMP)
+    # C. CLEANING & THICKNESS (Thickness Slider Logic)
+    # Morphological Operations - Jali ko mota/patla karne ke liye
+    kernel = np.ones((3,3), np.uint8)
+    
+    # Noise Removal (Chote daane hatana)
+    if noise_level > 0:
+        # Opening: Removes small noise (white dots)
+        binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=int(noise_level))
+
+    # Thickness (Erosion/Dilation)
+    if thickness > 0:
+        # Dilate: Makes white lines thicker (Jali moti hogi)
+        binary = cv2.dilate(binary, kernel, iterations=int(thickness))
+    elif thickness < 0:
+        # Erode: Makes lines thinner
+        binary = cv2.erode(binary, kernel, iterations=abs(int(thickness)))
+
+    # D. Save Temp BMP for Potrace
     temp_bmp = "temp_trace.bmp"
     cv2.imwrite(temp_bmp, binary)
 
-    # Step C: Vectorize using Potrace
-    # Command line magic: BMP -> SVG
-    # -t = turdsize (speckle noise remove karne ke liye)
-    # -a = alphamax (curves kitne smooth honge)
+    # E. Vectorize using Potrace
     output_svg = "ts_vector_design.svg"
     
-    # Smoothness slider logic (0.5 to 1.3)
-    corner_policy = "1" # Turn off optimization
-    
+    # Potrace Commands
     cmd = [
         "potrace", temp_bmp,
         "-s", # Output SVG
         "-o", output_svg,
-        "-t", "10", # Chote daag dhabbe ignore karo
-        "-a", str(smoothness), # Smoothness control
-        "--opaque" # Solid fill
+        "-t", "2", # Remove Speckles
+        "-a", str(smooth_factor), # Corner Smoothness
+        "--opaque" # Solid Fill
     ]
     
-    subprocess.run(cmd, check=True)
+    try:
+        subprocess.run(cmd, check=True)
+    except Exception as e:
+        print(f"Error: {e}")
+        return None, None
 
-    # Return SVG path and Preview Image (Black & White version)
+    # Return SVG and the B&W Preview
     return output_svg, Image.fromarray(binary)
 
-# --- 2. Premium TS Interface ---
+# --- 2. Interface Design ---
 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: #0f172a !important; /* Rich Dark Blue */
-    color: #e2e8f0 !important;
-}
+body, .gradio-container { font-family: 'Inter', sans-serif !important; background: #1e1e2e !important; color: white !important; }
 
 #main_card {
-    background: rgba(30, 41, 59, 0.7);
-    backdrop-filter: blur(20px);
+    background: rgba(255, 255, 255, 0.05);
     border: 1px solid rgba(255, 255, 255, 0.1);
-    border-radius: 24px;
+    border-radius: 20px;
     padding: 30px;
-    box-shadow: 0 25px 50px -12px rgba(0, 0, 0, 0.5);
-    max-width: 1000px;
-    margin: 30px auto;
 }
-
-#logo_text h1 {
-    background: linear-gradient(to right, #fbbf24, #d97706); /* Golden Gradient for CNC */
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
-    font-size: 3.5rem !important;
-    font-weight: 900;
-    text-align: center;
-    margin-bottom: 5px;
-}
-
-#subtitle_text h3 {
-    text-align: center;
-    color: #94a3b8;
-    font-size: 1.1rem;
-    margin-bottom: 30px;
-}
-
 .primary-btn {
-    background: linear-gradient(135deg, #fbbf24, #b45309) !important;
-    color: #000 !important;
+    background: linear-gradient(90deg, #FF9966, #FF5E62) !important;
+    color: white !important;
     border: none !important;
     font-size: 1.2rem !important;
-    font-weight: 800 !important;
-    padding: 16px !important;
-    box-shadow: 0 10px 15px -3px rgba(245, 158, 11, 0.3);
+    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", elem_id="logo_text")
-        gr.Markdown("### IMAGE TO CNC VECTOR CONVERTER", elem_id="subtitle_text")
+        gr.Markdown("# 📐 TS VECTOR PRO", elem_id="logo_text")
+        gr.Markdown("### Advanced CNC Tracer (Color/Light Proof)")
         
         with gr.Row():
-            # Left: Inputs
-            with gr.Column(scale=1):
-                inp_img = gr.Image(type="pil", label="Upload Jali Design (Jpg/Png)")
+            with gr.Column():
+                inp_img = gr.Image(type="pil", label="Upload Jali Image", height=300)
+                
+                # --- NEW CONTROLS ---
+                gr.Markdown("### 🎛️ Trace Settings")
+                
+                mode_drp = gr.Dropdown(
+                    ["Scanner Mode (Colored/Light Lights)", "Standard (High Contrast)", "Silhouette (Dark Object)"],
+                    value="Scanner Mode (Colored/Light Lights)",
+                    label="1. Tracing Mode (Best for Colored Lights)"
+                )
                 
-                with gr.Group():
-                    gr.Markdown("### ⚙️ Trace Settings")
-                    thresh = gr.Slider(0, 255, value=128, label="Detail Threshold (Darkness)")
-                    smooth = gr.Slider(0.0, 1.33, value=1.0, label="Curve Smoothness (Low=Sharp, High=Round)")
+                thick_sld = gr.Slider(-2, 5, value=0, step=1, label="2. Thickness (Mota/Patla)")
+                noise_sld = gr.Slider(0, 3, value=1, step=1, label="3. Clean Noise (Daane Hataye)")
+                smooth_sld = gr.Slider(0, 1.4, value=1.0, label="4. Smoothing")
                 
-                btn = gr.Button("🔥 CONVERT TO SVG", variant="primary", elem_classes=["primary-btn"])
+                btn = gr.Button("🚀 CONVERT TO VECTOR", variant="primary", elem_classes=["primary-btn"])
 
-            # Right: Outputs
-            with gr.Column(scale=1):
-                # Preview of what AI saw
-                preview_img = gr.Image(label="AI Vision (B&W Preview)", interactive=False)
-                # Download File
-                out_file = gr.File(label="Download Vector File (.SVG)")
-                gr.Markdown("*Note: Download SVG and open in CorelDraw/Illustrator.*")
+            with gr.Column():
+                preview_img = gr.Image(label="AI Vision (B&W)", interactive=False)
+                out_file = gr.File(label="Download SVG")
 
-    btn.click(convert_to_vector, inputs=[inp_img, thresh, smooth], outputs=[out_file, preview_img])
+    btn.click(process_vector, inputs=[inp_img, mode_drp, thick_sld, noise_sld, smooth_sld], outputs=[out_file, preview_img])
 
 app.launch()