feat: Add style prompt input to infer function in app.py

app.py

@@ -4,7 +4,7 @@ from gradio_client import Client
 
 
 
-def infer (prompt, inf_steps, guidance_scale, width, height, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):    
+def infer (prompt, style_prompt, inf_steps, guidance_scale, width, height, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):    
     custom_model="lichorosario/dott_remastered_style_lora_sdxl"
     weight_name="dott_style.safetensors"
 
@@ -18,7 +18,7 @@ def infer (prompt, inf_steps, guidance_scale, width, height, seed, lora_weight,
     result = client.predict(
 		custom_model=custom_model,
 		weight_name=weight_name,
-		prompt="dott style. "+prompt,
+		prompt="dott style. "+prompt+". "+style_prompt,
 		inf_steps=inf_steps,
 		guidance_scale=guidance_scale,
 		width=width,
@@ -76,6 +76,9 @@ with gr.Blocks(css=css) as demo:
           label="Your Prompt",
           info = "Dont' forget to include your trigger word if necessary"
         )   
+        style_prompt_in = gr.Textbox(
+          label="Your Style Prompt"
+        )   
         with gr.Accordion("Advanced Settings", open=False):
             with gr.Row():
                 inf_steps = gr.Slider(
@@ -133,7 +136,7 @@ with gr.Blocks(css=css) as demo:
 
     submit_btn.click(
         fn = infer,
-        inputs = [prompt_in, inf_steps, guidance_scale, width, height, seed, lora_weight],
+        inputs = [prompt_in, style_prompt_in, inf_steps, guidance_scale, width, height, seed, lora_weight],
         outputs = [image_out, last_used_seed]
     )