Update app.py

app.py

@@ -38,6 +38,7 @@ def query_tqa(query, search_level):
     return grag_response, grag_reference, grag_reference_text, rag_response, rag_reference, rag_reference_text
 
 
+
 def eval_llm(query, rag_response, grag_response):
     """
     Evaluate responses from both Graph-RAG and RAG models using an LLM.
@@ -50,9 +51,8 @@ def eval_llm(query, rag_response, grag_response):
     Returns:
         tuple: Evaluation results for both responses.
     """
-    grag_eval = evaluate_llm(query, grag_response)
-    rag_eval = evaluate_llm(query, rag_response)
-    return parse_evaluation_regex(grag_eval), parse_evaluation_regex(rag_eval)
+    eval_text = evaluate_llm(query, grag_response, rag_response)
+    return eval_text #, parse_evaluation_regex(grag_eval)
 
 
  
@@ -65,25 +65,29 @@ with gr.Blocks() as demo:
             search_level = gr.Slider(minimum=1, maximum=50, value=3, step=5, label="Search level")
             ask_button = gr.Button("Ask TelcomVA!!")
 
-            
+
         with gr.Row():
             with gr.Accordion("Graph-RAG!", open=True):
                 grag_output = gr.Textbox(label="Response")
                 grag_reference = gr.Textbox(label="Triplets")
-                grag_reference_text = gr.Textbox(label="Extracted Reference raw")
+                with gr.Accordion("Extracted Reference raw", open=False):
+                    grag_reference_text = gr.Textbox(label="Extracted Reference raw")
 
             with gr.Accordion("RAG", open=True):
                 rag_output = gr.Textbox(label="Response")
                 rag_reference = gr.Textbox(label="Extracted Reference")
-                rag_reference_text = gr.Textbox(label="Extracted Reference raw")
+                with gr.Accordion("Extracted Reference raw", open=False):
+                    rag_reference_text = gr.Textbox(label="Extracted Reference raw")
 
 
         with gr.Row():
             grag_performance = gr.Textbox(label="Graph-RAG Performance")
-            rag_performance = gr.Textbox(label="RAG Performance")
+            # rag_performance = gr.HTML(label="RAG Performance")
         eval_button = gr.Button("Evaluate LLMs!!")
 
-
+        with gr.Row():
+            grag_reasoning = gr.Textbox(label="Graph-RAG Reasoning")
+        reason_button = gr.Button("Get Graph Reasoning!!")
 
     with gr.Accordion("Explore KG!", open=False):
         kg_output = gr.HTML()
@@ -105,7 +109,14 @@ with gr.Blocks() as demo:
                       inputs=[query_input,rag_output,grag_output],
                       outputs=[
                           grag_performance,
-                          rag_performance
+                        #   rag_performance
+                      ]
+    )
+
+    reason_button.click(reasoning_graph,
+                      inputs=[query_input,grag_output,grag_reference],
+                      outputs=[
+                          grag_reasoning
                       ]
     )
 
@@ -118,4 +129,5 @@ with gr.Blocks() as demo:
     ],
     inputs=[query_input]
     )
+
 demo.launch(auth=(os.getenv('id'), os.getenv('pass')), share=True)