Update app.py

app.py

@@ -70,29 +70,35 @@ with gr.Blocks() as demo:
         image = gr.Image(label="Upload Image")
         prompt = gr.Textbox(label="Prompt")
         add_button = gr.Button("Add to Dataset")
-        add_button.click(add_to_dataset, inputs=[dataset_name, image, prompt], outputs="text")
+        add_output = gr.Textbox()
+        add_button.click(add_to_dataset, inputs=[dataset_name, image, prompt], outputs=add_output)
         
         index = gr.Number(label="Index to Edit/Delete", precision=0)
         new_prompt = gr.Textbox(label="New Prompt")
         edit_button = gr.Button("Edit Prompt")
-        edit_button.click(edit_dataset, inputs=[dataset_name, index, new_prompt], outputs="text")
+        edit_output = gr.Textbox()
+        edit_button.click(edit_dataset, inputs=[dataset_name, index, new_prompt], outputs=edit_output)
         
         delete_button = gr.Button("Delete from Dataset")
-        delete_button.click(delete_from_dataset, inputs=[dataset_name, index], outputs="text")
+        delete_output = gr.Textbox()
+        delete_button.click(delete_from_dataset, inputs=[dataset_name, index], outputs=delete_output)
     
     with gr.Tab("Upload Dataset"):
         upload_dataset_name = gr.Textbox(label="Dataset Name")
         upload_file = gr.File(label="Upload JSONL File")
         upload_button = gr.Button("Upload Dataset")
-        upload_button.click(load_dataset, inputs=[upload_dataset_name, upload_file], outputs="text")
+        upload_output = gr.Textbox()
+        upload_button.click(load_dataset, inputs=[upload_dataset_name, upload_file], outputs=upload_output)
     
     with gr.Tab("Generate/Download Dataset"):
         generate_button = gr.Button("Generate Dataset")
-        generate_button.click(generate_dataset, inputs=[dataset_name], outputs="file")
+        generate_output = gr.File()
+        generate_button.click(generate_dataset, inputs=[dataset_name], outputs=generate_output)
     
     with gr.Tab("View Dataset"):
         view_dataset_name = gr.Textbox(label="Dataset Name")
         view_button = gr.Button("View Dataset")
-        view_button.click(lambda name: datasets.get(name, []), inputs=[view_dataset_name], outputs="json")
+        view_output = gr.JSON()
+        view_button.click(lambda name: datasets.get(name, []), inputs=[view_dataset_name], outputs=view_output)
 
 demo.launch()