Update app.py

app.py

@@ -8,16 +8,32 @@ import time
 
 # Defining the repository information and the trigger word
 repo = "artificialguybr/LineAniRedmond-LinearMangaSDXL-V2"
-trigger_word = "lineart,LineAniAF,  "
+trigger_word = "lineart,LineAniAF,"
+
+# Hard set prompt template
+hard_set_prompt = (
+    "minimalist black and white single-line art illustration of [subject], created in one continuous, unbroken stroke from start to finish. "
+    "The design uses true black and true white values only, with no gradients or gray tones. The artwork emphasizes clean, connected, and flowing lines, "
+    "resulting in a sleek, modern aesthetic. The [subject] is depicted in a dynamic and recognizable pose, optimized for artistic purposes and ready for 3D printing applications."
+)
+
+constraints = (
+    "MUST: be only black and white, must be single-line art, must never use gray values/shading/tint or hue of any kind, only true black and true white. "
+    "NEVER use anything besides a single black line of varying widths as necessary to achieve a result of the user's request."
+)
 
 def generate_image(prompt):
     print("Generating image with prompt:", prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
-    #token = os.getenv("API_TOKEN")  # Uncomment and use your Hugging Face API token
+    # token = os.getenv("API_TOKEN")  # Uncomment and use your Hugging Face API token
     headers = {
-        #"Authorization": f"Bearer {token}"
+        # "Authorization": f"Bearer {token}"
     }
-    full_prompt = f"{prompt} {trigger_word}"
+    
+    # Incorporate the hard set prompt and constraints
+    full_prompt = (
+        f"{hard_set_prompt.replace('[subject]', prompt)} {constraints} {trigger_word}"
+    )
     payload = {
         "inputs": full_prompt,
         "parameters": {
@@ -35,7 +51,7 @@ def generate_image(prompt):
         print("API response status code:", response.status_code)
         if response.status_code == 200:
             print("Image generation successful!")
-            return Image.open(BytesIO(response.content))  # Changed to match the first code
+            return Image.open(BytesIO(response.content))
         elif response.status_code == 503:
             time.sleep(1)
             pbar.update(1)
@@ -48,11 +64,20 @@ def generate_image(prompt):
 
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(lines=2, placeholder="Type your prompt here..."),
+    inputs=gr.Textbox(lines=2, placeholder="Describe the subject here..."),
     outputs="image",
     title="LineArt XL Image Generator",
-    description="Powered by the generous GPU time from Redmond.AI, this LORA, fine-tuned on SD XL 1.0, excels at creating Lineart-themed images across a wide range of subjects. Optimized for 1024x1024 resolution, it incorporates the specific tag 'lineart,LineAniAF' directly in the HF Space for ease of use. If you appreciate this model and wish to support, consider a donation via [Patreon](https://www.patreon.com/user?u=81570187) or [Ko-fi](https://ko-fi.com/artificialguybr). Stay updated on new models by following on [Twitter](https://twitter.com/artificialguybr).",
-    examples=[["1girl, red hair"], ["1 boy, wearing sunglasses"]]
+    description=(
+        "Powered by the generous GPU time from Redmond.AI, this LORA, fine-tuned on SD XL 1.0, excels at creating Lineart-themed images across a wide range of subjects. "
+        "Optimized for 1024x1024 resolution, it incorporates the specific tag 'lineart,LineAniAF' directly in the HF Space for ease of use. "
+        "If you appreciate this model and wish to support, consider a donation via [Patreon](https://www.patreon.com/user?u=81570187) or [Ko-fi](https://ko-fi.com/artificialguybr). "
+        "Stay updated on new models by following on [Twitter](https://twitter.com/artificialguybr)."
+    ),
+    examples=[
+        ["a soaring eagle"], 
+        ["a running cheetah"], 
+        ["a majestic oak tree"]
+    ]
 )
 
 print("Launching Gradio interface...")