Update app.py

app.py

@@ -57,50 +57,50 @@ def query_tqa(query, search_level):
     )
 
 
-def eval_llm(query, rag_response, grag_response):
-    """
-    Evaluate the Graph-RAG and RAG responses using an LLM.
+# def eval_llm(query, rag_response, grag_response):
+#     """
+#     Evaluate the Graph-RAG and RAG responses using an LLM.
 
-    Args:
-    query (str): The query that was asked.
-    rag_response (str): The response from the Vanilla-RAG model.
-    grag_response (str): The response from the Graph-RAG model.
+#     Args:
+#     query (str): The query that was asked.
+#     rag_response (str): The response from the Vanilla-RAG model.
+#     grag_response (str): The response from the Graph-RAG model.
 
-    Returns:
-    str: The evaluation text on various criteria from the LLM.
-    """
+#     Returns:
+#     str: The evaluation text on various criteria from the LLM.
+#     """
 
-    if not query.strip() or not rag_response.strip() or not grag_response.strip():
-        raise gr.Error("Please ask a query and get responses before evaluating.")
+#     if not query.strip() or not rag_response.strip() or not grag_response.strip():
+#         raise gr.Error("Please ask a query and get responses before evaluating.")
 
-    eval_text = evaluate_llm(query, grag_response, rag_response)
-    return eval_text
+#     eval_text = evaluate_llm(query, grag_response, rag_response)
+#     return eval_text
 
 
-def reason_and_plot(query, grag_response, grag_reference):
-    """
-    Get the reasoning graph for a query and plot the knowledge graph.
+# def reason_and_plot(query, grag_response, grag_reference):
+#     """
+#     Get the reasoning graph for a query and plot the knowledge graph.
 
-    Args:
-    query (str): The query to ask the Graph-RAG.
-    grag_response (str): The response from the Graph-RAG model.
-    grag_reference (str): The reference text from the Graph-RAG model.
+#     Args:
+#     query (str): The query to ask the Graph-RAG.
+#     grag_response (str): The response from the Graph-RAG model.
+#     grag_reference (str): The reference text from the Graph-RAG model.
 
-    Returns:
-    tuple: The reasoning graph and the HTML to plot the knowledge graph.
-    """
+#     Returns:
+#     tuple: The reasoning graph and the HTML to plot the knowledge graph.
+#     """
 
-    if not query.strip() or not grag_response.strip() or not grag_reference.strip():
-        raise gr.Error(
-            "Please ask a query and get a Graph-RAG response before reasoning."
-        )
+#     if not query.strip() or not grag_response.strip() or not grag_reference.strip():
+#         raise gr.Error(
+#             "Please ask a query and get a Graph-RAG response before reasoning."
+#         )
 
-    graph_reasoning = reasoning_graph(query, grag_response, grag_reference)
-    escaped_html = plot_subgraph(grag_reference)
+#     graph_reasoning = reasoning_graph(query, grag_response, grag_reference)
+#     escaped_html = plot_subgraph(grag_reference)
 
-    iframe_html = f'<iframe srcdoc="{escaped_html}" width="100%" height="400px" frameborder="0"></iframe>'
+#     iframe_html = f'<iframe srcdoc="{escaped_html}" width="100%" height="400px" frameborder="0"></iframe>'
 
-    return graph_reasoning, iframe_html
+#     return graph_reasoning, iframe_html
 
 
 def show_graph():
@@ -170,43 +170,43 @@ with gr.Blocks() as demo:
         with gr.Column():
             gr.Markdown("### Graph-RAG")
             grag_output = gr.Textbox(label="Response", lines=5)
-            grag_reference = gr.Textbox(label="Triplets", lines=3)
-            with gr.Accordion("Extracted Reference (Raw)", open=False):
-                grag_reference_text = gr.Textbox(label="Raw Reference", lines=5)
+            # grag_reference = gr.Textbox(label="Triplets", lines=3)
+            # with gr.Accordion("Extracted Reference (Raw)", open=False):
+            #     grag_reference_text = gr.Textbox(label="Raw Reference", lines=5)
 
