added buttons to view context

app.py

@@ -65,7 +65,7 @@ def process_query(query):
 
     boolean_context_modified = miniWikiCollectionDict[boolean_ranking_modified[0]]
     tf_idf_context_modified = miniWikiCollectionDict[tf_idf_ranking_modified[0]]
-    bm25_context = miniWikiCollectionDict[str(bm25_ranking_modified[0])]
+    bm25_context_modified = miniWikiCollectionDict[str(bm25_ranking_modified[0])]
     vision_context_modified = miniWikiCollectionDict[vision_ranking_modified[0]]
     open_source_context_modified = miniWikiCollectionDict[open_source_ranking_modified[0]]
 
@@ -88,7 +88,7 @@ def process_query(query):
 
     boolean_answer_modified = generate_answer_withContext(modified_query, boolean_context_modified)
     tf_idf_answer_modified = generate_answer_withContext(modified_query, tf_idf_context_modified)
-    bm25_answer_modified = generate_answer_withContext(modified_query, bm25_context)
+    bm25_answer_modified = generate_answer_withContext(modified_query, bm25_context_modified)
     vision_answer_modified = generate_answer_withContext(modified_query, vision_context_modified)
     open_source_answer_modified = generate_answer_withContext(modified_query, open_source_context_modified)
 
@@ -127,22 +127,22 @@ def process_query(query):
     return (
         best_model,
         best_answer,
-        agent1_answer,
-        agent2_answer,
-        boolean_answer,
-        tf_idf_answer,
-        bm25_answer,
-        vision_answer,
-        open_source_answer,
-        boolean_answer_modified,
-        tf_idf_answer_modified,
-        bm25_answer_modified,
-        vision_answer_modified,
-        open_source_answer_modified,
-        tf_idf_bm25_open_RRF_Ranking_answer,
-        tf_idf_bm25_open_RRF_Ranking_modified_answer,
-        tf_idf_bm25_open_RRF_Ranking_combined_answer,
-        zeroShot,
+        agent1_answer, agent1_context,
+        agent2_answer, agent2_context,
+        boolean_answer, boolean_context,
+        tf_idf_answer, tf_idf_context,
+        bm25_answer, bm25_context,
+        vision_answer, vision_context,
+        open_source_answer, open_source_context,
+        boolean_answer_modified, boolean_context_modified,
+        tf_idf_answer_modified, tf_idf_context_modified,
+        bm25_answer_modified, bm25_context_modified,
+        vision_answer_modified, vision_context_modified,
+        open_source_answer_modified, open_source_context_modified,
+        tf_idf_bm25_open_RRF_Ranking_answer, tf_idf_bm25_open_RRF_Ranking_context,
+        tf_idf_bm25_open_RRF_Ranking_modified_answer, tf_idf_bm25_open_RRF_Ranking_modified_context,
+        tf_idf_bm25_open_RRF_Ranking_combined_answer, tf_idf_bm25_open_RRF_Ranking_combined_context,
+        zeroShot, "Zero-shot doesn't have a context."
     )
 
 # Gradio interface
@@ -155,32 +155,47 @@ def create_interface():
         with gr.Row():
             best_model_output = gr.Textbox(label="Best Model")
             best_answer_output = gr.Textbox(label="Best Answer")
-        
+
         gr.Markdown("---")  # Horizontal line
-        
-        gr.Markdown("## All Answers")
+        gr.Markdown("## All Answers and Contexts")
+
+        def create_answer_row(label, context):
+            with gr.Row():
+                answer_textbox = gr.Textbox(label=f"{label} Answer", interactive=False).style(container=True)
+                context_button = gr.Button(f"Show {label} Context")
+                context_textbox = gr.Textbox(label=f"{label} Context", visible=False).style(container=True)
+
+                context_button.click(
+                    fn=lambda: gr.update(visible=True, value=context),
+                    inputs=[],
+                    outputs=context_textbox,
+                )
+                return answer_textbox, context_textbox
+                
+
         with gr.Row():
