Update app.py

app.py

@@ -1,73 +1,64 @@
 import gradio as gr
 import cv2
 import numpy as np
-from PIL import Image, ImageOps
+from PIL import Image, ImageOps, ImageEnhance
 import subprocess
 
-# --- 1. CNC BRAIN V5.5 (Solid + Detail Preserved) ---
-def process_vector(image, invert_input, detail_level, solidify_strength, smooth_factor, remove_noise):
+# --- CNC BRAIN: THE ULTIMATE (Zero Blur, Pure Detail) ---
+def process_vector(image, invert_input, contrast_boost, detail_threshold, solid_force, remove_noise):
     if image is None: return None, None
 
-    # --- STEP 1: HD UPSCALE (V5 ka magic - Quality ke liye) ---
+    # 1. HD UPSCALE (Resolution is king for details)
     w, h = image.size
-    # Photo ko 2 guna bada karo taaki curves smooth aayein
+    # 2x बड़ा कर रहे हैं ताकि बारीक पिक्सेल अलग हो सकें
     image = image.resize((w * 2, h * 2), Image.LANCZOS)
 
+    # 2. PRE-PROCESSING (Contrast Boost)
+    # OpenCV से पहले Pillow से कंट्रास्ट बढ़ाएंगे ताकि लाइनें उभर कर आएं
+    enhancer = ImageEnhance.Contrast(image)
+    image = enhancer.enhance(contrast_boost) # User controlled boost
+
     # --- INVERT LOGIC ---
     if invert_input:
         image = ImageOps.invert(image.convert("RGB"))
 
-    # Step A: Grayscale
+    # Grayscale conversion
     img_np = np.array(image.convert("RGB"))
     gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)
-    
-    # --- 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 4: SOLIDIFY (Tera favorite feature) ---
-    # Ye lines ko jodne aur solid banane ke liye hai
+    # --- THE SURGICAL THRESHOLD ---
+    # कोई ब्लर नहीं, कोई ऑटोमैटिक दिमाग नहीं। सीधा थ्रेशोल्ड।
+    # यह स्लाइडर तय करेगा कि किस शेड को काला मानना है।
+    # बारीक डिज़ाइन के लिए यह सबसे सटीक तरीका है।
+    _, binary = cv2.threshold(gray, detail_threshold, 255, cv2.THRESH_BINARY_INV)
+
+    # --- SOLID FORCE (Optional) ---
+    # सिर्फ तब इस्तेमाल करें जब लाइनें टूट रही हों।
+    # बारीक डिज़ाइन में इसे 0 रखना ही बेहतर है।
     kernel = np.ones((3,3), np.uint8)
-    
-    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 5: Noise Removal
+    if solid_force > 0:
+        binary = cv2.dilate(binary, kernel, iterations=int(solid_force))
+
+    # --- NOISE REMOVAL (Dangerous for fine detail) ---
+    # इसे बहुत सोच समझ कर ऑन करना, यह बारीक बिन्दुओं को उड़ा सकता है।
     if remove_noise:
         binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)
 
-    # Step 6: Invert for Potrace
+    # Final Invert for Potrace
     final_binary = cv2.bitwise_not(binary)
 
     # Save Temp
     temp_bmp = "temp_trace.bmp"
     cv2.imwrite(temp_bmp, final_binary)
 
-    # Step 7: Vectorize
-    output_svg = "ts_vector_solid_detail.svg"
+    # Vectorize with precision settings
+    output_svg = "ts_vector_ultimate.svg"
     
     cmd = [
         "potrace", temp_bmp,
         "-s", "-o", output_svg,
-        "-t", "2",
-        "-a", str(smooth_factor),
+        "-t", "0", # Turdsize 0: मतलब एक पिक्सेल भी नहीं छोड़ना है। सब पकड़ो।
+        "-a", "0", # Alphamax 0: कोई अपनी मर्जी से स्मूथिंग नहीं। जैसा है वैसा ट्रेस करो।
         "--opaque"
     ]
     
@@ -79,43 +70,43 @@ def process_vector(image, invert_input, detail_level, solidify_strength, smooth_
 
     return output_svg, Image.fromarray(final_binary)
 
-# --- 2. Interface ---
+# --- 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: #00ff88 !important; color: black !important; font-weight: bold !important; }
+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; }
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as app:
     with gr.Column(elem_id="main_card"):
-        gr.Markdown("# 🏆 TS VECTOR V5.5", elem_id="logo_text")
-        gr.Markdown("### Solid Shapes + Fine Detail Control")
+        gr.Markdown("# 💀 TS VECTOR ULTIMATE", elem_id="logo_text")
+        gr.Markdown("### The Final Attempt for Extreme Detail (Zero Compromise)")
         
         with gr.Row():
             with gr.Column():
-                inp_img = gr.Image(type="pil", label="Upload Photo", height=300)
+                inp_img = gr.Image(type="pil", label="Upload The Hardest Photo", height=300)
                 
-                gr.Markdown("### 🎨 Image Type")
+                gr.Markdown("### ⚙️ Critical Controls (ध्यान से सेट करें)")
                 inv_chk = gr.Checkbox(label="Is Design WHITE on BLACK?", value=False)
                 
-                gr.Markdown("### 🎛️ Settings (Adjust for Details)")
-                
-                # 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)")
+                # 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 (इसे घुमाकर प्रीव्यू में चेक करें)")
                 
-                # Tera favorite Solidify slider wapas aa gaya
-                solid_sld = gr.Slider(-2, 5, value=1, step=1, label="2. Solidify (Gap Bharne ke liye)")
+                # 3. Solid Force
+                solid_sld = gr.Slider(0, 3, value=0, step=1, label="3. Solid Force (बारीक डिज़ाइन के लिए 0 रखें)")
                 
-                noise_chk = gr.Checkbox(label="3. Clean Noise", value=True)
-                smooth_sld = gr.Slider(0, 1.3, value=1.0, label="4. Smoothing")
+                noise_chk = gr.Checkbox(label="4. Clean Noise (बारीक डिज़ाइन में OFF रखें)", value=False)
                 
-                btn = gr.Button("🔥 CREATE SOLID VECTOR", variant="primary", elem_classes=["primary-btn"])
+                btn = gr.Button("⚡ TRACE EXACT PIXELS", variant="primary", elem_classes=["primary-btn"])
 
             with gr.Column():
-                preview_img = gr.Image(label="Computer Vision Preview (Black = Solid Material)", interactive=False)
+                preview_img = gr.Image(label="Computer Vision Preview (What you see is what you get)", interactive=False)
                 out_file = gr.File(label="Download SVG")
 
-    btn.click(process_vector, inputs=[inp_img, inv_chk, detail_sld, solid_sld, smooth_sld, noise_chk], outputs=[out_file, preview_img])
+    btn.click(process_vector, inputs=[inp_img, inv_chk, cont_sld, thresh_sld, solid_sld, noise_chk], outputs=[out_file, preview_img])
 
 app.launch()