-        with gr.Column():
-            gr.Markdown("### Vanilla RAG")
-            rag_output = gr.Textbox(label="Response", lines=5)
-            rag_reference = gr.Textbox(label="Extracted Reference", lines=3)
-            with gr.Accordion("Extracted Reference (Raw)", open=False):
-                rag_reference_text = gr.Textbox(label="Raw Reference", lines=5)
-
-    gr.Markdown("### Coupon")
-    with gr.Row():    
-        with gr.Column():
-            coupon = gr.Text(label="Coupon", lines=1)
-        with gr.Column():
-            reveal = gr.Button("Reveal Coupon", variant="secondary")
+        # with gr.Column():
+        #     gr.Markdown("### Vanilla RAG")
+        #     rag_output = gr.Textbox(label="Response", lines=5)
+        #     rag_reference = gr.Textbox(label="Extracted Reference", lines=3)
+        #     with gr.Accordion("Extracted Reference (Raw)", open=False):
+        #         rag_reference_text = gr.Textbox(label="Raw Reference", lines=5)
 
-    with gr.Row():
-        gr.Markdown("### Evaluate and Compare")
+    # gr.Markdown("### Coupon")
+    # with gr.Row():    
+    #     with gr.Column():
+    #         coupon = gr.Text(label="Coupon", lines=1)
+    #     with gr.Column():
+    #         reveal = gr.Button("Reveal Coupon", variant="secondary")
 
-    with gr.Row():
-        eval_button = gr.Button("Evaluate LLMs", variant="secondary")
+    # with gr.Row():
+    #     gr.Markdown("### Evaluate and Compare")
 
-    grag_performance = gr.Textbox(label="Evaluation", lines=3)
+    # with gr.Row():
+    #     eval_button = gr.Button("Evaluate LLMs", variant="secondary")
 
-    with gr.Row():
-        gr.Markdown("### Graph Reasoning")
+    # grag_performance = gr.Textbox(label="Evaluation", lines=3)
 
-    with gr.Row():
-        reason_button = gr.Button("Get Graph Reasoning", variant="secondary")
+    # with gr.Row():
+    #     gr.Markdown("### Graph Reasoning")
 
-    with gr.Row():
-        with gr.Column():
-            grag_reasoning = gr.Textbox(label="Graph-RAG Reasoning", lines=5)
-        with gr.Column():
-            subgraph_plot = gr.HTML()
+    # with gr.Row():
+    #     reason_button = gr.Button("Get Graph Reasoning", variant="secondary")
+
+    # with gr.Row():
+    #     with gr.Column():
+    #         grag_reasoning = gr.Textbox(label="Graph-RAG Reasoning", lines=5)
+    #     with gr.Column():
+    #         subgraph_plot = gr.HTML()
 
     with gr.Row():
         plot_button = gr.Button("Plot Knowledge Graph", variant="secondary")
@@ -218,35 +218,35 @@ with gr.Blocks() as demo:
         inputs=[query_input, search_level],
         outputs=[
             grag_output,
-            grag_reference,
-            grag_reference_text,
-            rag_output,
-            rag_reference,
-            rag_reference_text,
+            # grag_reference,
+            # grag_reference_text,
+            # rag_output,
+            # rag_reference,
+            # rag_reference_text,
         ],
     )
 
-    eval_button.click(
-        eval_llm,
-        inputs=[query_input, rag_output, grag_output],
-        outputs=[grag_performance],
-    )
+    # eval_button.click(
+    #     eval_llm,
+    #     inputs=[query_input, rag_output, grag_output],
+    #     outputs=[grag_performance],
+    # )
 
-    reason_button.click(
-        reason_and_plot,
-        inputs=[query_input, grag_output, grag_reference],
-        outputs=[grag_reasoning, subgraph_plot],
-    )
+    # reason_button.click(
+    #     reason_and_plot,
+    #     inputs=[query_input, grag_output, grag_reference],
+    #     outputs=[grag_reasoning, subgraph_plot],
+    # )
 
     plot_button.click(
         show_graph,
         outputs=[kg_output],
     )
 
-    reveal.click(
-        reveal_coupon,
-        inputs=[query_input, grag_output],
-        outputs=[coupon],
-    )
+    # reveal.click(
+    #     reveal_coupon,
+    #     inputs=[query_input, grag_output],
+    #     outputs=[coupon],
+    # )
 
 demo.launch(auth=(os.getenv("ID"), os.getenv("PASS")), share=False)