fix file upload (#6)

- fix file upload (54f1b6147198f6e7641611d13c40115477dac11d)


Co-authored-by: Radamés Ajna <[email protected]>

app.py

@@ -54,7 +54,8 @@ fetch(url)
 get_js_image = """
 async (image_in_img, prompt, image_file_live_opt, live_conditioning) => {
   const canvasEl = document.getElementById("canvas-root");
-  return [image_in_img, prompt, image_file_live_opt, canvasEl._data]
+  const imageData = canvasEl? canvasEl._data : null;
+  return [image_in_img, prompt, image_file_live_opt, imageData]
 }
 """
 
@@ -145,14 +146,15 @@ def generate_images(image_in_img, prompt, image_file_live_opt='file', live_condi
     if image_in_img is None and 'image' not in live_conditioning:
         raise gr.Error("Please provide an image")
 
-    if image_file_live_opt == 'file':
-      conditioning = get_conditioning(image_in_img)
-    elif image_file_live_opt == 'webcam':
-      base64_img = live_conditioning['image']
-      image_data = base64.b64decode(base64_img.split(',')[1])
-      conditioning = Image.open(BytesIO(image_data)).convert('RGB').resize((512,512))
-
     try:
+        if image_file_live_opt == 'file':
+            conditioning = get_conditioning(image_in_img)
+        elif image_file_live_opt == 'webcam':
+            base64_img = live_conditioning['image']
+            image_data = base64.b64decode(base64_img.split(',')[1])
+            conditioning = Image.open(BytesIO(image_data)).convert(
+                'RGB').resize((512, 512))
+
         output = pipe(
             prompt,
             conditioning,
@@ -164,7 +166,6 @@ def generate_images(image_in_img, prompt, image_file_live_opt='file', live_condi
     except Exception as e:
         raise gr.Error(str(e))
 
-
 def toggle(choice):
     if choice == "file":
         return gr.update(visible=True, value=None), gr.update(visible=False, value=None)