Update app.py

app.py

@@ -30,8 +30,6 @@ def convert_svg_to_ai(svg_path, image_width, image_height):
     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
@@ -68,16 +66,11 @@ def convert_svg_to_ai(svg_path, image_width, image_height):
         ai_file.write("\n%%EndDocument\n")
         ai_file.write("\n%%Trailer\n%%EOF\n")
     
-
-
-    # AI 파일 헤더 생성 및 파일 변환 로직 (이전과 동일)
-    
     # HTML 미리보기 생성
     preview_html = create_preview_html(svg_path, image_width, image_height)
     
     return ai_path, preview_html
 
-    
 def convert_to_vector(
     image, 
     save_svg,
@@ -129,16 +122,14 @@ def convert_to_vector(
 
         # 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>')
+            preview = gr.HTML(create_preview_html(svg_path, image.width, image.height))
             outputs.append(svg_path)
         
         # AI 파일 처리
         if save_ai:
-            ai_path, preview_html = convert_svg_to_ai(svg_path)
+            ai_path, ai_preview_html = convert_svg_to_ai(svg_path, image.width, image.height)
             outputs.append(ai_path)
-            ai_preview = preview_html
+            ai_preview = gr.HTML(ai_preview_html)
 
         if not save_svg:  # SVG가 선택되지 않았다면 임시 파일 삭제
             os.remove(svg_path)