Quality tags v1

app.py

@@ -254,6 +254,16 @@ with open('characterfull.txt', 'r') as f:
 def get_random_character():
     return random.choice(characters)
 
+# Function to add quality tags
+def add_quality_tags(prompt, negative_prompt):
+    positive_tags = "score_9, score_8_up, score_7_up, score_6_up, dramatic lighting, "
+    negative_tags = "worst quality, low quality, text, censored, deformed, bad hand, blurry, (watermark), mutated hands, monochrome, "
+    
+    updated_prompt = positive_tags + prompt
+    updated_negative_prompt = negative_tags + negative_prompt
+    
+    return updated_prompt, updated_negative_prompt
+
 if torch.cuda.is_available():
     pipe = load_pipeline(MODEL)
     logger.info("Loaded on Device!")
@@ -282,6 +292,9 @@ with gr.Blocks(css="style.css") as demo:
                 variant="primary", 
                 scale=0
             )
+        with gr.Row():
+            random_character_button = gr.Button("Random Character")
+            add_quality_tags_button = gr.Button("Add quality tags")
         result = gr.Image(
             label="Result", 
             show_label=False
@@ -293,7 +306,7 @@ with gr.Blocks(css="style.css") as demo:
             placeholder="Enter a negative prompt",
             value=""
         )
-        aspect_ratio_selector = gr.Radio(
+    aspect_ratio_selector = gr.Radio(
             label="Aspect Ratio",
             choices=config.aspect_ratios,
             value="1024 x 1024",
@@ -365,9 +378,6 @@ with gr.Blocks(css="style.css") as demo:
         json_input = gr.TextArea(label="Edit/Paste JSON Parameters", placeholder="Paste or edit JSON parameters here")
         generate_from_json = gr.Button("Generate from JSON")
 
-    with gr.Accordion("Randomize", open=False):
-        random_character_button = gr.Button("Random Character")
-
     with gr.Accordion("Generation History", open=False) as history_accordion:
         history_gallery = gr.Gallery(
             label="History",
@@ -469,6 +479,12 @@ with gr.Blocks(css="style.css") as demo:
         outputs=[prompt]
     )
 
+    add_quality_tags_button.click(
+        fn=add_quality_tags,
+        inputs=[prompt, negative_prompt],
+        outputs=[prompt, negative_prompt]
+    )
+
     history_gallery.select(
         fn=handle_image_click,
         inputs=[],