Changes

app.py

@@ -41,7 +41,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):
+async def lynx(input_style_dropdown,document_input,question_input,answer_input,collinear_output):
     start_time = time.time()
     if input_style_dropdown=='QA format':
         client = AsyncOpenAI(
@@ -79,6 +79,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,collinear_output)
     return results, lynx_time
 # Function to judge reliability based on the selected input format
 
@@ -241,12 +242,8 @@ 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],
+        inputs=[input_style_dropdown,document_input,question_input,answer_input,result_output],
         outputs=[lynx_output, lynx_time_output]
-    ).then(
-        fn=add_to_dataset, 
-        inputs=[input_style_dropdown,document_input,question_input,answer_input,claim_input,conversation_input,lynx_output,result_output],
-        outputs=[]
     )