main.py

from fastapi import FastAPI, UploadFile, HTTPException
from fastapi.responses import FileResponse
import pandas as pd
from classify import classify_log
import uvicorn

app= FastAPI()

@app.post("/classify")
def classify(file: UploadFile):

    df = pd.read_csv(file.file)

    required_cols = {"source", "log_message"}
    if not required_cols.issubset(df.columns):
        missing = required_cols - set(df.columns)
        raise HTTPException(status_code=400, detail=f"Missing columns: {missing}")

    # Perform classification
    df["target_label"] = classify_log(list(zip(df["source"], df["log_message"])))

    # Save the modified file
    output_file = "test/test_results.csv"
    df.to_csv(output_file, index=False)

    return FileResponse(output_file)


if __name__== "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)