Update app.py

app.py

@@ -18,13 +18,13 @@ app = FastAPI()
 # Configure CORS to allow requests from Figma plugin
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["https://www.figma.com"],
+    allow_origins=["https://www.figma.com", "*"],  # Allow Figma and local testing
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# Existing vtracer conversion function
+# VTracer conversion function
 def convert_to_vector(
     image,
     colormode="color",
@@ -45,6 +45,7 @@ def convert_to_vector(
     try:
         # Save the input image to a temporary file
         image.save(input_path)
+        logger.info(f"Saved image to {input_path}")
 
         # Convert the image to SVG using VTracer
         vtracer.convert_image_to_svg_py(
@@ -62,21 +63,23 @@ def convert_to_vector(
             splice_threshold=int(splice_threshold),
             path_precision=int(path_precision)
         )
+        logger.info(f"Converted image to SVG at {output_path}")
 
         # Read the SVG output
         with open(output_path, "r") as f:
             svg_content = f.read()
 
         return svg_content
-    # except Exception as e:
-    #     logger.error(f"Error in convert_to_vector: {str(e)}")
-    #     raise HTTPException(status_code=500, detail=f"Conversion failed: {str(e)}")
+    except Exception as e:
+        logger.error(f"Error in convert_to_vector: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Conversion failed: {str(e)}")
     finally:
         # Clean up temporary files
         for path in [input_path, output_path]:
             if os.path.exists(path):
                 try:
                     os.remove(path)
+                    logger.info(f"Removed {path}")
                 except Exception as e:
                     logger.warning(f"Failed to remove {path}: {str(e)}")
 
@@ -97,6 +100,7 @@ async def convert_image(
     path_precision: int = Form(3)
 ):
     try:
+        logger.info("Received request to /convert")
         # Read the uploaded image
         image_data = await file.read()
         image = Image.open(io.BytesIO(image_data))
@@ -112,9 +116,9 @@ async def convert_image(
             layer_difference,
             corner_threshold,
             length_threshold,
-            max_iterations,
-            splice_threshold,
-            path_precision
+            max iterations=max_iterations,
+            splice_threshold=splice_threshold,
+            path_precision=path_precision
         )
 
         return JSONResponse(content={"svg": svg_content})
@@ -122,14 +126,21 @@ async def convert_image(
         logger.error(f"Error in convert_image: {str(e)}")
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
+# Health check endpoint
+@app.get("/")
+async def health_check():
+    logger.info("Health check requested")
+    return {"status": "healthy"}
+
 # Gradio interface
 def handle_color_mode(value):
     return value
 
+# Check if examples directory exists, else use empty list
+examples_dir = "examples"
 examples = [
-    "examples/11.jpg",
-    "examples/02.jpg",
-    "examples/03.jpg",
+    os.path.join(examples_dir, f) for f in ["11.jpg", "02.jpg", "03.jpg"]
+    if os.path.exists(os.path.join(examples_dir, f))
 ]
 
 css = """
@@ -162,39 +173,41 @@ with gr.Blocks(css=css) as gradio_app:
                 image_input = gr.Image(type="pil", label="Upload Image")
                 with gr.Accordion("Advanced Settings", open=False):
                     with gr.Accordion("Clustering", open=False):
-                        colormode = gr.Radio([("COLOR","color"),("B/W", "binary")], value="color", label="Color Mode", show_label=False)
+                        colormode = gr.Radio([("COLOR", "color"), ("B/W", "binary")], value="color", label="Color Mode", show_label=False)
                         filter_speckle = gr.Slider(0, 128, value=4, step=1, label="Filter Speckle", info="Cleaner")
                         color_precision = gr.Slider(1, 8, value=6, step=1, label="Color Precision", info="More accurate")
                         layer_difference = gr.Slider(0, 128, value=16, step=1, label="Gradient Step", info="Less layers")
-                        hierarchical = gr.Radio([("STACKED","stacked"), ("CUTOUT","cutout")], value="stacked", label="Hierarchical Mode",show_label=False)
+                        hierarchical = gr.Radio([("STACKED", "stacked"), ("CUTOUT", "cutout")], value="stacked", label="Hierarchical Mode", show_label=False)
                     with gr.Accordion("Curve Fitting", open=False):
-                        mode = gr.Radio([("SPLINE","spline"),("POLYGON", "polygon"), ("PIXEL","none")], value="spline", label="Mode", show_label=False)
+                        mode = gr.Radio([("SPLINE", "spline"), ("POLYGON", "polygon"), ("PIXEL", "none")], value="spline", label="Mode", show_label=False)
                         corner_threshold = gr.Slider(0, 180, value=60, step=1, label="Corner Threshold", info="Smoother")
-                        length_threshold = gr.Slider(3.5, 10, value=4.0, step=0.1, label="Segment Length", info ="More coarse")
+                        length_threshold = gr.Slider(3.5, 10, value=4.0, step=0.1, label="Segment Length", info="More coarse")
                         splice_threshold = gr.Slider(0, 180, value=45, step=1, label="Splice Threshold", info="Less accurate")
                         max_iterations = gr.Slider(1, 20, value=10, step=1, label="Max Iterations", visible=False)
                         path_precision = gr.Slider(1, 10, value=3, step=1, label="Path Precision", visible=False)
                         output_text = gr.Textbox(label="Selected Mode", visible=False)
                 with gr.Row():
                     clear_button = gr.Button("Clear")
-                    convert_button = gr.Button("✨ Convert to SVG", variant='primary', elem_classes=["generate-btn"])
+                    convert_button = gr.Button("✨ Convert to SVG", variant="primary", elem_classes=["generate-btn"])
 
             with gr.Column():
                 html = gr.HTML(label="SVG Output")
                 svg_output = gr.File(label="Download SVG")
-        
-        gr.Examples(
-            examples=examples,
-            fn=convert_to_vector,
-            inputs=[image_input],
-            outputs=[html, svg_output],
-            cache_examples=False,
-            run_on_click=True
-        )
+
+        if examples:
+            gr.Examples(
+                examples=examples,
+                fn=convert_to_vector,
+                inputs=[image_input],
+                outputs=[html, svg_output],
+                cache_examples=False,
+                run_on_click=True
+            )
 
     colormode.change(handle_color_mode, inputs=colormode, outputs=output_text)
     hierarchical.change(handle_color_mode, inputs=hierarchical, outputs=output_text)
     mode.change(handle_color_mode, inputs=mode, outputs=output_text)
+
     default_values = {
         "color_precision": 6,
         "layer_difference": 16
@@ -255,11 +268,10 @@ with gr.Blocks(css=css) as gradio_app:
         outputs=[html, svg_output]
     )
 
-# Mount Gradio app to FastAPI at a subpath
-from gradio import mount_gradio_app
-app = mount_gradio_app(app, gradio_app, path="/gradio")
-
-# Health check endpoint
-@app.get("/")
-async def health_check():
-    return {"status": "healthy"}
+# Mount Gradio app to FastAPI at /gradio
+try:
+    from gradio import mount_gradio_app
+    app = mount_gradio_app(app, gradio_app, path="/gradio")
+    logger.info("Gradio app mounted successfully at /gradio")
+except Exception as e:
+    logger.error(f"Failed to mount Gradio app: {str(e)}")