Update app.py

app.py

@@ -1,18 +1,29 @@
 import gradio as gr
+import numpy as np
+import matplotlib.pyplot as plt
+import logging
 
-def display_sketch(sketch):
-    return sketch
+logging.basicConfig(level=logging.DEBUG)
 
-# Create a sketchpad interface
-sketchpad = gr.Sketchpad(label="Draw Something")
-output_image = gr.Image(label="Your Sketch")
+def display_sketch(sketch):
+    try:
+        image_data = np.array(sketch)  # Convert the dict data to a numpy array
+        logging.debug(f"Image data type: {type(image_data)}")  # Log the type of image data
+        logging.debug(f"Image data shape: {image_data.shape}")  # Log the shape of image data
+        
+        # Plot the image
+        fig, ax = plt.subplots()
+        ax.imshow(image_data, cmap='gray')
+        plt.axis('off')
+        plt.savefig("/mnt/data/output.png", bbox_inches='tight')
+        return "/mnt/data/output.png"
+    except Exception as e:
+        logging.error(f"Error processing sketch: {e}")
+        return str(e)
 
-# Define the interface
-interface = gr.Interface(
-    fn=display_sketch,
-    inputs=sketchpad,
-    outputs=output_image
-)
+with gr.Blocks() as demo:
+    sketchpad = gr.Sketchpad(label="Draw Something", tool="sketch")
+    output_image = gr.Image(label="Your Sketch")
+    sketchpad.submit(display_sketch, inputs=sketchpad, outputs=output_image)
 
-# Launch the interface
-interface.launch()
+demo.launch()