Update app.py

app.py

@@ -4,45 +4,36 @@ import numpy as np
 from PIL import Image, ImageOps, ImageEnhance
 import subprocess
 
-# --- CNC BRAIN: THE ULTIMATE (Zero Blur, Pure Detail) ---
-def process_vector(image, invert_input, contrast_boost, detail_threshold, solid_force, remove_noise):
+# --- CNC BRAIN V8: SMOOTH MASTER (Detail + Curves Balanced) ---
+def process_vector(image, invert_input, contrast_boost, detail_threshold, solid_force, smooth_factor):
     if image is None: return None, None
 
-    # 1. HD UPSCALE (Resolution is king for details)
+    # 1. HD UPSCALE (Resolution zaroori hai smooth curves ke liye)
     w, h = image.size
-    # 2x बड़ा कर रहे हैं ताकि बारीक पिक्सेल अलग हो सकें
     image = image.resize((w * 2, h * 2), Image.LANCZOS)
 
-    # 2. PRE-PROCESSING (Contrast Boost)
-    # OpenCV से पहले Pillow से कंट्रास्ट बढ़ाएंगे ताकि लाइनें उभर कर आएं
+    # 2. PRE-PROCESSING
     enhancer = ImageEnhance.Contrast(image)
-    image = enhancer.enhance(contrast_boost) # User controlled boost
+    image = enhancer.enhance(contrast_boost)
 
     # --- INVERT LOGIC ---
     if invert_input:
         image = ImageOps.invert(image.convert("RGB"))
 
-    # Grayscale conversion
     img_np = np.array(image.convert("RGB"))
     gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)
 
-    # --- THE SURGICAL THRESHOLD ---
-    # कोई ब्लर नहीं, कोई ऑटोमैटिक दिमाग नहीं। सीधा थ्रेशोल्ड।
-    # यह स्लाइडर तय करेगा कि किस शेड को काला मानना है।
-    # बारीक डिज़ाइन के लिए यह सबसे सटीक तरीका है।
+    # --- THRESHOLDING ---
     _, binary = cv2.threshold(gray, detail_threshold, 255, cv2.THRESH_BINARY_INV)
 
-    # --- SOLID FORCE (Optional) ---
-    # सिर्फ तब इस्तेमाल करें जब लाइनें टूट रही हों।
-    # बारीक डिज़ाइन में इसे 0 रखना ही बेहतर है।
+    # --- SOLID FORCE & CLEANING ---
     kernel = np.ones((3,3), np.uint8)
-    if solid_force > 0:
-        binary = cv2.dilate(binary, kernel, iterations=int(solid_force))
+    
+    # Noise cleaning (Thoda zaroori hai smooth vector ke liye)
+    binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)
 
-    # --- NOISE REMOVAL (Dangerous for fine detail) ---
-    # इसे बहुत सोच समझ कर ऑन करना, यह बारीक बिन्दुओं को उड़ा सकता है।
-    if remove_noise:
-        binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)
+    if solid_force > 0:
+        binary = cv2.dilate(binary, kernel, iterations=int(solidify_strength))
 
     # Final Invert for Potrace
     final_binary = cv2.bitwise_not(binary)
