Update app.py

app.py

@@ -4,68 +4,65 @@ from PIL import Image
 import vtracer
 import tempfile
 
-def convert_image(image, color_mode, hierarchical, mode, filter_speckle,
-                  color_precision, layer_difference, corner_threshold,
-                  length_threshold, max_iterations, splice_threshold, path_precision):
-    """Converts an image to SVG using vtracer with customizable parameters."""
+def converter_imagem(imagem, modo_cor, hierarquico, modo, filtro_ruido,
+                     precisao_cor, diferenca_camada, limite_canto,
+                     limite_comprimento, max_iteracoes, limite_emenda, precisao_caminho):
+    """Converte uma imagem para SVG usando vtracer com parâmetros personalizáveis."""
 
-    # Convert Gradio image to bytes for vtracer compatibility
+    # Converter a imagem do Gradio para bytes para compatibilidade com o vtracer
     img_byte_array = io.BytesIO()
-    image.save(img_byte_array, format='PNG')
+    imagem.save(img_byte_array, format='PNG')
     img_bytes = img_byte_array.getvalue()
 
-    # Perform the conversion
+    # Realizar a conversão
     svg_str = vtracer.convert_raw_image_to_svg(
         img_bytes,
         img_format='png',
-        colormode=color_mode.lower(),
-        hierarchical=hierarchical.lower(),
-        mode=mode.lower(),
-        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)
+        colormode=modo_cor.lower(),
+        hierarchical=hierarquico.lower(),
+        mode=modo.lower(),
+        filter_speckle=int(filtro_ruido),
+        color_precision=int(precisao_cor),
+        layer_difference=int(diferenca_camada),
+        corner_threshold=int(limite_canto),
+        length_threshold=float(limite_comprimento),
+        max_iterations=int(max_iteracoes),
+        splice_threshold=int(limite_emenda),
+        path_precision=int(precisao_caminho)
     )
 
-    # Save the SVG string to a temporary file
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.svg')
-    temp_file.write(svg_str.encode('utf-8'))
-    temp_file.close()
+    # Salvar o SVG em um arquivo temporário
+    arquivo_temporario = tempfile.NamedTemporaryFile(delete=False, suffix='.svg')
+    arquivo_temporario.write(svg_str.encode('utf-8'))
+    arquivo_temporario.close()
 
-    return gr.HTML(f'<svg viewBox="0 0 {image.width} {image.height}">{svg_str}</svg>'), temp_file.name
+    return gr.HTML(f'<svg viewBox="0 0 {imagem.width} {imagem.height}">{svg_str}</svg>'), arquivo_temporario.name
 
-# Gradio interface
+# Interface Gradio
 iface = gr.Blocks()
 
 with iface:
     
     gr.Interface(
-    fn=convert_image,
-    inputs=[
-        gr.Image(type="pil", label="Upload Image"),
-        gr.Radio(choices=["Color", "Binary"], value="Color", label="Color Mode"),
-        gr.Radio(choices=["Stacked", "Cutout"], value="Stacked", label="Hierarchical"),
-        gr.Radio(choices=["Spline", "Polygon", "None"], value="Spline", label="Mode"),
-        gr.Slider(minimum=1, maximum=10, value=4, step=1, label="Filter Speckle"),
-        gr.Slider(minimum=1, maximum=8, value=6, step=1, label="Color Precision"),
-        gr.Slider(minimum=1, maximum=32, value=16, step=1, label="Layer Difference"),
-        gr.Slider(minimum=10, maximum=90, value=60, step=1, label="Corner Threshold"),
-        gr.Slider(minimum=3.5, maximum=10, value=4.0, step=0.5, label="Length Threshold"),
-        gr.Slider(minimum=1, maximum=20, value=10, step=1, label="Max Iterations"),
-        gr.Slider(minimum=10, maximum=90, value=45, step=1, label="Splice Threshold"),
-        gr.Slider(minimum=1, maximum=10, value=8, step=1, label="Path Precision")
-    ],
-    outputs=[
-        gr.HTML(label="SVG Output"),
-        gr.File(label="Download SVG")
-    ],
-    title="Convert Image to SVG vectors",
-    description="Upload an image and customize the conversion parameters as needed.<br><h2>Support me USDT (TRC-20): TAe7hsSVWtMEYz3G5V1UiUdYPQVqm28bKx</h2>",
-)
+        fn=converter_imagem,
+        inputs=[
+            gr.Image(type="pil", label="Enviar Imagem"),
+            gr.Radio(choices=["Colorido", "Binário"], value="Colorido", label="Modo de Cor"),
+            gr.Radio(choices=["Empilhado", "Recortado"], value="Empilhado", label="Hierárquico"),
+            gr.Radio(choices=["Spline", "Polígono", "Nenhum"], value="Spline", label="Modo"),
+            gr.Slider(minimum=1, maximum=10, value=4, step=1, label="Filtro de Ruído"),
+            gr.Slider(minimum=1, maximum=8, value=6, step=1, label="Precisão de Cor"),
+            gr.Slider(minimum=1, maximum=32, value=16, step=1, label="Diferença de Camada"),
+            gr.Slider(minimum=10, maximum=90, value=60, step=1, label="Limite de Canto"),
+            gr.Slider(minimum=3.5, maximum=10, value=4.0, step=0.5, label="Limite de Comprimento"),
+            gr.Slider(minimum=1, maximum=20, value=10, step=1, label="Máx. Iterações"),
+            gr.Slider(minimum=10, maximum=90, value=45, step=1, label="Limite de Emenda"),
+            gr.Slider(minimum=1, maximum=10, value=8, step=1, label="Precisão do Caminho")
+        ],
+        outputs=[
+            gr.HTML(label="Saída SVG"),
+            gr.File(label="Baixar SVG")
+        ]
+    )
 
-    
-iface.launch()
+iface.launch()