Update app.py

app.py

@@ -180,37 +180,29 @@ class BatchPredictionResponse(BaseModel):
 async def root():
     return {"message": "BERT Compliance Predictor API"}
 
-@app.get("/health")
+@app.get("/v1/bert/health")
 async def health_check():
     return {"status": "healthy"}
 
-@app.get("/training-status")
+@app.get("/v1/bert/training-status")
 async def get_training_status():
     return training_status
 
-@app.post("/upload")
-async def upload_file(file: UploadFile = File(...)):
-    """Upload a CSV file for training or validation"""
-    if not file.filename.endswith('.csv'):
-        raise HTTPException(status_code=400, detail="Only CSV files are allowed")
-    
-    file_path = UPLOAD_DIR / file.filename
-    with file_path.open("wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
-    
-    return {"message": f"File {file.filename} uploaded successfully", "file_path": str(file_path)}
-
-@app.post("/bert/train", response_model=TrainingResponse)
+@app.post("/v1/bert/train", response_model=TrainingResponse)
 async def start_training(
     config: TrainingConfig,
     background_tasks: BackgroundTasks,
-    file_path: str
+    file: UploadFile = File(...)
 ):
     if training_status["is_training"]:
         raise HTTPException(status_code=400, detail="Training is already in progress")
     
-    if not os.path.exists(file_path):
-        raise HTTPException(status_code=404, detail="Training file not found")
+    if not file.filename.endswith('.csv'):
+        raise HTTPException(status_code=400, detail="Only CSV files are allowed")
+    
+    file_path = UPLOAD_DIR / file.filename
+    with file_path.open("wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
     
     training_id = datetime.now().strftime("%Y%m%d_%H%M%S")
     
@@ -222,9 +214,9 @@ async def start_training(
         "status": "starting"
     })
     
-    background_tasks.add_task(train_model_task, config, file_path, training_id)
+    background_tasks.add_task(train_model_task, config, str(file_path), training_id)
     
-    download_url = f"/bert/download-model/{training_id}"
+    download_url = f"/v1/bert/download-model/{training_id}"
     
     return TrainingResponse(
         message="Training started successfully",
@@ -233,7 +225,7 @@ async def start_training(
         download_url=download_url
     )
 
-@app.post("/bert/validate")
+@app.post("/v1/bert/validate")
 async def validate_model(
     file: UploadFile = File(...),
     model_name: str = "BERT_model"
@@ -319,7 +311,7 @@ async def validate_model(
         if os.path.exists(file_path):
             os.remove(file_path)
 
-@app.post("/bert/predict")
+@app.post("/v1/bert/predict")
 async def predict(
     request: Optional[PredictionRequest] = None,
     file: Optional[UploadFile] = File(None),
@@ -492,7 +484,7 @@ async def predict(
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.get("/bert/download-model/{model_id}")
+@app.get("/v1/bert/download-model/{model_id}")
 async def download_model(model_id: str):
     """Download a trained model"""
     model_path = MODEL_SAVE_DIR / f"{model_id}.pth"