Update app.py

app.py

@@ -193,22 +193,13 @@ 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):
-    # Assuming 'data' is a dictionary that we want to save as CSV
+    # Assuming 'data' is a dictionary to convert into a DataFrame and save as CSV
     df = pd.DataFrame([data])
-    # 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)
+    csv_buffer = StringIO()
+    df.to_csv(csv_buffer, index=False)
+    csv_buffer.seek(0)  # Move to the start of the StringIO buffer
+    return ("annotations.csv", csv_buffer.getvalue())
     
 css = """
 #container{
@@ -244,12 +235,16 @@ with gr.Blocks(css = css) as app:
         with gr.Column():
             answer = gr.TextArea(label="Answer")
             save_btn = gr.Button("Save as CSV")
+            download_link = gr.File(label="Download CSV")
+
+    # Make sure the inputs and outputs match in your click function
     
     
     btn.click(gradio_predict, inputs=[video,image, question], outputs=answer)
 
     # Button to save the answer as CSV
-    save_btn.click(convert_and_save, inputs=answer, outputs=gr.File(label="Download CSV"))
+    save_btn.click(fn=convert_and_save, inputs=answer, outputs=download_link)
+
 
     gr.Examples(
         examples=test_examples,