Add remove output

app.py

@@ -106,8 +106,8 @@ def detect_features(image_paths, garment_type, language="English"):
         }
 
         payload = {
-        # "model": "gpt-4o",
-        "model": "gpt-4-turbo",
+        "model": "gpt-4o",
+        # "model": "gpt-4-turbo",
         "messages": [
             {
             "role": "user",
@@ -120,6 +120,7 @@ def detect_features(image_paths, garment_type, language="English"):
                 Make sure to output the alt text in %s language.
                 If the photo does not contain a garment, return 'No garment detected'.
                 If the photo contains a garment, return the result in in the following JSON format without any preceding or trailing text: 
+
                 {
                     "features": [list of comma separated features], 
                     "intended_use": [list of comma separated intended uses], 
@@ -184,7 +185,7 @@ def generate(*data):
         messages = [
             SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites."""),
             HumanMessage(content=f"""Write a product description with the following features.
-                                Make sure that the description follows the structure of the reference structure. The first line in the reference structure include the document type and the language code. The next lines define the structure of the document to be generated.
+                                Make sure that the description follows the structure of the reference structure. 
                                 Make sure to use markdown format for the output.
                                 Make sure that the entire output is written entirely in language defined in the reference structure. 
                                 Use language that is suitable for the type of document specified in the reference structure. 
@@ -212,7 +213,7 @@ def generate(*data):
     return result_md, result_json
 
 
-visible = 1
+visible = 0
 def add_output_click(*struct_ref):
     global visible
     print("Adding output ", visible)
@@ -227,6 +228,19 @@ def add_output_click(*struct_ref):
     return struct_ref
 
 
+def remove_output_click(*struct_ref):
+    global visible
+    print("Removing output", visible)
+    visible -= 1
+    structure_texts = struct_ref[::2]
+    reference_texts = struct_ref[1::2]
+    structures = [gr.Textbox(label=f"Structure {i}", lines=10, value=structure_texts[i] if i <= visible else "", interactive=True, visible=i <= visible) for i in range(10)]
+    references = [gr.Textbox(label=f"Reference copy {i}", lines=3, value=reference_texts[i] if i <= visible else "", interactive=True, visible=i <= visible) for i in range(10)]
+
+    struct_ref = [val for pair in zip(structures, references) for val in pair]
+    return struct_ref
+
+
 def show_advanced(model, temperature):
     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)
@@ -251,9 +265,12 @@ with gr.Blocks() as demo:
                           [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)]]
             struct_ref[0].value = structure_text_1
-            struct_ref[2].value = structure_text_2
-            add_output = gr.Button(value="Add Output")
-            add_output.click(add_output_click, inputs=struct_ref, outputs=struct_ref)
+            # struct_ref[2].value = structure_text_2
+            with gr.Row():
+                add_output = gr.Button(value="Add Output")
+                remove_output = gr.Button(value="Remove Output")
+                add_output.click(add_output_click, inputs=struct_ref, outputs=struct_ref)
+                remove_output.click(remove_output_click, inputs=struct_ref, outputs=struct_ref)
         with gr.Column():
             md_output = gr.Markdown(label="Output", show_label=True)
             json_output = gr.JSON(label="JSON Output")