Update app.py

app.py

@@ -60,6 +60,39 @@ def reject(prompt, negative_prompt, seed, randomize_seed, width, height, guidanc
     
     return image
 
+
+@spaces.GPU
+def accept(*args):
+    
+    conv_id = args[0]
+    '''
+    markdown_blocks = list(args[1:-2])
+    labels = args[-2]
+
+    data = {'conv_id': conv_id, 'labels': labels}
+    with open(f"./labels/conv_{conv_id}", 'w') as f:
+        json.dump(data, f, indent=4)
+
+    new_conversation = get_conversation()
+    new_conv_id = new_conversation['conv_id']
+    new_transcript = pad_transcript(new_conversation['transcript'], max_conversation_length)
+
+    for i in range(max_conversation_length):
+        if new_transcript[i]['speaker'] != '':
+            markdown_blocks[i] = gr.Markdown(f"""<br>**{new_transcript[i]['speaker']}**: {new_transcript[i]['response']}""",
+                                             visible=True)
+        else:
+            markdown_blocks[i] = gr.Markdown("", visible=False)
+
+    new_labels = gr.CheckboxGroup(choices=checkbox_choices,
+                                  label="",
+                                  visible=False)
+    conv_len = gr.Number(value=len(new_transcript), visible=False)
+
+    return [new_conv_id] + list(markdown_blocks) + [labels] + [new_labels] + [conv_len]
+    '''
+ 
+
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
     "An astronaut riding a green horse",
@@ -100,6 +133,7 @@ with gr.Blocks(css=css) as demo:
             right_button = gr.Button("Right", scale=0)
         
         result = gr.Image(label="Result", show_label=False)
+        conv_id_element = gr.Text(value=1, visible=False)
 
         with gr.Accordion("Advanced Settings", open=False):
             
@@ -161,12 +195,6 @@ with gr.Blocks(css=css) as demo:
             inputs = [prompt]
         )
 
-    run_button.click(
-        fn = infer,
-        inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs = [result]
-    )
-
     left_button.click(
         fn = reject,
         inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
@@ -174,9 +202,9 @@ with gr.Blocks(css=css) as demo:
     )
 
     right_button.click(
-        fn = reject,
+        fn = accept,
         inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs = [result]        
+        outputs = [conv_id_element]        
     )
 
 demo.queue().launch()