Update app.py

app.py

@@ -174,6 +174,13 @@ def invert_displayed_image(image):
         return Image.fromarray(inverted)
     return inverted
 
+
+MAX_TOKENS = 100
+
+def count_tokens(text):
+    """Count the number of tokens in the text."""
+    return len(text.split())
+
 @spaces.GPU()
 def inference(
     qr_code_content: str,
@@ -207,6 +214,13 @@ def inference(
         if qrcode_image is None and qr_code_content == "":
             raise gr.Error("QR Code Image or QR Code Content is required")
 
+        # Validate the length of the prompt
+        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:
+            raise gr.Error(f"Negative prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
+
         pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
 
         if seed == -1:
@@ -265,7 +279,7 @@ def inference(
                 controlnet_conditioning_scale=float(controlnet_conditioning_scale),
                 generator=generator,
                 strength=float(strength),
-                num_inference_steps=50,
+                num_inference_steps=100,
             )
             final_image = out.images[0] if final_image is None else final_image
 
@@ -354,7 +368,7 @@ with gr.Blocks(theme='Hev832/Applio', css=css) as blocks:
             prompt = gr.Textbox(
                 label="Artistic Prompt",
                 placeholder="Describe the style or theme for your QR code art",
-                value="A high-resolution, photo-realistic minimalist rendering of Mount Fuji, depicted as a sharp, semi-realistic silhouette of a mountain range on the horizon. The mountain evokes strength and motion with clean, crisp lines and a sense of natural flow. The scene should feature detailed snow textures, subtle highlights on the mountain ridges, and a powerful yet serene atmosphere. The rendering should emphasize the strength of the mountain with a focus on clarity and precision in both texture and light. (Sharp outlines:1.5), (Photo-realistic:1.4), (Detailed textures:1.3), (Minimalist:1.3), (Semi-realistic:1.3), (Monochrome contrast:1.2), (Crisp detail:1.2), (Evoking strength:1.2), inspired by traditional Japanese woodblock prints, nature photography, and minimalist design principles.",
+                value="A high-res, photo-realistic minimalist rendering of Mount Fuji as a sharp, semi-realistic silhouette on the horizon. The mountain conveys strength and motion with clean, crisp lines and natural flow. Features detailed snow textures, subtle ridge highlights, and a powerful yet serene atmosphere. Emphasizes strength with clarity and precision in texture and light. (Sharp outlines:1.5), (Photo-realistic:1.4), (Detailed textures:1.3), (Minimalist:1.3), (Semi-realistic:1.3), (Monochrome contrast:1.2), (Crisp detail:1.2), (Evoking strength:1.2). Inspired by traditional Japanese woodblock prints, nature photography, and minimalist design principles.",
                 info="Be specific and creative! This guides the AI in creating your unique QR code art.",
             )
             negative_prompt = gr.Textbox(