Update app.py

app.py

@@ -193,14 +193,22 @@ def gradio_predict(video,image, question):
     return answer
 
 
+# def convert_and_save(data):
+#     # Convert the answer dictionary to a DataFrame
+#     df = pd.DataFrame([data])
+#     # Convert DataFrame to CSV format
+#     csv_buffer = StringIO()
+#     df.to_csv(csv_buffer, index=False)
+#     # Gradio file components expect (filename, filecontents), hence the StringIO contents are read
+#     return ("annotations.csv", csv_buffer.getvalue())
+
 def convert_and_save(data):
-    # Convert the answer dictionary to a DataFrame
+    # Assuming 'data' is a dictionary that we want to save as CSV
     df = pd.DataFrame([data])
-    # Convert DataFrame to CSV format
-    csv_buffer = StringIO()
-    df.to_csv(csv_buffer, index=False)
-    # Gradio file components expect (filename, filecontents), hence the StringIO contents are read
-    return ("annotations.csv", csv_buffer.getvalue())
+    # Convert the DataFrame to a CSV string
+    csv_string = df.to_csv(index=False)
+    # Return a tuple with the filename and the CSV string
+    return ("annotations.csv", csv_string)
     
 css = """
 #container{