Update app.py

app.py

@@ -1,20 +1,66 @@
-import gradio as gr
+import json
+import os
+import tempfile
 from pathlib import Path
 
-running_markdown_content = """
-# Something is running (hopefully!)
+import gradio as gr
+from huggingface_hub import duplicate_space, upload_folder
+
 
-If all went to plan, something should be running. check the logs.
-"""
+def configure_training(this_space_id, csv_data, character, do_extract_vocals=False):
+    character = character.strip().replace('-', '').replace('_', '').replace(" ", "").lower()
+    ds_cfg = {
+        "character": character,
+        "do_extract_vocals": do_extract_vocals,
+    }
+    with tempfile.TemporaryDirectory() as tempdir:
+        temp_path = Path(tempdir)
+        (temp_path / 'data.csv').write_text(csv_data)
+        (temp_path / 'dataset_config.csv').write_text(json.dumps(ds_cfg))
+        upload_folder(repo_id=this_space_id, folder_path=tempdir, path_in_repo=".", repo_type="space")
+        print("Would normally upload here!")
+        print(list(temp_path.glob("*")))
+    return "OK! Rebooting here in a sec to start training"
 
-startup_markdown_content = """
-# Welcome! Lets get started...
+description = """
+Configure training session for voice cloning.
 
-Here we'd want some input from the user to start a run.
+Please provide a CSV containing YouTube IDs, start times, and end times that we can use to gather the dataset for you.
+
+It should look like this:
+
+```
+ytid,start,end
+YYiQxHM0L-w,300,660
+Ga-CcToGiUM,3105,3300
+```
 """
 
-with gr.Blocks() as demo:
-    gr.Markdown(running_markdown_content if Path('data.csv').exists() else startup_markdown_content)
+if os.environ.get("HF_TOKEN", None) is not None:
+    interface = gr.Interface(
+        configure_training,
+        inputs=[
+            gr.Textbox(label="This Space's Repo ID", info="The repo ID of this space (ex. nateraw/voice-cloning-training-ui)."),
+            gr.TextArea(value="ytid,start,end\n", label="CSV Data", max_lines=50),
+            gr.Textbox(placeholder="Name of character that you're cloning."),
+            gr.Checkbox(
+                False,
+                label="Isolate Vocals",
+                info="If checked, we use demucs to isolate vocals from each audio file. You want to use this if the provided clips contain background music"
+            )
+        ],
+        outputs="text",
+        title="Configure Training Session",
+        description=description,
+    )
+else:
+    with gr.Blocks() as interface:
+        gr.Markdown("""
+## Please Set The HF_TOKEN Environment Variable
+
+Go to the settings tab of this space and add a new environment variable named `HF_TOKEN` with its value being **a token with write access** from [here](https://hf.co/settings/tokens).
+""")
+
 
 if __name__ == '__main__':
-    demo.launch()
+    interface.launch()