Update app.py (#6)

- Update app.py (af059198a3f33fd7b630e95be9308abe0c16a181)

app.py

@@ -59,7 +59,7 @@ def update_selection(selected_state: gr.SelectData, flux_loras):
     trigger_word = flux_loras[selected_state.index]["trigger_word"]
     
     updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo})"
-    new_placeholder = f"optional description, e.g. 'a man with glasses and a beard'"
+    new_placeholder = f"opt - describe the person/subject, e.g. 'a man with glasses and a beard'"
     
     return updated_text, gr.update(placeholder=new_placeholder), selected_state.index
 
@@ -172,17 +172,22 @@ def infer_with_lora(input_image, prompt, selected_index, custom_lora, seed=42, r
     trigger_word = lora_to_use["trigger_word"]
     is_kontext_lora = lora_to_use["lora_type"] == "kontext"
     if not is_kontext_lora:
-        if trigger_word == ", How2Draw":
-            prompt = f"create a How2Draw sketch of the person of the photo {prompt}, maintain the facial identity of the person and general features"
-        elif trigger_word == ", video game screenshot in the style of THSMS":
-            prompt = f"create a video game screenshot in the style of THSMS with the person from the photo, {prompt}. maintain the facial identity of the person and general features"
-        else:
-            prompt = f"convert the style of this portrait photo to {trigger_word} while maintaining the identity of the person. {prompt}. Make sure to maintain the person's facial identity and features, while still changing the overall style to {trigger_word}."
-    else: 
         if portrait_mode:
-            prompt = f"{trigger_word} while maintaining the identity of the person. {prompt}. Make sure to maintain the person's facial identity and features."
+            if trigger_word == ", How2Draw":
+                prompt = f"create a How2Draw sketch of the person of the photo {prompt}, maintain the facial identity of the person and general features"
+            elif trigger_word == ", video game screenshot in the style of THSMS":
+                prompt = f"create a video game screenshot in the style of THSMS with the person from the photo, {prompt}. maintain the facial identity of the person and general features"
+            else:
+                prompt = f"convert the style of this portrait photo to {trigger_word} while maintaining the identity of the person. {prompt}. Make sure to maintain the person's facial identity and features, while still changing the overall style to {trigger_word}."
         else:
-            prompt = f"{trigger_word}. {prompt}."
+           if trigger_word == ", How2Draw":
+                prompt = f"create a How2Draw sketch of the photo {prompt}"
+           elif trigger_word == ", video game screenshot in the style of THSMS":
+                prompt = f"create a video game screenshot in the style of THSMS of the photo, {prompt}."
+           else:
+                prompt = f"convert the style of this photo {prompt} to {trigger_word}." 
+    else: 
+        prompt = f"{trigger_word}. {prompt}."
     try:
         image = pipe(
             image=input_image, 
@@ -239,7 +244,7 @@ with gr.Blocks(css=css) as demo:
     
     title = gr.HTML(
         """<h1> FLUX.1 Kontext Portrait 👩🏻‍🎤
-        <br><small style="font-size: 13px; opacity: 0.75;">Flux.1 Kontext [dev] + community Flux LoRAs 🤗</small></h1>""",
+        <br><small style="font-size: 13px; opacity: 0.75;">Flux.1 Kontext [dev] + community Kontext & Flux LoRAs 🤗</small></h1>""",
     )
     
     selected_state = gr.State(value=None)
@@ -274,7 +279,7 @@ with gr.Blocks(css=css) as demo:
                     show_label=False,
                     lines=1,
                     max_lines=1,
-                    placeholder="optional description, e.g. 'a man with glasses and a beard'",
+                    placeholder="opt - describe the person/subject, e.g. 'a man with glasses and a beard'",
                     elem_id="prompt"
                 )
                 run_button = gr.Button("Generate", elem_id="run_button")