Update app.py

app.py

@@ -194,12 +194,12 @@ def gradio_predict(video,image, question):
 
 
 def convert_and_save(data):
-    # Assuming 'data' is a dictionary to convert into a DataFrame and save as CSV
+    # Assuming 'data' is a dictionary that we want to save as CSV
     df = pd.DataFrame([data])
-    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())
+    # 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{
@@ -235,7 +235,7 @@ 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")
+            download_link = gr.Dataframe(label="Download CSV")
 
     # Make sure the inputs and outputs match in your click function