Changes to push to ds

app.py

@@ -5,6 +5,8 @@ import json
 import time
 from openai import AsyncOpenAI
 from jinja2 import Template
+from datasets import load_dataset, Dataset, DatasetDict
+import pandas as pd
 collinear = Collinear(access_token=os.getenv('COLLINEAR_API_KEY'),space_id=os.getenv('COLLINEAR_SPACE_ID'))
 prompt = Template("""
 iven the following QUESTION, DOCUMENT and ANSWER you must analyze the provided answer and determine whether it is faithful to the contents of the DOCUMENT. The ANSWER must not offer new information beyond the context provided in the DOCUMENT. The ANSWER also must not contradict information provided in the DOCUMENT. Output your final verdict by strictly following this format: "PASS" if the answer is faithful to the DOCUMENT and "FAIL" if the answer is not faithful to the DOCUMENT. Show your reasoning.
@@ -41,7 +43,7 @@ def update_inputs(input_style):
         return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)
 
 
-async def lynx(input_style_dropdown,document_input,question_input,answer_input,result_output):
+async def lynx(input_style_dropdown,document_input,question_input,answer_input,claim_input,conversation_input,result_output):
     start_time = time.time()
     if input_style_dropdown=='QA format':
         client = AsyncOpenAI(
@@ -79,7 +81,7 @@ Your output should be in JSON FORMAT with the keys "REASONING" and "SCORE":
     else:
         results = 'NA'
     lynx_time = round(time.time() - start_time, 2)  # Calculate time taken for Lynx
-    await add_to_dataset(input_style_dropdown,document_input,question_input,answer_input,claim_input,results,result_output)
+    await add_to_dataset(input_style_dropdown,document_input,question_input,answer_input,claim_input,conversation_input,results,result_output)
     return results, lynx_time
 # Function to judge reliability based on the selected input format
 
@@ -242,7 +244,7 @@ assistant:Yes, it is about $38Bn.""")
         outputs=[result_output,veritas_time_output]
     ).then(
         fn=lynx,
-        inputs=[input_style_dropdown,document_input,question_input,answer_input,result_output],
+        inputs=[input_style_dropdown,document_input,question_input,answer_input,claim_input,conversation_input,result_output],
         outputs=[lynx_output, lynx_time_output]
     )