-            agent1_output = gr.Textbox(label="Agent 1")
-            agent2_output = gr.Textbox(label="Agent 2")
-            boolean_output = gr.Textbox(label="Boolean")
-            tf_idf_output = gr.Textbox(label="TF-IDF")
-            bm25_output = gr.Textbox(label="BM25")
-        
+            agent1_output, agent1_context_box = create_answer_row("Agent 1", agent1_context)
+            agent2_output, agent2_context_box = create_answer_row("Agent 2", agent_text_context)
+            boolean_output, boolean_context_box = create_answer_row("Boolean", boolean_context)
+            tf_idf_output, tf_idf_context_box = create_answer_row("TF-IDF", tf_idf_context)
+            bm25_output, bm25_context_box = create_answer_row("BM25", bm25_context)
+            vision_output, vision_context_box = create_answer_row("Vision", vision_context)
+            open_source_output, open_source_context_box = create_answer_row("Open Source", open_source_context)
+
         with gr.Row():
-            vision_output = gr.Textbox(label="Vision")
-            open_source_output = gr.Textbox(label="Open Source")
-            boolean_mod_output = gr.Textbox(label="Boolean (Modified)")
-            tf_idf_mod_output = gr.Textbox(label="TF-IDF (Modified)")
-            bm25_mod_output = gr.Textbox(label="BM25 (Modified)")
-        
+            boolean_mod_output, boolean_mod_context_box = create_answer_row("Boolean (Modified)", boolean_context_modified)
+            tf_idf_mod_output, tf_idf_mod_context_box = create_answer_row("TF-IDF (Modified)", tf_idf_context_modified)
+            bm25_mod_output, bm25_mod_context_box = create_answer_row("BM25 (Modified)", bm25_context_modified)
+            vision_mod_output, vision_mod_context_box = create_answer_row("Vision (Modified)", vision_context_modified)
+            open_source_mod_output, open_source_mod_context_box = create_answer_row("Open Source (Modified)", open_source_context_modified)
+
         with gr.Row():
-            vision_mod_output = gr.Textbox(label="Vision (Modified)")
-            open_source_mod_output = gr.Textbox(label="Open Source (Modified)")
-            tf_idf_rrf_output = gr.Textbox(label="TF-IDF + BM25 + Open RRF")
-            tf_idf_rrf_mod_output = gr.Textbox(label="TF-IDF + BM25 + Open RRF (Modified)")
-            tf_idf_rrf_combined_output = gr.Textbox(label="TF-IDF + BM25 + Open RRF (Combined)")
-        
-        zero_shot_output = gr.Textbox(label="Zero Shot")
+            tf_idf_rrf_output, tf_idf_rrf_context_box = create_answer_row("TF-IDF + BM25 + Open RRF", tf_idf_bm25_open_RRF_Ranking_context)
+            tf_idf_rrf_mod_output, tf_idf_rrf_mod_context_box = create_answer_row("TF-IDF + BM25 + Open RRF (Modified)", tf_idf_bm25_open_RRF_Ranking_modified_context)
+            tf_idf_rrf_combined_output, tf_idf_rrf_combined_context_box = create_answer_row("TF-IDF + BM25 + Open RRF (Combined)", tf_idf_bm25_open_RRF_Ranking_combined_context)
+
+        with gr.Row():
+            zero_shot_output, zero_shot_context_box = create_answer_row("Zero Shot", "Zero-shot doesn't have a context.")
 
         gr.Button("Submit").click(
             fn=process_query,
@@ -188,13 +203,22 @@ def create_interface():
             outputs=[
                 best_model_output,
                 best_answer_output,
-                agent1_output, agent2_output,
-                boolean_output, tf_idf_output, bm25_output,
-                vision_output, open_source_output,
-                boolean_mod_output, tf_idf_mod_output, bm25_mod_output,
-                vision_mod_output, open_source_mod_output,
-                tf_idf_rrf_output, tf_idf_rrf_mod_output,
-                tf_idf_rrf_combined_output, zero_shot_output,
+                agent1_output, agent1_context_box,
+                agent2_output, agent2_context_box,
+                boolean_output, boolean_context_box,
+                tf_idf_output, tf_idf_context_box,
+                bm25_output, bm25_context_box,
+                vision_output, vision_context_box,
+                open_source_output, open_source_context_box,
+                boolean_mod_output, boolean_mod_context_box,
+                tf_idf_mod_output, tf_idf_mod_context_box,
+                bm25_mod_output, bm25_mod_context_box,
+                vision_mod_output, vision_mod_context_box,
+                open_source_mod_output, open_source_context_box,
+                tf_idf_rrf_output, tf_idf_rrf_context_box,
+                tf_idf_rrf_mod_output, tf_idf_rrf_mod_context_box,
+                tf_idf_rrf_combined_output, tf_idf_rrf_combined_context_box,
+                zero_shot_output, zero_shot_context_box,
             ]
         )