@@ -51,14 +42,16 @@ def process_vector(image, invert_input, contrast_boost, detail_threshold, solid_
     temp_bmp = "temp_trace.bmp"
     cv2.imwrite(temp_bmp, final_binary)
 
-    # Vectorize with precision settings
-    output_svg = "ts_vector_ultimate.svg"
+    # --- VECTORIZE WITH SMOOTHING ENGINE ---
+    output_svg = "ts_vector_smooth.svg"
     
+    # Potrace Commands (Ye hai magic settings)
     cmd = [
         "potrace", temp_bmp,
         "-s", "-o", output_svg,
-        "-t", "0", # Turdsize 0: मतलब एक पिक्सेल भी नहीं छोड़ना है। सब पकड़ो।
-        "-a", "0", # Alphamax 0: कोई अपनी मर्जी से स्मूथिंग नहीं। जैसा है वैसा ट्रेस करो।
+        "-t", "2",   # Despeckle: Chote daane hatayega taaki curve kharab na ho
+        "-a", str(smooth_factor), # Alphamax: YE HAI GOLAI KA SLIDER (0=Sharp, 1.3=Very Round)
+        "--longcurve", # CNC ke liye lambi smooth lines banane ki koshish karega
         "--opaque"
     ]
     
@@ -74,39 +67,37 @@ def process_vector(image, invert_input, contrast_boost, detail_threshold, solid_
 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: #0a0a0a !important; color: white !important; }
-#main_card { border: 2px solid #FFD700; border-radius: 15px; padding: 25px; background: #111; }
-.primary-btn { background: linear-gradient(45deg, #FFD700, #FF8C00) !important; color: black !important; font-weight: 900 !important; letter-spacing: 1px; }
+#main_card { border: 2px solid #00b8ff; border-radius: 15px; padding: 25px; background: #111; }
+.primary-btn { background: linear-gradient(45deg, #00b8ff, #00ff88) !important; color: black !important; font-weight: 900 !important; letter-spacing: 1px; }
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as app:
     with gr.Column(elem_id="main_card"):
-        gr.Markdown("# 💀 TS VECTOR ULTIMATE", elem_id="logo_text")
-        gr.Markdown("### The Final Attempt for Extreme Detail (Zero Compromise)")
+        gr.Markdown("# 🌀 TS VECTOR V8 (Smooth)", elem_id="logo_text")
+        gr.Markdown("### Perfect Curves for CNC Cutting")
         
         with gr.Row():
             with gr.Column():
-                inp_img = gr.Image(type="pil", label="Upload The Hardest Photo", height=300)
+                inp_img = gr.Image(type="pil", label="Upload Photo", height=300)
                 
-                gr.Markdown("### ⚙️ Critical Controls (ध्यान से सेट करें)")
+                gr.Markdown("### ⚙️ Controls")
                 inv_chk = gr.Checkbox(label="Is Design WHITE on BLACK?", value=False)
                 
-                # 1. Contrast Boost
-                cont_sld = gr.Slider(1.0, 3.0, value=1.5, step=0.1, label="1. Contrast Booster (लाइनों को उभारने के लिए)")
-
-                # 2. The Main Threshold
-                thresh_sld = gr.Slider(0, 255, value=180, step=1, label="2. Detail Cutoff (इसे घुमाकर प्रीव्यू में चेक करें)")
-                
-                # 3. Solid Force
-                solid_sld = gr.Slider(0, 3, value=0, step=1, label="3. Solid Force (बारीक डिज़ाइन के लिए 0 रखें)")
+                # Basic Controls
+                cont_sld = gr.Slider(1.0, 2.5, value=1.5, step=0.1, label="1. Contrast (Detail Ubharo)")
+                thresh_sld = gr.Slider(0, 255, value=170, step=1, label="2. Detail Cutoff (Preview dekho)")
+                solid_sld = gr.Slider(0, 3, value=0, step=1, label="3. Solid Force (Mota karne ke liye)")
                 
-                noise_chk = gr.Checkbox(label="4. Clean Noise (बारीक डिज़ाइन में OFF रखें)", value=False)
+                # --- NEW SMOOTH SLIDER ---
+                gr.Markdown("### 🌀 Curve Control (Golai)")
+                smooth_sld = gr.Slider(0.0, 1.34, value=1.0, step=0.1, label="4. Smoothing Factor (0=Kata hua, 1.0=Mast Golai)")
                 
-                btn = gr.Button("⚡ TRACE EXACT PIXELS", variant="primary", elem_classes=["primary-btn"])
+                btn = gr.Button("⚡ CREATE SMOOTH VECTOR", variant="primary", elem_classes=["primary-btn"])
 
             with gr.Column():
-                preview_img = gr.Image(label="Computer Vision Preview (What you see is what you get)", interactive=False)
+                preview_img = gr.Image(label="Computer Vision Preview (Black = Cut)", interactive=False)
                 out_file = gr.File(label="Download SVG")
 
-    btn.click(process_vector, inputs=[inp_img, inv_chk, cont_sld, thresh_sld, solid_sld, noise_chk], outputs=[out_file, preview_img])
+    btn.click(process_vector, inputs=[inp_img, inv_chk, cont_sld, thresh_sld, solid_sld, smooth_sld], outputs=[out_file, preview_img])
 
 app.launch()