Update app.py

app.py

@@ -1,9 +1,29 @@
 import gradio as gr
 import vtracer
 import os
+from svglib.svglib import svg2rlg
+from reportlab.graphics import renderPDF
+import subprocess
+
+def convert_svg_to_ai(svg_path):
+    # SVG를 PDF로 먼저 변환
+    drawing = svg2rlg(svg_path)
+    pdf_path = svg_path.replace('.svg', '.pdf')
+    renderPDF.drawToFile(drawing, pdf_path)
+    
+    # PDF를 AI로 변환
+    ai_path = svg_path.replace('.svg', '.ai')
+    subprocess.run(['gs', '-dNOPAUSE', '-dBATCH', '-sDEVICE=pdfwrite',
+                   '-sOutputFile=' + ai_path, pdf_path])
+    
+    # 임시 PDF 파일 삭제
+    os.remove(pdf_path)
+    return ai_path
 
 def convert_to_vector(
     image, 
+    save_svg,
+    save_ai,
     colormode="color", 
     hierarchical="stacked", 
     mode="spline", 
@@ -16,16 +36,21 @@ def convert_to_vector(
     splice_threshold=45, 
     path_precision=3
 ):
+    if not (save_svg or save_ai):
+        return None, None, None  # 둘 다 선택되지 않은 경우
+
     input_path = "temp_input.jpg"
-    output_path = "svg_output.svg"
+    svg_path = "svg_output.svg"
+    outputs = []
+    preview = None
 
-    # Save the input image to a temporary file
+    # 입력 이미지를 임시 파일로 저장
     image.save(input_path)
 
-    # Convert the image to SVG using VTracer
+    # VTracer를 사용하여 이미지를 SVG로 변환
     vtracer.convert_image_to_svg_py(
         input_path,
-        output_path,
+        svg_path,
         colormode=colormode,
         hierarchical=hierarchical,
         mode=mode,
@@ -38,24 +63,23 @@ def convert_to_vector(
         splice_threshold=int(splice_threshold),
         path_precision=int(path_precision)
     )
-    # Read the SVG output
-    with open(output_path, "r") as f:
-        svg_content = f.read()
-
-    # Return the SVG file path instead of content
-    return gr.HTML(f'<svg viewBox="0 0 {image.width} {image.height}">{svg_content}</svg>'), output_path
-    # return output_path,output_path
 
-def handle_color_mode(value):
-    # You can change this to display the selected value without any prefix
-    return value
+    # SVG 파일 처리
+    if save_svg:
+        with open(svg_path, "r") as f:
+            svg_content = f.read()
+        preview = gr.HTML(f'<svg viewBox="0 0 {image.width} {image.height}">{svg_content}</svg>')
+        outputs.append(svg_path)
+    
+    # AI 파일 처리
+    if save_ai:
+        ai_path = convert_svg_to_ai(svg_path)
+        outputs.append(ai_path)
 
+    if not save_svg:  # SVG가 선택되지 않았다면 임시 파일 삭제
+        os.remove(svg_path)
 
-examples = [
-    "examples/11.jpg",
-    "examples/02.jpg",
-    "examples/03.jpg",
-]
+    return preview, outputs[0] if outputs else None, outputs[1] if len(outputs) > 1 else None
 
 css = """
 #col-container {
@@ -73,18 +97,21 @@ css = """
 }
 """
 
-# Define the Gradio interface
+# Gradio 인터페이스 정의
 with gr.Blocks(css=css) as app:
     with gr.Column(elem_id="col-container"):
         gr.HTML("""
                 <div style="text-align: center;">
                     <h2>Image to Vector Converter ⚡</h2>
-                    <p>Converts raster images (JPG, PNG, WEBP) to vector graphics (SVG).</p>
+                    <p>Converts raster images (JPG, PNG, WEBP) to vector graphics (SVG, AI).</p>
                 </div>
             """)
         with gr.Row():
             with gr.Column():
                 image_input = gr.Image(type="pil", label="Upload Image")
+                with gr.Row():
+                    save_svg = gr.Checkbox(value=True, label="Save as SVG")
+                    save_ai = gr.Checkbox(value=False, label="Save as AI")
                 with gr.Accordion("Advanced Settings", open=False):
                     with gr.Accordion("Clustering", open=False):
                         colormode = gr.Radio([("COLOR","color"),("B/W", "binary")], value="color", label="Color Mode", show_label=False)
@@ -95,72 +122,65 @@ with gr.Blocks(css=css) as app:
                     with gr.Accordion("Curve Fitting", open=False):
                         mode = gr.Radio([("SPLINE","spline"),("POLYGON", "polygon"), ("PIXEL","none")], value="spline", label="Mode", show_label=False)
                         corner_threshold = gr.Slider(0, 180, value=60, step=1, label="Corner Threshold", info="Smoother")
-                        length_threshold = gr.Slider(3.5, 10, value=4.0, step=0.1, label="Segment Length", info ="More coarse")
+                        length_threshold = gr.Slider(3.5, 10, value=4.0, step=0.1, label="Segment Length", info="More coarse")
                         splice_threshold = gr.Slider(0, 180, value=45, step=1, label="Splice Threshold", info="Less accurate")
                         max_iterations = gr.Slider(1, 20, value=10, step=1, label="Max Iterations", visible=False)
                         path_precision = gr.Slider(1, 10, value=3, step=1, label="Path Precision", visible=False)
                         output_text = gr.Textbox(label="Selected Mode", visible=False)
                 with gr.Row():
                     clear_button = gr.Button("Clear")
-                    convert_button = gr.Button("✨ Convert to SVG", variant='primary', elem_classes=["generate-btn"])
+                    convert_button = gr.Button("✨ Convert", variant='primary', elem_classes=["generate-btn"])
 
             with gr.Column():
-                html = gr.HTML(label="SVG Output") # container=True, show_label=True
-                svg_output = gr.File(label="Download SVG")
+                preview = gr.HTML(label="Preview")
+                svg_output = gr.File(label="SVG Output", visible=True)
+                ai_output = gr.File(label="AI Output", visible=True)
+
+        examples = [
+            "examples/11.jpg",
+            "examples/02.jpg",
+            "examples/03.jpg",
+        ]
         
         gr.Examples(
-            examples = examples,
-            fn = convert_to_vector,
-            inputs = [image_input],
-            outputs = [html,svg_output],
+            examples=examples,
+            fn=convert_to_vector,
+            inputs=[
+                image_input,
+                save_svg,
+                save_ai
+            ],
+            outputs=[preview, svg_output, ai_output],
             cache_examples=False,
-            run_on_click = True
+            run_on_click=True
         )
-    # Store default values for restoration
-    colormode.change(handle_color_mode, inputs=colormode,outputs=output_text)
-    hierarchical.change(handle_color_mode, inputs=hierarchical,outputs=output_text)
-    mode.change(handle_color_mode, inputs=mode,outputs=output_text)
-    default_values = {
-        "color_precision": 6,
-        "layer_difference": 16
-    }
 
+    # 이벤트 핸들러
     def clear_inputs():
-        return gr.Image(value=None), gr.Radio(value="color"), gr.Radio(value="stacked"), gr.Radio(value="spline"), gr.Slider(value=4), gr.Slider(value=6), gr.Slider(value=16), gr.Slider(value=60), gr.Slider(value=4.0), gr.Slider(value=10), gr.Slider(value=45), gr.Slider(value=3)
+        return [None] * 12
 
+    def update_output_visibility(save_svg, save_ai):
+        return gr.update(visible=save_svg), gr.update(visible=save_ai)
 
-    def update_interactivity_and_visibility(colormode, color_precision_value, layer_difference_value):
-        is_color_mode = colormode == "color"
-        return (
-            gr.update(interactive=is_color_mode),
-            gr.update(interactive=is_color_mode),
-            gr.update(visible=is_color_mode)  # Show/Hide Hierarchical Mode
-        )
-
-    colormode.change(
-        update_interactivity_and_visibility,
-        inputs=[colormode, color_precision, layer_difference],
-        outputs=[color_precision, layer_difference, hierarchical]
+    # 체크박스 상태에 따른 출력 파일 컴포넌트 표시/숨김
+    save_svg.change(
+        update_output_visibility,
+        inputs=[save_svg, save_ai],
+        outputs=[svg_output, ai_output]
     )
-
-    def update_interactivity_and_visibility_for_mode(mode):
-        is_spline_mode = mode == "spline"
-        return (
-            gr.update(interactive=is_spline_mode),
-            gr.update(interactive=is_spline_mode),
-            gr.update(interactive=is_spline_mode)
-        )
-
-    mode.change(
-        update_interactivity_and_visibility_for_mode,
-        inputs=[mode],
-        outputs=[corner_threshold,length_threshold,splice_threshold]
+    save_ai.change(
+        update_output_visibility,
+        inputs=[save_svg, save_ai],
+        outputs=[svg_output, ai_output]
     )
 
-    clear_button.click(
-        clear_inputs,
-        outputs=[
+    # 변환 버튼 클릭 이벤트
+    convert_button.click(
+        convert_to_vector,
+        inputs=[
             image_input,
+            save_svg,
+            save_ai,
             colormode,
             hierarchical,
             mode,
@@ -172,12 +192,14 @@ with gr.Blocks(css=css) as app:
             max_iterations,
             splice_threshold,
             path_precision
-        ]
+        ],
+        outputs=[preview, svg_output, ai_output]
     )
 
-    convert_button.click(
-        convert_to_vector,
-        inputs=[
+    # Clear 버튼 이벤트
+    clear_button.click(
+        clear_inputs,
+        outputs=[
             image_input,
             colormode,
             hierarchical,
@@ -190,9 +212,7 @@ with gr.Blocks(css=css) as app:
             max_iterations,
             splice_threshold,
             path_precision
-        ],
-        outputs=[html,svg_output]
+        ]
     )
 
-# Launch the app
-app.launch(debug=True)
+app.launch(debug=True)