Update app.py

app.py

@@ -4,39 +4,24 @@ from PIL import Image
 import vtracer
 import tempfile
 
-# Localization dictionary
 TRANSLATIONS = {
     'en': {
         'title': 'Convert Image to SVG Vectors',
-        'description': 'Upload an image and customize the conversion parameters as needed.',
-        'upload_label': 'Upload Image',
-        'convert_btn': 'Convert to SVG',
-        'svg_output': 'SVG Output',
-        'download_svg': 'Download SVG'
+        'description': 'Upload an image and customize the conversion parameters as needed.'
     },
     'de': {
         'title': 'Bild in SVG-Vektoren umwandeln',
-        'description': 'Laden Sie ein Bild hoch und passen Sie die Konvertierungsparameter nach Bedarf an.',
-        'upload_label': 'Bild hochladen',
-        'convert_btn': 'In SVG umwandeln',
-        'svg_output': 'SVG-Ausgabe',
-        'download_svg': 'SVG herunterladen'
+        'description': 'Laden Sie ein Bild hoch und passen Sie die Konvertierungsparameter nach Bedarf an.'
     }
 }
 
 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."""
-    if image is None:
-        return None, None
-        
-    # Convert Gradio image to bytes for vtracer compatibility
+ color_precision, layer_difference, corner_threshold,
+ length_threshold, max_iterations, splice_threshold, path_precision, language):
     img_byte_array = io.BytesIO()
     image.save(img_byte_array, format='PNG')
     img_bytes = img_byte_array.getvalue()
 
-    # Perform the conversion
     svg_str = vtracer.convert_raw_image_to_svg(
         img_bytes,
         img_format='png',
@@ -53,30 +38,26 @@ def convert_image(image, color_mode, hierarchical, mode, filter_speckle,
         path_precision=int(path_precision)
     )
 
-    # 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()
 
     return (
-        f'<svg viewBox="0 0 {image.width} {image.height}" style="max-width: 100%; height: auto;">{svg_str}</svg>', 
+        gr.HTML(f'<svg viewBox="0 0 {image.width} {image.height}">{svg_str}</svg>'), 
         temp_file.name
     )
 
-def update_interface(language):
-    """Update interface language"""
-    return (
-        f"# {TRANSLATIONS[language]['title']}",
-        TRANSLATIONS[language]['description'],
-        TRANSLATIONS[language]['upload_label'],
-        TRANSLATIONS[language]['convert_btn'],
-        TRANSLATIONS[language]['svg_output'],
-        TRANSLATIONS[language]['download_svg']
+with gr.Blocks() as vector_converter_interface:
+    language_dropdown = gr.Dropdown(
+        choices=['en', 'de'], 
+        value='en', 
+        label='Language'
     )
 
-# Create the interface using gr.Interface instead of gr.Blocks
-def create_interface():
-    return gr.Interface(
+    title = gr.Textbox(value=TRANSLATIONS['en']['title'], visible=False)
+    description = gr.Textbox(value=TRANSLATIONS['en']['description'], visible=False)
+
+    gr.Interface(
         fn=convert_image,
         inputs=[
             gr.Image(type="pil", label="Upload Image"),
@@ -90,18 +71,24 @@ def create_interface():
             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")
+            gr.Slider(minimum=1, maximum=10, value=8, step=1, label="Path Precision"),
+            language_dropdown
         ],
         outputs=[
             gr.HTML(label="SVG Output"),
             gr.File(label="Download SVG")
         ],
-        title="Convert Image to SVG Vectors / Bild in SVG-Vektoren umwandeln",
-        description="Upload an image and customize the conversion parameters as needed. / Laden Sie ein Bild hoch und passen Sie die Konvertierungsparameter nach Bedarf an."
+        title=TRANSLATIONS['en']['title'],
+        description=TRANSLATIONS['en']['description']
     )
 
-# Create and launch the interface
-vector_converter_interface = create_interface()
+    def update_language(language):
+        return TRANSLATIONS[language]['title'], TRANSLATIONS[language]['description']
+
+    language_dropdown.change(
+        update_language,
+        inputs=language_dropdown,
+        outputs=[title, description]
+    )
 
-if __name__ == "__main__":
-    vector_converter_interface.launch()
+vector_converter_interface.launch()