Generation workflow

app.py

@@ -75,7 +75,7 @@ sample_dict = {
 
 user_id = "NO_ID"
 
-with gr.Blocks(theme=gr.themes.Soft()) as user_eval:
+with gr.Blocks() as user_eval:
     # Title text introducing study
     gr.Markdown("""
     # Legal Retriever Evaluation Study
@@ -129,6 +129,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as user_eval:
             step += 1
             print(eval_1 + eval_2 + eval_3)
             if step == len(passage_texts):
+                # Step 10: all sources 
                 collapsible_string = ""
                 for i, passage in enumerate(passage_texts):
                     collapsible_string += """
@@ -148,10 +149,35 @@ with gr.Blocks(theme=gr.themes.Soft()) as user_eval:
                 }
         
         def next_g(eval_helps, eval_satisfied): 
+            global step 
+            step += 1
             print(eval_helps + eval_satisfied)
-            return {
-                selection: gr.Markdown("next summary")
-            }
+            if step == 11:
+                # Step 11: guaranteed summary
+                return {
+                    selection: gr.HTML("""
+                        <h2> Autogenerated Response </h2> <br />
+                        """ + generation)
+                }
+            # Steps 12 and 13 are gold passage + gold passage generation IF it is applicable
+            if step > 11 and not sample_dict["top10_contains_gold_passage"]: 
+                if step == 12: 
+                    return {
+                    selection: gr.HTML("""
+                        <h2> Retrieved Passage </h2> <br />
+                        """ + gold_passage)
+                    }
+                else: 
+                   return {
+                    selection: gr.HTML("""
+                        <h2> Autogenerated Response </h2> <br />
+                        """ + gold_passage_generation)
+                    }
+            else: 
+                return {
+                    selection: gr.HTML("TOGGLE INCOMING")
+                }
+
 
         btn_p.click(fn = next_p, inputs = [eval_1, eval_2, eval_3], outputs = [selection, scores_p, scores_g])
         btn_g.click(fn = next_g, inputs = [eval_helps, eval_satisfied], outputs = [selection])