Create app.py

app.py

@@ -0,0 +1,142 @@
+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
+
+    # 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
+
+    # Step B: Prepare for Tracing (Black & White conversion)
+    # Alpha channel (transparency) ko mask banao
+    img_np = np.array(output_image)
+    
+    # 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)
+
+    # Save temp BMP (Potrace needs BMP)
+    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)
+    output_svg = "ts_vector_design.svg"
+    
+    # Smoothness slider logic (0.5 to 1.3)
+    corner_policy = "1" # Turn off optimization
+    
+    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
+    ]
+    
+    subprocess.run(cmd, check=True)
+
+    # Return SVG path and Preview Image (Black & White version)
+    return output_svg, Image.fromarray(binary)
+
+# --- 2. Premium TS 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: #0f172a !important; /* Rich Dark Blue */
+    color: #e2e8f0 !important;
+}
+
+#main_card {
+    background: rgba(30, 41, 59, 0.7);
+    backdrop-filter: blur(20px);
+    border: 1px solid rgba(255, 255, 255, 0.1);
+    border-radius: 24px;
+    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;
+    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);
+}
+"""
+
+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")
+        
+        with gr.Row():
+            # Left: Inputs
+            with gr.Column(scale=1):
+                inp_img = gr.Image(type="pil", label="Upload Jali Design (Jpg/Png)")
+                
+                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)")
+                
+                btn = gr.Button("🔥 CONVERT TO SVG", 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.*")
+
+    btn.click(convert_to_vector, inputs=[inp_img, thresh, smooth], outputs=[out_file, preview_img])
+
+app.launch()