Update app.py

app.py

@@ -234,7 +234,7 @@ def inference(
         if count_tokens(prompt) > MAX_TOKENS:
             raise gr.Error(f"Prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
 
-        if count_tokens(negative_prompt) > MAX_TOKENS:
+        if negative_prompt and count_tokens(negative_prompt) > MAX_TOKENS:
             raise gr.Error(f"Negative prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
 
         pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
@@ -243,7 +243,7 @@ def inference(
             seed = torch.seed()  # Generate a truly random seed
         generator = torch.manual_seed(seed)
 
-        if qr_code_content != "" or qrcode_image.size == (1, 1):
+        if qr_code_content != "" or (qrcode_image is not None and qrcode_image.size == (1, 1)):
             print("Generating QR Code from content")
             qr = qrcode.QRCode(
                 version=1,
@@ -256,9 +256,11 @@ def inference(
 
             qrcode_image = qr.make_image(fill_color=qr_color, back_color=bg_color)
             qrcode_image = resize_for_condition_image(qrcode_image, 1024)
-        else:
+        elif qrcode_image is not None:
             print("Using QR Code Image")
             qrcode_image = resize_for_condition_image(qrcode_image, 1024)
+        else:
+            raise gr.Error("No valid QR code content or image provided")
 
         # Determine which image to use as init_image and control_image
         if use_qr_code_as_init_image: