Update app.py

app.py

@@ -1,10 +1,14 @@
+import os
+import pickle
 import tempfile
 import warnings
-import pickle
+from io import BytesIO
 from pathlib import Path
+from uuid import uuid4
 
 import gradio as gr
 import joblib
+from huggingface_hub import upload_file
 from skops import io as sio
 
 title = "skops converter"
@@ -41,11 +45,20 @@ https://github.com/skops-dev/skops/issues/new?title=CONVERSION+error+from+hf.spa
 """
 
 
-def convert(file):
+def convert(file, store):
     msg = ""
     try:
         with warnings.catch_warnings(record=True) as record:
             in_file = Path(file.name)
+            if store:
+                upload_file(
+                    path_or_fileobj=str(in_file),
+                    path_in_repo=f"{uuid4()}/{in_file.name}",
+                    repo_id="adrin/pickle-to-skops",
+                    repo_type="dataset",
+                    token=os.environ["HF_TOKEN"],
+                )
+
             try:
                 obj = joblib.load(in_file)
             except:
@@ -66,11 +79,21 @@ def convert(file):
             msg = "\n".join([repr(w.message) for w in record])
     except Exception as e:
         return None, None, repr(e)
+
     return out_file, unknown_types, msg
 
 
 with gr.Blocks(title=title) as iface:
     gr.Markdown(desc)
+    store = gr.Checkbox(
+        label=(
+            "Store a copy: if you leave this box checked, we store a copy of your"
+            " pickle file in a private place, only used for us to find issues and"
+            " improve the skops format. Please uncheck this box if your pickle file"
+            " includes any personal or sensitive data."
+        ),
+        value=True,
+    )
     upload_button = gr.UploadButton(
         "Click to Upload a File",
         file_types=None,
@@ -79,13 +102,13 @@ with gr.Blocks(title=title) as iface:
     file_output = gr.File(label="Converted File")
     upload_button.upload(
         convert,
-        upload_button,
+        [upload_button, store],
         [
             file_output,
             gr.Text(label="Unknown Types"),
             gr.Text(label="Errors and Warnings"),
         ],
-        api_name="upload-file"
+        api_name="upload-file",
     )