updated dataset creation

app.py

@@ -91,7 +91,7 @@ def func(file, number_of_pages, secret):
 
 
     # output, parsed_document = client.predict(file, number_of_pages)
-    output = client.predict(file, number_of_pages)
+    output, latex_output, latex_time, openai_time = client.predict(file, number_of_pages)
     
     if "Error" in output:
         return output   
@@ -111,12 +111,15 @@ def func(file, number_of_pages, secret):
     
     return output
 
-def save_output(unique_filename: str, output: str, num_pages:int, date_string: str) -> None:
+def save_output(unique_filename: str, output: str, num_pages:int, date_string: str, latex_output: str, latex_time: float, openai_time: float) -> None:
 
     # Append outputs and using a thread lock to avoid concurrent writes from different users.
     with scheduler.lock:
         with feedback_file.open("a") as f:
-            f.write(json.dumps({"input_name": unique_filename, "output": output, "num_pages": num_pages, "timestamp": date_string}))
+            f.write(json.dumps({"input_name": unique_filename, "output": output, 
+                                "num_pages": num_pages, "timestamp": date_string,
+                                "latex": latex_output, "latex_extraction_time": latex_time, 
+                                "openai_call_time": openai_time}))
             f.write("\n")