Update app.py

app.py

@@ -189,13 +189,18 @@ test_examples = [[None, "Images/cat_dog.jpeg", promt_cat_dog],
 def gradio_predict(video,image, question):
     answer = predict_answer(video,image, question)
 
-    # Convert the dictionary to a DataFrame
-    df = pd.DataFrame([answer])
+    return answer
 
 
-
-    return answer, df
-
+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())
+    
 css = """
 #container{
     display: block;
@@ -229,11 +234,14 @@ with gr.Blocks(css = css) as app:
             btn = gr.Button("Annotate")
         with gr.Column():
             answer = gr.TextArea(label="Answer")
-            df = gr.Dataframe(label="CSV")
+            save_btn = gr.Button("Save as CSV")
     
     
     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"))
+
     gr.Examples(
         examples=test_examples,
         inputs=[video,image, question],