Add guidance prompt as a post-processing step

app.py

@@ -77,6 +77,9 @@ structure_text_1 = """[type: UK website, style=true, language=English]
 structure_text_2 = """[type: Japanese newsletter, style=true, language=Japanese]
 {{ introduction_paragraph | motivational, passionate, 3-6 sentences }}"""
 
+guidance_prompt_text = """If it is made of 100% cotton, say "pure cotton" instead.
+If it is lightweight, say "featherlight" instead."""
+
 languages = ["American English",
              "British English",
              "German",
@@ -236,11 +239,24 @@ def get_language(struct_lang, copy_lang):
     return "English"
 
 
-def post_process(text):
-    # text = text.strip().replace('<br>', '  ')
-    # with open(f"postprocess_{time.time()}.txt", "w") as f:
-    #     f.write(text)
-    # return text
+def post_process(text: str, guidance_prompt: str, chat: ChatOpenAI):
+    messages = [
+        SystemMessage(content=f"""You are a helpful assistant that edit documents based on the guidlines provided. Make sure to write in English language."""),
+        HumanMessage(content=f"""Given the following product description, your task is to
+make minimal modification to the product description such that the resulting description 
+follows the rules defined in the guidlines. Make sure to preserve the structure of the 
+original text as much as possible. Do not modify the structure of the original text.
+Output only the modified text in markdown format.
+
+Text: 
+{text}
+
+Guildlines: 
+{guidance_prompt}""")
+    ]
+
+    response = chat.invoke(messages, temperature=0.0)
+    text = response.content
     return text  
 
 
@@ -306,8 +322,8 @@ def generate(*data):
     global visible
     print("visible", visible)
     
-    nargs = 10 
-    feature, image, garment_type, model, temperature, nversions, excluded_words, included_words, glossary_upload, debug = data[:nargs]
+    nargs = 11 
+    feature, image, garment_type, model, temperature, nversions, excluded_words, included_words, glossary_upload, debug, guidance_prompt = data[:nargs]
     struct_ref = data[nargs:]
 
     print(f"{feature=}")
@@ -572,7 +588,7 @@ Excluded words: {excluded_words}"""),]
         bestid = jr["best_version"]["id"]
         for d in jr["versions"]:
             if d["id"] == bestid:
-                bestd = post_process(d["content"]) + (f"\n\nDebug info:\n\nScore: {d['score']}\n\nExplanation: {jr['best_version']['explanation']}" if debug else "")
+                bestd = d["content"] + (f"\n\nDebug info:\n\nScore: {d['score']}\n\nExplanation: {jr['best_version']['explanation']}" if debug else "")
                 bests  = d["score"]
                 break
         # bests = 0
@@ -603,7 +619,15 @@ Excluded words: {excluded_words}"""),]
 
     alt_text_dict = {k[0]: v for (k, v) in zip(image, alt_texts)} if len(alt_texts) > 0 else {}
     result_json = {"outputs": jresponse if debug else descriptions, "alt_text": alt_text_dict}
-    result_md = md_content + alt_texts_str + '\n'.join([f'![Product photo](data:image/png;base64,{base64_image} "{alt_text}")' if base64_image != "" else "" for (base64_image, alt_text) in zip(base64_images, alt_texts)])
+
+    post_content = post_process(text=md_content, guidance_prompt=guidance_prompt, chat=chat)
+
+    result_md = "Original\n\n" + md_content + "\n\n---\n\nModified\n\n" + post_content + "\n\n" \
+                    + alt_texts_str \
+                    + '\n'.join([f'![Product photo](data:image/png;base64,{base64_image} "{alt_text}")' if base64_image != "" else "" for (base64_image, alt_text) in zip(base64_images, alt_texts)])
+
+    
+
     return result_md, result_json
 
 
@@ -668,6 +692,7 @@ with gr.Blocks() as demo:
             
         with gr.Column():
             visible = 0
+            guidance_prompt = gr.Textbox(label="Guidance Prompt", value=guidance_prompt_text, lines=3, interactive=True)
             struct_ref = [val for i in range(10) for val in 
                           [gr.Textbox(label=f"Structure {i}", lines=10, value="", interactive=True, visible=i <= visible), 
                            gr.Textbox(label=f"Reference copy {i}", lines=3, value="", interactive=True, visible=i <= visible)]]
@@ -683,7 +708,7 @@ with gr.Blocks() as demo:
             md_output = gr.Markdown(label="Output", show_label=True, line_breaks=True)
             json_output = gr.JSON(label="JSON Output")  
         submit.click(generate, inputs=[feature, image, garment_type, model, temperature, nversions,
-                                       excluded_words, included_words, glossary_upload, debug, *struct_ref], 
+                                       excluded_words, included_words, glossary_upload, debug, guidance_prompt, *struct_ref], 
                      outputs=[md_output, json_output])
         # advanced.click(show_advanced, inputs=[], outputs=[model, temperature])