Update app.py

app.py

@@ -1,6 +1,19 @@
 import gradio as gr
 import vtracer
 import os
+from svglib.svglib import svg2rlg
+from reportlab.graphics import renderPDF
+
+def create_pdf_preview(svg_path):
+    """SVG를 PDF로 변환하여 미리보기용 PDF 생성"""
+    try:
+        drawing = svg2rlg(svg_path)
+        pdf_path = svg_path.replace('.svg', '_preview.pdf')
+        renderPDF.drawToFile(drawing, pdf_path)
+        return pdf_path
+    except Exception as e:
+        print(f"PDF 미리보기 생성 실패: {str(e)}")
+        return None
 
 def convert_svg_to_ai(svg_path):
     """SVG를 AI 파일로 변환"""
@@ -14,7 +27,7 @@ def convert_svg_to_ai(svg_path):
     ai_header = """%!PS-Adobe-3.0
 %%Creator: Adobe Illustrator(TM) SVG Converter
 %%AI8_CreatorVersion: 24.0.0
-%%For: SVG to AI Converter
+%%For: VectorFlow
 %%Title: Generated AI File
 %%CreationDate: %(date)s
 %%BoundingBox: 0 0 800 600
@@ -34,10 +47,7 @@ def convert_svg_to_ai(svg_path):
     
     # 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")
@@ -47,11 +57,12 @@ def convert_svg_to_ai(svg_path):
         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
+    # PDF 미리보기 생성
+    pdf_preview = create_pdf_preview(svg_path)
+    
+    return ai_path, pdf_preview
 
 def convert_to_vector(
     image, 
@@ -70,15 +81,16 @@ def convert_to_vector(
     path_precision=3
 ):
     if not (save_svg or save_ai):
-        return None, None, None  # 둘 다 선택되지 않은 경우
-
+        return None, None, None, None  # Preview, SVG output, AI output, AI preview
+    
     if image is None:
-        return None, None, None
+        return None, None, None, None
 
     input_path = "temp_input.jpg"
     svg_path = "svg_output.svg"
     outputs = []
     preview = None
+    ai_preview = None
 
     try:
         # 입력 이미지를 임시 파일로 저장
@@ -110,17 +122,24 @@ def convert_to_vector(
         
         # AI 파일 처리
         if save_ai:
-            ai_path = convert_svg_to_ai(svg_path)
+            ai_path, pdf_preview = convert_svg_to_ai(svg_path)
             outputs.append(ai_path)
+            if pdf_preview:
+                ai_preview = pdf_preview
 
         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
+        return (
+            preview,
+            outputs[0] if outputs else None,
+            outputs[1] if len(outputs) > 1 else None,
+            ai_preview
+        )
 
     except Exception as e:
         print(f"Error during conversion: {str(e)}")
-        return None, None, None
+        return None, None, None, None
 
     finally:
         # 임시 파일 정리
@@ -144,15 +163,34 @@ css = """
     transform: translateY(-2px);
     box-shadow: 0 5px 15px rgba(0,0,0,0.2);
 }
+.container {
+    max-width: 1200px;
+    margin: 0 auto;
+    padding: 20px;
+}
+.header {
+    text-align: center;
+    margin-bottom: 40px;
+}
+.header h1 {
+    font-size: 2.5em;
+    color: #2c3e50;
+    margin-bottom: 10px;
+}
+.header p {
+    color: #7f8c8d;
+    font-size: 1.2em;
+}
 """
 
 # 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, AI).</p>
+                <div class="header">
+                    <h1>VectorFlow ⚡</h1>
+                    <p>Transform your images into professional vector graphics</p>
+                    <p style="font-size: 0.9em; color: #95a5a6;">Supports JPG, PNG, WEBP → SVG, AI</p>
                 </div>
             """)
         with gr.Row():
@@ -181,9 +219,13 @@ with gr.Blocks(css=css) as app:
                     convert_button = gr.Button("✨ Convert", variant='primary', elem_classes=["generate-btn"])
 
             with gr.Column():
-                preview = gr.HTML(label="Preview")
-                svg_output = gr.File(label="SVG Output", visible=True)
-                ai_output = gr.File(label="AI Output", visible=True)
+                with gr.Tabs():
+                    with gr.Tab("SVG Preview"):
+                        preview = gr.HTML(label="SVG Preview")
+                        svg_output = gr.File(label="Download SVG", visible=True)
+                    with gr.Tab("AI Preview"):
+                        ai_preview = gr.PDF(label="AI Preview")
+                        ai_output = gr.File(label="Download AI", visible=True)
 
         examples = [
             ["examples/11.jpg", True, False],
@@ -199,7 +241,7 @@ with gr.Blocks(css=css) as app:
                 save_svg,
                 save_ai
             ],
-            outputs=[preview, svg_output, ai_output],
+            outputs=[preview, svg_output, ai_output, ai_preview],
             cache_examples=False,
             run_on_click=True
         )
@@ -209,18 +251,23 @@ with gr.Blocks(css=css) as app:
         return [None] * 12
 
     def update_output_visibility(save_svg, save_ai):
-        return gr.update(visible=save_svg), gr.update(visible=save_ai)
+        return (
+            gr.update(visible=save_svg), 
+            gr.update(visible=save_ai), 
+            gr.update(visible=save_svg), 
+            gr.update(visible=save_ai)
+        )
 
     # 체크박스 상태에 따른 출력 파일 컴포넌트 표시/숨김
     save_svg.change(
         update_output_visibility,
         inputs=[save_svg, save_ai],
-        outputs=[svg_output, ai_output]
+        outputs=[preview, ai_preview, svg_output, ai_output]
     )
     save_ai.change(
         update_output_visibility,
         inputs=[save_svg, save_ai],
-        outputs=[svg_output, ai_output]
+        outputs=[preview, ai_preview, svg_output, ai_output]
     )
 
     # 변환 버튼 클릭 이벤트
@@ -242,7 +289,7 @@ with gr.Blocks(css=css) as app:
             splice_threshold,
             path_precision
         ],
-        outputs=[preview, svg_output, ai_output]
+        outputs=[preview, svg_output, ai_output, ai_preview]
     )
 
     # Clear 버튼 이벤트