add number of versions slider

app.py

@@ -266,8 +266,8 @@ def generate(*data):
     global visible
     print("visible", visible)
     
-    nargs = 9 
-    feature, image, garment_type, model, temperature, excluded_words, included_words, glossary_upload, debug = data[:nargs]
+    nargs = 10 
+    feature, image, garment_type, model, temperature, nversions, excluded_words, included_words, glossary_upload, debug = data[:nargs]
     struct_ref = data[nargs:]
 
     print(f"{feature=}")
@@ -323,8 +323,9 @@ def generate(*data):
                 print("Using both copy and structure")
                 messages = [
                     SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
-                    HumanMessage(content=f"""Generate 5 versions of the product description for a product with the following information.
+                    HumanMessage(content=f"""Generate {nversions} versions of the product description for a product with the following information.
 Write in a way that target the customer.                          
+Use feedbacks from the previous versions to improve the quality of the next versions.         
 Make sure that the structure of each output follows the reference structure.
 Make sure to use the tone of voice, rythm, cadence and style of the reference copy for each output.
 Use markdown format for each output.
@@ -378,8 +379,9 @@ Excluded words: {excluded_words}"""),]
                 print("Using copy")
                 messages = [
                     SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
-                    HumanMessage(content=f"""Generate 5 versions of the product description for a product with the following information.
-Write in a way that target the customer.                                
+                    HumanMessage(content=f"""Generate {nversions} versions of the product description for a product with the following information.
+Write in a way that target the customer.                            
+Use feedbacks from the previous versions to improve the quality of the next versions.                                     
 Make sure that the structure of each output follows the structure of the reference copy.
 Make sure to use the tone of voice, rythm, cadence and style of the reference copy for each output.
 Use markdown format for each output.
@@ -434,8 +436,9 @@ Excluded words: {excluded_words}"""),]
                 print("Using structure")
                 messages = [
                     SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
-                    HumanMessage(content=f"""Generate 5 versions of the product description for a product with the following information.
-Write in a way that target the customer.                                 
+                    HumanMessage(content=f"""Generate {nversions} versions of the product description for a product with the following information.
+Write in a way that target the customer.  
+Use feedbacks from the previous versions to improve the quality of the next versions.                                                                         
 Make sure that the structure of each output follows the reference structure.
 Use markdown format for each output.
 Do not include any part of the reference structure in the output.
@@ -579,6 +582,7 @@ with gr.Blocks() as demo:
             with gr.Accordion(label="Advanced Options", open=False):
                 model = gr.Dropdown(models, value="claude-3-5-sonnet-20240620", interactive=True, label="Model", visible=True)
                 temperature = gr.Slider(minimum=0., maximum=1.0, value=0., interactive=True, label="Temperature", visible=True)
+                nversions = gr.Slider(minimum=1, maximum=10, value=1, step=int, interactive=True, label="Number of versions", visible=True)
                 excluded_words = gr.Textbox(label="Excluded words", interactive=True, lines=2)
                 included_words = gr.Textbox(label="Included words", interactive=True, lines=2)
                 # glossary = gr.Dataframe(row_count = (2, "dynamic"), col_count=(2,"static"), headers=["Description", "Way of writing"], label="Glossary", interactive=True)
@@ -603,7 +607,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, 
+        submit.click(generate, inputs=[feature, image, garment_type, model, temperature, nversions,
                                        excluded_words, included_words, glossary_upload, debug, *struct_ref], 
                      outputs=[md_output, json_output])
         # advanced.click(show_advanced, inputs=[], outputs=[model, temperature])