Update app.py

app.py

@@ -51,9 +51,6 @@ def process_query(query):
         tf_idf_ranking_modified, bm25_ranking_modified, open_source_ranking_modified
     )
 
-
-
-
     agent1_context = wiki_data[0]
     agent2_context = article
 
@@ -73,10 +70,7 @@ def process_query(query):
     tf_idf_bm25_open_RRF_Ranking_modified_context = miniWikiCollectionDict[tf_idf_bm25_open_RRF_Ranking_modified[0]]
     tf_idf_bm25_open_RRF_Ranking_combined_context = miniWikiCollectionDict[tf_idf_bm25_open_RRF_Ranking_combined[0]]
 
-
-
-    #Generating answers
-
+    # Generating answers
     agent1_answer = generate_answer_withContext(query, agent1_context)
     agent2_answer = generate_answer_withContext(query, agent2_context)
 
@@ -98,7 +92,6 @@ def process_query(query):
 
     zeroShot = generate_answer_zeroShot(query)
 
-
     # Ranking the best answer
     rankerAgentInput = {
         "query": query,
@@ -122,8 +115,6 @@ def process_query(query):
 
     best_model, best_answer = rankerAgent(rankerAgentInput)
 
-
-
     return (
         best_model,
         best_answer,
@@ -145,57 +136,47 @@ def process_query(query):
         zeroShot, "Zero-shot doesn't have a context."
     )
 
-# Gradio interface
+# Interface creation
 def create_interface():
     with gr.Blocks() as interface:
-        gr.Markdown("# Query Answering System")
-        gr.Markdown("Enter a query to get the best model and the best answer using multiple retrieval models and ranking techniques.")
         query_input = gr.Textbox(label="Enter your query")
-        
-        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 and Contexts")
+        best_model_output = gr.Textbox(label="Best Model", interactive=False)
+        best_answer_output = gr.Textbox(label="Best Answer", interactive=False)
 
-        def create_answer_row(label, context):
+        def create_answer_row(label):
             with gr.Row():
-                answer_textbox = gr.Textbox(label=f"{label} Answer", interactive=False).style(container=True)
+                answer_textbox = gr.Textbox(label=f"{label} Answer", interactive=False)
                 context_button = gr.Button(f"Show {label} Context")
-                context_textbox = gr.Textbox(label=f"{label} Context", visible=False).style(container=True)
+                context_textbox = gr.Textbox(label=f"{label} Context", visible=False)
 
+                # Event to show the context
                 context_button.click(
-                    fn=lambda: gr.update(visible=True, value=context),
-                    inputs=[],
-                    outputs=context_textbox,
+                    fn=lambda x: gr.update(visible=True, value=x),
+                    inputs=None,
+                    outputs=context_textbox
                 )
                 return answer_textbox, context_textbox
-                
-
-        with gr.Row():
-            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():
-            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():
-            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.")
+
+        agent1_output, agent1_context_output = create_answer_row("Agent 1")
+
+        agent2_output, agent2_context_output = create_answer_row("Agent 2")
+        boolean_output, boolean_context_output = create_answer_row("Boolean")
+        tf_idf_output, tf_idf_context_output = create_answer_row("TF-IDF")
+        bm25_output, bm25_context_output = create_answer_row("BM25")
+        vision_output, vision_context_output = create_answer_row("Vision")
+        open_source_output, open_source_context_output = create_answer_row("Open Source")
+
+        boolean_mod_output, boolean_mod_context_output = create_answer_row("Boolean (Modified)")
+        tf_idf_mod_output, tf_idf_mod_context_output = create_answer_row("TF-IDF (Modified)")
+        bm25_mod_output, bm25_mod_context_output = create_answer_row("BM25 (Modified)")
+        vision_mod_output, vision_mod_context_output = create_answer_row("Vision (Modified)")
+        open_source_mod_output, open_source_context_output = create_answer_row("Open Source (Modified)")
+
+        tf_idf_rrf_output, tf_idf_rrf_context_output = create_answer_row("TF-IDF + BM25 + Open RRF")
+        tf_idf_rrf_mod_output, tf_idf_rrf_mod_context_output = create_answer_row("TF-IDF + BM25 + Open RRF (Modified)")
+        tf_idf_rrf_combined_output, tf_idf_rrf_combined_context_output = create_answer_row("TF-IDF + BM25 + Open RRF (Combined)")
+
+        zero_shot_output, zero_shot_context_output = create_answer_row("Zero Shot")
 
         gr.Button("Submit").click(
             fn=process_query,
@@ -203,26 +184,28 @@ def create_interface():
             outputs=[
                 best_model_output,
                 best_answer_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,
+                agent1_output, agent1_context_output,
+                agent2_output, agent2_context_output,
+                boolean_output, boolean_context_output,
+                tf_idf_output, tf_idf_context_output,
+                bm25_output, bm25_context_output,
+                vision_output, vision_context_output,
+                open_source_output, open_source_context_output,
+                boolean_mod_output, boolean_mod_context_output,
+                tf_idf_mod_output, tf_idf_mod_context_output,
+                bm25_mod_output, bm25_mod_context_output,
+                vision_mod_output, vision_mod_context_output,
+                open_source_mod_output, open_source_context_output,
+                tf_idf_rrf_output, tf_idf_rrf_context_output,
+                tf_idf_rrf_mod_output, tf_idf_rrf_mod_context_output,
+                tf_idf_rrf_combined_output, tf_idf_rrf_combined_context_output,
+                zero_shot_output, zero_shot_context_output
             ]
         )
-    
+
     return interface
 
+# Launch the interface
 if __name__ == "__main__":
-    create_interface().launch()
+    interface = create_interface()
+    interface.launch()