feat: add optional Google Drive upload for recordings

README.md

@@ -37,3 +37,14 @@ Uploads may take a few seconds. Large batches could hit rate limits; keep per-su
 
 The repository view shows only Git-tracked content. Runtime-generated files live only in the ephemeral container filesystem until the Space restarts. Upload or commit them if you need persistence.
 
+## Google Drive Upload (Alternative)
+
+If you prefer uploading to Google Drive:
+
+1. Create a Google Cloud service account with Drive API enabled and grant it access to a Drive folder.
+2. Put the service account JSON in a Space secret named `GDRIVE_SERVICE_ACCOUNT_JSON`. You can paste the JSON string or mount a path and store the path in the secret.
+3. Add another secret `GDRIVE_FOLDER_ID` with the target folder ID.
+4. In the app UI, tick "Upload to Google Drive" before Submit.
+
+The app uses `google-api-python-client` to upload each WAV file into that folder. Errors will be shown in the results area if credentials or permissions are incorrect.
+

app.py

@@ -1,4 +1,5 @@
 import os
+import json
 import re
 import time
 import math
@@ -14,6 +15,16 @@ except Exception:  # package might be missing in some local runs
     HfApi = None
     HfFolder = None
 
+# Google Drive API (service account) optional imports
+try:
+    from google.oauth2 import service_account
+    from googleapiclient.discovery import build
+    from googleapiclient.http import MediaFileUpload
+except Exception:
+    service_account = None
+    build = None
+    MediaFileUpload = None
+
 # Output directory for saved recordings
 OUT_DIR = "recordings"
 os.makedirs(OUT_DIR, exist_ok=True)
@@ -146,6 +157,57 @@ def upload_recordings(paths: Sequence[str]) -> Tuple[int, Optional[str]]:
     return uploaded, None
 
 
+def upload_recordings_to_gdrive(paths: Sequence[str]) -> Tuple[int, Optional[str]]:
+    """Upload files to Google Drive into a folder using a service account.
+
+    Requires secrets:
+    - GDRIVE_SERVICE_ACCOUNT_JSON: full JSON credentials for a service account
+    - GDRIVE_FOLDER_ID: target Drive folder ID
+    Returns (uploaded_count, error_message).
+    """
+    if not paths:
+        return 0, None
+    if not (service_account and build and MediaFileUpload):
+        return 0, "Google API client not installed."
+    svc_json = os.getenv("GDRIVE_SERVICE_ACCOUNT_JSON")
+    folder_id = os.getenv("GDRIVE_FOLDER_ID")
+    if not svc_json:
+        return 0, "Missing GDRIVE_SERVICE_ACCOUNT_JSON secret."
+    if not folder_id:
+        return 0, "Missing GDRIVE_FOLDER_ID secret."
+
+    try:
+        creds = None
+        if svc_json.strip().startswith("{"):
+            data = json.loads(svc_json)
+            creds = service_account.Credentials.from_service_account_info(
+                data,
+                scopes=["https://www.googleapis.com/auth/drive.file"],
+            )
+        else:
+            # if not JSON string, maybe it's a file path provided via secret
+            creds = service_account.Credentials.from_service_account_file(
+                svc_json,
+                scopes=["https://www.googleapis.com/auth/drive.file"],
+            )
+        drive = build("drive", "v3", credentials=creds)
+    except Exception as e:
+        return 0, f"Auth error: {e}"
+
+    uploaded = 0
+    try:
+        for p in paths:
+            if not os.path.isfile(p):
+                continue
+            media = MediaFileUpload(p, mimetype="audio/wav", resumable=False)
+            body = {"name": os.path.basename(p), "parents": [folder_id]}
+            drive.files().create(body=body, media_body=media, fields="id").execute()
+            uploaded += 1
+    except Exception as e:
+        return uploaded, f"Drive upload error: {e}"
+    return uploaded, None
+
+
 def submit_recordings(
     username: str,
     lumos_path: Optional[str],
@@ -155,6 +217,7 @@ def submit_recordings(
     accio_path: Optional[str],
     reparo_path: Optional[str],
     upload_flag: bool,
+    gdrive_flag: bool,
 ) -> str:
     user = sanitize_username(username)
 
@@ -199,6 +262,14 @@ def submit_recordings(
             lines.append(f"Hub upload: {uploaded} file(s) committed to repo.")
             lines.append("(It may take a few seconds to appear in the file browser.)")
 
+    if gdrive_flag:
+        gup, gerr = upload_recordings_to_gdrive(saved_paths)
+        lines.append("")
+        if gerr:
+            lines.append(f"Drive upload attempted: {gup} succeeded, error: {gerr}")
+        else:
+            lines.append(f"Drive upload: {gup} file(s) uploaded to folder.")
+
     return "\n".join(lines)
 
 
@@ -226,12 +297,13 @@ def build_ui() -> gr.Blocks:
 
         with gr.Row():
             upload_checkbox = gr.Checkbox(label="Upload to Hub (requires HF_TOKEN)", value=False)
+            gdrive_checkbox = gr.Checkbox(label="Upload to Google Drive (service account)", value=False)
         submit = gr.Button("Submit")
         result = gr.Markdown()
 
         submit.click(
             fn=submit_recordings,
-            inputs=[username, lumos, nox, alohomora, wingardium, accio, reparo, upload_checkbox],
+            inputs=[username, lumos, nox, alohomora, wingardium, accio, reparo, upload_checkbox, gdrive_checkbox],
             outputs=[result],
         )
 
@@ -239,6 +311,7 @@ def build_ui() -> gr.Blocks:
     Notes:
     - Files are saved locally in `recordings/` with `<spell>_<username>_<timestamp>.wav`.
     - Check "Upload to Hub" to commit them to the repo (needs HF_TOKEN secret).
+    - Or check "Upload to Google Drive" to upload via a service account.
     - 16 kHz mono WAV ensures consistent model training.
     - You can submit partial sets; only provided spells are saved.
         """)

requirements.txt

@@ -2,4 +2,8 @@ gradio
 numpy
 soundfile
 scipy
-huggingface_hub
+huggingface_hub
+google-api-python-client
+google-auth
+google-auth-httplib2
+google-auth-oauthlib