Update app.py

app.py

@@ -1,18 +1,21 @@
 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 생성"""
+def create_preview_html(svg_path):
+    """SVG를 HTML 미리보기로 변환"""
     try:
-        drawing = svg2rlg(svg_path)
-        pdf_path = svg_path.replace('.svg', '_preview.pdf')
-        renderPDF.drawToFile(drawing, pdf_path)
-        return pdf_path
+        with open(svg_path, 'r', encoding='utf-8') as f:
+            svg_content = f.read()
+        # SVG를 HTML로 감싸서 반환
+        preview_html = f'''
+        <div style="width:100%; height:100%; background-color: white; padding: 20px;">
+            {svg_content}
+        </div>
+        '''
+        return preview_html
     except Exception as e:
-        print(f"PDF 미리보기 생성 실패: {str(e)}")
+        print(f"미리보기 생성 실패: {str(e)}")
         return None
 
 def convert_svg_to_ai(svg_path):
@@ -59,10 +62,10 @@ def convert_svg_to_ai(svg_path):
         ai_file.write("\n%%EndDocument\n")
         ai_file.write("\n%%Trailer\n%%EOF\n")
     
-    # PDF 미리보기 생성
-    pdf_preview = create_pdf_preview(svg_path)
+    # HTML 미리보기 생성
+    preview_html = create_preview_html(svg_path)
     
-    return ai_path, pdf_preview
+    return ai_path, preview_html
 
 def convert_to_vector(
     image, 
@@ -122,10 +125,9 @@ def convert_to_vector(
         
         # AI 파일 처리
         if save_ai:
-            ai_path, pdf_preview = convert_svg_to_ai(svg_path)
+            ai_path, preview_html = convert_svg_to_ai(svg_path)
             outputs.append(ai_path)
-            if pdf_preview:
-                ai_preview = pdf_preview
+            ai_preview = preview_html
 
         if not save_svg:  # SVG가 선택되지 않았다면 임시 파일 삭제
             os.remove(svg_path)
@@ -224,7 +226,7 @@ with gr.Blocks(css=css) as app:
                         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_preview = gr.HTML(label="AI Preview")
                         ai_output = gr.File(label="Download AI", visible=True)
 
         examples = [