Update with finished gallery label parser

app.py

@@ -219,13 +219,27 @@ def image_prompt(prompt, guidance, steps, seed, height, width, negative_prompt="
   print(f"is_prompt_clean: {is_clean}")
   print("-------------------------------------------------------")
 
-  trimmed_prompt = prompt.replace("<", "").replace(">", "").replace("ahx-beta-", "").replace("ahx-model-", "")
-  # trimmed_prompt_array = trimmed_prompt.split(" ")
+  input_prompt = prompt.replace(">", "").replace("<", "")
+  input_prompt = input_prompt.split(" ")
+
+  tokens = []
+  prompt_words = []
+
+  for word in input_prompt:
+    if "ahx" in word:
+      tokens.append(word.replace("ahx-beta-", "").replace("ahx-model-", ""))
+    else:
+      prompt_words.append(word)
+
+  joined_prompt_text = f"\"{' '.join(prompt_words)}\""
+  file_name = f"ahx-{'-'.join(tokens)}-{seed}.png"
+
+  gallery_label = f"{joined_prompt_text} | {file_name}"
     
   if is_clean:
     return (
       pipe(prompt=prompt, guidance_scale=guidance, num_inference_steps=steps, generator=generator, height=height, width=width, negative_prompt=negative_prompt).images[0], 
-      f"prompt: '{prompt}', seed = {int(seed)},\nheight: {height}px, width: {width}px,\nguidance: {guidance}, steps: {steps}, negative prompt: {negative_prompt}\n\n\n\"{trimmed_prompt}\" | ahx-{seed}.png"
+      f"{gallery_label}\n\nprompt: '{prompt}', seed = {int(seed)},\nheight: {height}px, width: {width}px,\nguidance: {guidance}, steps: {steps}, negative prompt: {negative_prompt}"
     )
   else:
     return (