Update server.py

server.py

@@ -14,7 +14,9 @@ from DISTILLBERTmodel import *
 import os
 import zipfile
 import shutil
-
+from fastapi import Form
+from fastapi import UploadFile, File, Form
+from pathlib import Path
 
 VISUALIZER_CLASSES = {
     "BERT": BERTVisualizer,
@@ -65,67 +67,29 @@ def ping():
 
     
 
+@app.post("/upload_to_path")
+async def upload_to_path(
+    file: UploadFile = File(...),
+    dest_path: str = Form(...)  # e.g., "models/model.pt"
+):
+    full_path = Path("/data") / dest_path
+    full_path.parent.mkdir(parents=True, exist_ok=True)
 
-@app.post("/upload")
-async def upload(file: UploadFile = File(...)):
-    path = f"/data/{file.filename}"
-    with open(path, "wb") as f:
+    with open(full_path, "wb") as f:
         f.write(await file.read())
-    return {"status": "uploaded", "path": path}
- 
-
-@app.post("/migrate_cache")
-def migrate_hf_cache():
-    src_root = Path("/hf_cache")
-    dst_root = Path("/data/hf_cache")
-
-    if not src_root.exists():
-        return {"status": "error", "message": "Source directory does not exist"}
-
-    migrated_files = []
-
-    for src_path in src_root.rglob("*"):
-        if src_path.is_file():
-            relative_path = src_path.relative_to(src_root)
-            dst_path = dst_root / relative_path
 
-            # Create destination directory if needed
-            dst_path.parent.mkdir(parents=True, exist_ok=True)
-
-            # Copy file
-            shutil.copy2(src_path, dst_path)
-            migrated_files.append(str(relative_path))
-
-    return {
-        "status": "done",
-        "files_migrated": migrated_files,
-        "total": len(migrated_files)
-    }
+    return {"status": "uploaded", "path": str(full_path)}
 
 
 
 
-@app.get("/copy_and_extract")
-def copy_and_extract():
-
-    printstr = copy_extract_and_report()
-    
-    return {"message": "done", "log": printstr}
-
-
-
-
-
-@app.get("/data_check")
-def data_check():
-    with open("/data/marker.txt", "w") as f:
-        f.write("hello from server.py\n")
-        
-    files = os.listdir("/data")
-    return {
-        "message": "done",
-        "contents": files
-    }
+@app.post("/make_dir")
+def make_directory(
+    dir_path: str = Form(...)  # e.g., "logs/test_run"
+):
+    full_dir = Path("/data") / dir_path
+    full_dir.mkdir(parents=True, exist_ok=True)
+    return {"status": "created", "directory": str(full_dir)}