Update app.py

app.py

@@ -4,12 +4,18 @@ import numpy as np
 from PIL import Image, ImageOps
 import subprocess
 
-# --- 1. CNC BRAIN V4.5 (Solid Shape + Invert Logic) ---
+# --- 1. CNC BRAIN V5 (HD Upscale + Smooth Logic) ---
 def process_vector(image, invert_input, threshold_val, solidify_strength, smooth_factor, remove_noise):
     if image is None: return None, None
 
-    # --- NEW FIX: INVERT LOGIC ---
-    # Agar design White hai aur Background Black, toh hum pehle hi usse palat denge
+    # --- 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.
+    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"))
 
@@ -17,10 +23,10 @@ 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)
     
-    # Blur to mix highlights
-    gray_blurred = cv2.GaussianBlur(gray, (5, 5), 0)
+    # Blur thoda badhaya hai taaki pixelate na ho
+    gray_blurred = cv2.GaussianBlur(gray, (9, 9), 0)
 
-    # Step B: Thresholding (Solid Black logic)
+    # Step B: Thresholding
     _, binary = cv2.threshold(gray_blurred, threshold_val, 255, cv2.THRESH_BINARY_INV)
 
     # Step C: SOLIDIFY (Double Line Fixer)
@@ -42,13 +48,14 @@ def process_vector(image, invert_input, threshold_val, solidify_strength, smooth
     cv2.imwrite(temp_bmp, final_binary)
 
     # Step F: Vectorize
-    output_svg = "ts_vector_solid.svg"
+    output_svg = "ts_vector_hd.svg"
     
     cmd = [
         "potrace", temp_bmp,
         "-s", "-o", output_svg,
         "-t", "10", 
-        "-a", str(smooth_factor),
+        "-a", str(smooth_factor), # Curve smoothness controlled here
+        "--alphamax", "1.34",     # Maximize curves
         "--opaque"
     ]
     
@@ -65,28 +72,28 @@ 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 V4.5", elem_id="logo_text")
+        gr.Markdown("# 🏆 TS VECTOR HD (V5)", elem_id="logo_text")
+        gr.Markdown("### Auto-Upscale for Smooth Curves")
         
         with gr.Row():
             with gr.Column():
                 inp_img = gr.Image(type="pil", label="Upload Photo", height=300)
                 
-                gr.Markdown("### 🎨 Image Type (Ye naya hai)")
-                # Yahi wo magic button hai jo White Design ko fix karega
+                gr.Markdown("### 🎨 Image Type")
                 inv_chk = gr.Checkbox(label="Is Design WHITE on BLACK Background?", value=False)
                 
                 gr.Markdown("### 🎛️ Settings")
                 thresh_sld = gr.Slider(0, 255, value=140, step=1, label="Darkness Threshold")
-                solid_sld = gr.Slider(0, 5, value=2, step=1, label="Solidify (Double Line Fixer)")
+                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.3, value=1.0, label="Smoothing")
+                smooth_sld = gr.Slider(0, 1.5, value=1.2, label="Smoothing (High for Round shapes)")
                 
-                btn = gr.Button("🔥 CREATE VECTOR", variant="primary", elem_classes=["primary-btn"])
+                btn = gr.Button("🔥 CREATE HD VECTOR", variant="primary", elem_classes=["primary-btn"])
 
             with gr.Column():
                 preview_img = gr.Image(label="Computer Vision Preview", interactive=False)