add exlucded, included words

app.py

@@ -157,8 +157,8 @@ def generate(*data):
     global visible
     print("visible", visible)
 
-    feature, image, garment_type, model, temperature = data[:5]
-    struct_ref = data[5:]
+    feature, image, garment_type, model, temperature, excluded_words, included_words = data[:7]
+    struct_ref = data[7:]
 
     print(f"features {feature},\n image {image},\n garment_type {garment_type},\n model {model},\n temperature {temperature},\n struct_ref {struct_ref}")
 
@@ -193,7 +193,15 @@ def generate(*data):
                                 If the reference text is not empty, write the product description in the tone of voice of the reference text. 
                                 Make sure to output the product description only, do not include any preceeding text like "Here is your product description".
                                 Do not include any part of the reference structure in the output.
-                                \n\n {feature + detected_features} \n{intended_use} \nReference structure: {structure}\n Reference text: {reference}""")]
+                                Do not use any of the excluded words in the output.
+                                Make sure to include all of the included words in the output.
+                                
+                                {feature + detected_features} 
+                                {intended_use} 
+                                Reference structure: {structure}
+                                Reference text: {reference}
+                                Included words: {included_words}
+                                Excluded words: {excluded_words}"""),]
         batch.append(messages)
 
     description = ""
@@ -260,6 +268,8 @@ with gr.Blocks() as demo:
             with gr.Accordion(label="Advanced Options", open=False):
                 model = gr.Dropdown(models, value="gpt-4-turbo", interactive=True, label="Model", visible=True)
                 temperature = gr.Slider(minimum=0., maximum=1.0, value=0., interactive=True, label="Temperature", visible=True)
+                excluded_words = gr.Textbox(label="Excluded words", interactive=True, lines=2)
+                included_words = gr.Textbox(label="Included words", interactive=True, lines=2)
             with gr.Row():
                 submit = gr.Button(value="Submit")
                 # advanced = gr.Button(value="Advanced")
@@ -278,7 +288,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             md_output = gr.Markdown(label="Output", show_label=True)
             json_output = gr.JSON(label="JSON Output")  
-        submit.click(generate, inputs=[feature, image, garment_type, model, temperature, *struct_ref], 
+        submit.click(generate, inputs=[feature, image, garment_type, model, temperature, excluded_words, included_words, *struct_ref], 
                      outputs=[md_output, json_output])
         # advanced.click(show_advanced, inputs=[], outputs=[model, temperature])
         
@@ -300,4 +310,5 @@ def authf(username, password):
 
 
 if __name__ == '__main__':
-    demo.launch(server_name="0.0.0.0", auth=authf)
+    # demo.launch(server_name="0.0.0.0", auth=authf)
+    demo.launch()