Update app.py

app.py

@@ -17,6 +17,9 @@ logging.basicConfig(filename='svgmaker.log', level=logging.INFO,
 
 def process_single_image(image_file, stroke_width, fill, stroke, opacity, output_size, output_dir):
     try:
+        if image_file is None:
+            raise ValueError("No image file provided.")
+        
         logging.info(f"Processing image: {image_file.name}")
         with Image.open(image_file.name) as image:
             # Calculate new dimensions maintaining aspect ratio
@@ -46,13 +49,17 @@ def process_single_image(image_file, stroke_width, fill, stroke, opacity, output
             "potrace",
             temp_pgm_path,
             "--svg",
-            "--o", output_svg            
+            "-o", output_svg            
         ], check=True)
 
         # Clean up temporary PGM
         os.unlink(temp_pgm_path)
 
-
+        # Create a base64 encoded version of the SVG for preview
+        with open(output_svg, 'r') as f:
+            svg_content = f.read()
+        svg_base64 = base64.b64encode(svg_content.encode('utf-8')).decode('utf-8')
+        preview = f"data:image/svg+xml;base64,{svg_base64}"
 
         return output_svg, preview
     except Exception as e:
@@ -60,6 +67,9 @@ def process_single_image(image_file, stroke_width, fill, stroke, opacity, output
         return None, None
 
 def vectorize_icons(images, stroke_width, fill, stroke, opacity, output_size, output_dir, progress=gr.Progress()):
+    if images is None or len(images) == 0:
+        raise ValueError("No images were uploaded. Please upload at least one image.")
+
     vectorized_icons = []
     svg_previews = []
     total_images = len(images)
@@ -144,6 +154,9 @@ with gr.Blocks() as app:
             svg_preview = gr.HTML(label="SVG Previews")
             
             def process_and_display(images, stroke_width, fill, stroke, opacity, output_size, output_dir):
+                if images is None or len(images) == 0:
+                    raise ValueError("No images were uploaded. Please upload at least one image.")
+                
                 vectorized_icons, svg_previews = vectorize_icons(images, stroke_width, fill, stroke, opacity, output_size, output_dir)
                 preview_html = create_preview_grid(svg_previews)
                 # Filter out None values from vectorized_icons
@@ -163,4 +176,4 @@ if __name__ == "__main__":
         app.launch()
     except Exception as e:
         logging.error(f"Failed to launch app: {str(e)}")
-        raise
+        raise