Update app.py

app.py

@@ -1,23 +1,56 @@
 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로 변환
+    """SVG를 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)
+    # SVG 파일 읽기
+    with open(svg_path, 'r') as svg_file:
+        svg_content = svg_file.read()
+    
+    # AI 파일 헤더
+    ai_header = """%!PS-Adobe-3.0
+%%Creator: Adobe Illustrator(TM) SVG Converter
+%%AI8_CreatorVersion: 24.0.0
+%%For: SVG to AI Converter
+%%Title: Generated AI File
+%%CreationDate: %(date)s
+%%BoundingBox: 0 0 800 600
+%%HiResBoundingBox: 0 0 800 600
+%%DocumentData: Clean7Bit
+%%LanguageLevel: 2
+%%DocumentNeededResources: procset Adobe_packedarray 2.0 0
+%%+ procset Adobe_cmykcolor 1.1 0
+%%+ procset Adobe_cshow 1.1 0
+%%+ procset Adobe_customcolor 1.0 0
+%%+ procset Adobe_typography_AI5 1.0 1
+%%+ procset Adobe_pattern_AI3 1.0 1
+%%+ procset Adobe_Illustrator_AI3 1.0 1
+%%EndComments
+%%BeginProlog
+"""
+    
+    # AI 파일 생성
+    with open(ai_path, 'w', encoding='utf-8') as ai_file:
+        # 헤더 작성
+        ai_file.write(ai_header)
+        
+        # SVG 내용을 AI 파일에 포함
+        ai_file.write("\n%%BeginDocument\n")
+        ai_file.write("/SVGContent\n<<\n")
+        ai_file.write("/Type /SVG\n")
+        ai_file.write("/Version 1.1\n")
+        ai_file.write("/Content [\n")
+        ai_file.write(svg_content)
+        ai_file.write("\n]\n")
+        ai_file.write(">>\ndef\n")
+        ai_file.write("\n%%EndDocument\n")
+        
+        # 파일 종료
+        ai_file.write("\n%%Trailer\n%%EOF\n")
+    
     return ai_path
 
 def convert_to_vector(
@@ -39,47 +72,63 @@ def convert_to_vector(
     if not (save_svg or save_ai):
         return None, None, None  # 둘 다 선택되지 않은 경우
 
+    if image is None:
+        return None, None, None
+
     input_path = "temp_input.jpg"
     svg_path = "svg_output.svg"
     outputs = []
     preview = None
 
-    # 입력 이미지를 임시 파일로 저장
-    image.save(input_path)
-
-    # VTracer를 사용하여 이미지를 SVG로 변환
-    vtracer.convert_image_to_svg_py(
-        input_path,
-        svg_path,
-        colormode=colormode,
-        hierarchical=hierarchical,
-        mode=mode,
-        filter_speckle=int(filter_speckle),
-        color_precision=int(color_precision),
-        layer_difference=int(layer_difference),
-        corner_threshold=int(corner_threshold),
-        length_threshold=float(length_threshold),
-        max_iterations=int(max_iterations),
-        splice_threshold=int(splice_threshold),
-        path_precision=int(path_precision)
-    )
+    try:
+        # 입력 이미지를 임시 파일로 저장
+        image.save(input_path)
 
-    # 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)
+        # VTracer를 사용하여 이미지를 SVG로 변환
+        vtracer.convert_image_to_svg_py(
+            input_path,
+            svg_path,
+            colormode=colormode,
+            hierarchical=hierarchical,
+            mode=mode,
+            filter_speckle=int(filter_speckle),
+            color_precision=int(color_precision),
+            layer_difference=int(layer_difference),
+            corner_threshold=int(corner_threshold),
+            length_threshold=float(length_threshold),
+            max_iterations=int(max_iterations),
+            splice_threshold=int(splice_threshold),
+            path_precision=int(path_precision)
+        )
+
+        # SVG 파일 처리
+        if save_svg:
+            with open(svg_path, "r", encoding='utf-8') 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)
+
+        return preview, outputs[0] if outputs else None, outputs[1] if len(outputs) > 1 else None
 
-    if not save_svg:  # SVG가 선택되지 않았다면 임시 파일 삭제
-        os.remove(svg_path)
+    except Exception as e:
+        print(f"Error during conversion: {str(e)}")
+        return None, None, None
 
-    return preview, outputs[0] if outputs else None, outputs[1] if len(outputs) > 1 else None
+    finally:
+        # 임시 파일 정리
+        if os.path.exists(input_path):
+            try:
+                os.remove(input_path)
+            except:
+                pass
 
 css = """
 #col-container {