Update app.py

app.py

@@ -123,54 +123,59 @@ def inference(
     use_qr_code_as_init_image = True,
     sampler = "DPM++ Karras SDE",
 ):
-    if prompt is None or prompt == "":
-        raise gr.Error("Prompt is required")
+    try:
+        if prompt is None or prompt == "":
+            raise gr.Error("Prompt is required")
+
+        if qrcode_image is None and qr_code_content == "":
+            raise gr.Error("QR Code Image or QR Code Content is required")
 
-    if qrcode_image is None and qr_code_content == "":
-        raise gr.Error("QR Code Image or QR Code Content is required")
+        pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
 
-    pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
+        generator = torch.manual_seed(seed) if seed != -1 else torch.Generator()
 
-    generator = torch.manual_seed(seed) if seed != -1 else torch.Generator()
+        if qr_code_content != "" or qrcode_image.size == (1, 1):
+            print("Generating QR Code from content")
+            qr = qrcode.QRCode(
+                version=1,
+                error_correction=qrcode.constants.ERROR_CORRECT_H,
+                box_size=10,
+                border=4,
+            )
+            qr.add_data(qr_code_content)
+            qr.make(fit=True)
 
-    if qr_code_content != "" or qrcode_image.size == (1, 1):
-        print("Generating QR Code from content")
-        qr = qrcode.QRCode(
-            version=1,
-            error_correction=qrcode.constants.ERROR_CORRECT_H,
-            box_size=10,
-            border=4,
+            qrcode_image = qr.make_image(fill_color="black", back_color="white")
+            qrcode_image = resize_for_condition_image(qrcode_image, 768)
+        else:
+            print("Using QR Code Image")
+            qrcode_image = resize_for_condition_image(qrcode_image, 768)
+
+        # hack due to gradio examples
+        init_image = qrcode_image
+
+        out = pipe(
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            image=qrcode_image,
+            control_image=qrcode_image,  # type: ignore
+            width=768,  # type: ignore
+            height=768,  # type: ignore
+            guidance_scale=float(guidance_scale),
+            controlnet_conditioning_scale=float(controlnet_conditioning_scale),  # type: ignore
+            generator=generator,
+            strength=float(strength),
+            num_inference_steps=50,
         )
-        qr.add_data(qr_code_content)
-        qr.make(fit=True)
-
-        qrcode_image = qr.make_image(fill_color="black", back_color="white")
-        qrcode_image = resize_for_condition_image(qrcode_image, 768)
-    else:
-        print("Using QR Code Image")
-        qrcode_image = resize_for_condition_image(qrcode_image, 768)
-
-    # hack due to gradio examples
-    init_image = qrcode_image
-
-    out = pipe(
-        prompt=prompt,
-        negative_prompt=negative_prompt,
-        image=qrcode_image,
-        control_image=qrcode_image,  # type: ignore
-        width=768,  # type: ignore
-        height=768,  # type: ignore
-        guidance_scale=float(guidance_scale),
-        controlnet_conditioning_scale=float(controlnet_conditioning_scale),  # type: ignore
-        generator=generator,
-        strength=float(strength),
-        num_inference_steps=50,
-    )
-    return out.images[0]  # type: ignore
+        return out.images[0]  # type: ignore
+    except Exception as e:
+        print(f"Error in inference: {str(e)}")
+        # Return a blank image in case of an error
+        return Image.new('RGB', (768, 768), color='white')
 
 
 
-with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
+with gr.Blocks(theme='Hev832/Applio'()) as blocks:
     gr.Markdown(
         """
         ![Yamamoto Logo](https://cdn-uploads.huggingface.co/production/uploads/64740cf7485a7c8e1bd51ac9/_VyYxp5qE_nRZ_LJqBxmL.webp)