Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 
 import gradio as gr
 import requests
-# from openai import OpenAI
+from openai import OpenAI
 from func_timeout import FunctionTimedOut, func_timeout
 from tqdm import tqdm
 
@@ -402,9 +402,14 @@ def img_to_code_img(sketchpad_img):
     #     img.save(f)
 
     # img = sketchpad_img['layers'][0]
-    img = sketchpad_img['composite']
-    image_array = np.array(img)
-    image_array = image_array[:,:,0]
+    if np.all(sketchpad_img['background']==0):
+        img = sketchpad_img['layers'][0]
+        image_array = np.array(img)
+        image_array = 255 - image_array[:,:,3]
+    else:
+        img = sketchpad_img['composite']
+        image_array = np.array(img)
+        image_array = image_array[:,:,0]
     # image_array = 255 - image_array[:,:,3]
 
     # height, width = image_array.shape
@@ -466,6 +471,9 @@ def main():
         for file in files:
             if file.endswith(".png"):
                 example_input_images.append(os.path.join(root, file))
+    SHOWN_SAMPLES = [58, 63, 73, 82, 90, 98]
+    shown_example_input_images = [f"demo_example/img_{i}.png" for i in SHOWN_SAMPLES]
+    rest_of_example_images = [path for path in example_input_images if path not in shown_example_input_images]
 
     canvas = gr.Sketchpad(canvas_size=(512,512), brush=Brush(colors=["black"], default_size=2, color_mode='fixed'))
     with gr.Blocks(theme=theme) as demo:
@@ -478,8 +486,10 @@ def main():
             with gr.Column(scale=1):
                 canvas.render()
                 submit_button = gr.Button("Generate Programs")
+                with gr.Accordion("More Examples"):
+                    gr.Examples(rest_of_example_images, inputs=canvas, label='Example Input')
             with gr.Column(min_width=60):
-                gr.Examples(example_input_images, inputs=canvas, label='Example Input')
+                gr.Examples(shown_example_input_images, inputs=canvas, label='Example Input')
             with gr.Column(scale=6):
                 output_gallery = gr.Gallery(
         label="Generated Images", show_label=True, elem_id="gallery"