JaishnaCodz commited on
Commit
16e02f2
·
verified ·
1 Parent(s): 02f79df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -95,12 +95,17 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as app:
95
  outputs.extend(["", "", False]) # clear unused slots
96
  return outputs
97
 
98
- def collect_decisions(origs, suggs, accepts):
 
 
 
 
99
  return finalize_text(origs, suggs, accepts)
100
 
 
101
  # Wire actions
102
  extract_btn.click(fn=extract_both, inputs=[blog_url, image_url], outputs=combined_text)
103
  review_btn.click(fn=process_review, inputs=combined_text, outputs=[el for group in line_outputs for el in group])
104
- finalize_btn.click(fn=collect_decisions, inputs=[originals, suggestions, decisions], outputs=final_output)
105
 
106
  app.launch()
 
95
  outputs.extend(["", "", False]) # clear unused slots
96
  return outputs
97
 
98
+ def collect_decisions(*args):
99
+ count = len(args) // 3
100
+ origs = args[:count]
101
+ suggs = args[count:2*count]
102
+ accepts = args[2*count:]
103
  return finalize_text(origs, suggs, accepts)
104
 
105
+
106
  # Wire actions
107
  extract_btn.click(fn=extract_both, inputs=[blog_url, image_url], outputs=combined_text)
108
  review_btn.click(fn=process_review, inputs=combined_text, outputs=[el for group in line_outputs for el in group])
109
+ finalize_btn.click(fn=collect_decisions, inputs=originals + suggestions + decisions, outputs=final_output)
110
 
111
  app.launch()