commit b4853282ddbab28c24fa0787183caa961d227f28
parent 9f475c68189b69b8f2f6fbee25b853ec6fd8013f
Author: Amit Dutta <mail@amit.is-a.dev>
Date: Mon, 20 Jul 2026 20:10:28 +0530
Update and rename index.py to api/index.py
Diffstat:
| A | api/index.py | | | 58 | ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ |
| D | index.py | | | 56 | -------------------------------------------------------- |
2 files changed, 58 insertions(+), 56 deletions(-)
diff --git a/api/index.py b/api/index.py
@@ -0,0 +1,58 @@
+import os
+import random
+import string
+from datetime import datetime
+from fastapi import FastAPI, File, UploadFile, Form, HTTPException
+from huggingface_hub import HfApi
+
+# Create FastAPI app for Vercel
+app = FastAPI(docs_url=None, redoc_url=None)
+
+# Settings
+DATASET_REPO = "notamitgamer/uploads"
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# In Vercel, the route needs to match exactly what the frontend is calling.
+# Since vercel.json routes /api/(.*) to this file, we handle the specific endpoint.
+@app.post("/api/upload")
+async def upload_file(file: UploadFile = File(...), custom_id: str = Form(None)):
+ if not HF_TOKEN:
+ raise HTTPException(status_code=500, detail="HF_TOKEN environment variable is not set")
+
+ try:
+ ext = os.path.splitext(file.filename)[1]
+
+ # Generate unique code if not provided
+ if not custom_id:
+ timestamp = datetime.now().strftime("%Y%m%d-%H%M%S")
+ random_str = ''.join(random.choices(string.ascii_letters + string.digits, k=6))
+ item_id = f"{timestamp}-{random_str}{ext}"
+ else:
+ item_id = custom_id
+ if ext and not item_id.endswith(ext):
+ item_id += ext
+
+ content = await file.read()
+
+ # Vercel allows writing temporary files to /tmp/
+ temp_path = f"/tmp/{item_id}"
+ with open(temp_path, "wb") as f:
+ f.write(content)
+
+ # Push to Hugging Face Dataset
+ api = HfApi()
+ api.upload_file(
+ path_or_fileobj=temp_path,
+ path_in_repo=item_id,
+ repo_id=DATASET_REPO,
+ repo_type="dataset",
+ token=HF_TOKEN
+ )
+
+ # Cleanup tmp file
+ os.remove(temp_path)
+
+ return {"item_id": item_id}
+
+ except Exception as e:
+ raise HTTPException(status_code=500, detail=str(e))
diff --git a/index.py b/index.py
@@ -1,56 +0,0 @@
-import os
-import random
-import string
-from datetime import datetime
-from fastapi import FastAPI, File, UploadFile, Form, HTTPException
-from huggingface_hub import HfApi
-
-# Create FastAPI app for Vercel
-app = FastAPI(docs_url=None, redoc_url=None)
-
-# Settings
-DATASET_REPO = "notamitgamer/uploads"
-HF_TOKEN = os.getenv("HF_TOKEN")
-
-@app.post("/api/upload")
-async def upload_file(file: UploadFile = File(...), custom_id: str = Form(None)):
- if not HF_TOKEN:
- raise HTTPException(status_code=500, detail="HF_TOKEN environment variable is not set")
-
- try:
- ext = os.path.splitext(file.filename)[1]
-
- # Generate unique code if not provided
- if not custom_id:
- timestamp = datetime.now().strftime("%Y%m%d-%H%M%S")
- random_str = ''.join(random.choices(string.ascii_letters + string.digits, k=6))
- item_id = f"{timestamp}-{random_str}{ext}"
- else:
- item_id = custom_id
- if ext and not item_id.endswith(ext):
- item_id += ext
-
- content = await file.read()
-
- # Vercel allows writing temporary files to /tmp/
- temp_path = f"/tmp/{item_id}"
- with open(temp_path, "wb") as f:
- f.write(content)
-
- # Push to Hugging Face Dataset
- api = HfApi()
- api.upload_file(
- path_or_fileobj=temp_path,
- path_in_repo=item_id,
- repo_id=DATASET_REPO,
- repo_type="dataset",
- token=HF_TOKEN
- )
-
- # Cleanup tmp file
- os.remove(temp_path)
-
- return {"item_id": item_id}
-
- except Exception as e:
- raise HTTPException(status_code=500, detail=str(e))
