File size: 1,371 Bytes
cd218ce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
import uvicorn
import pandas as pd
from fastapi import FastAPI, UploadFile, HTTPException
from fastapi.responses import FileResponse
from classify import classify
app = FastAPI()
@app.post("/classify/")
async def classify_logs(file: UploadFile):
if not file.filename.endswith('.csv'):
raise HTTPException(status_code=400, detail="File must be a CSV.")
try:
# Read the uploaded CSV
df = pd.read_csv(file.file)
if "source" not in df.columns or "log_message" not in df.columns:
raise HTTPException(status_code=400, detail="CSV must contain 'source' and 'log_message' columns.")
# Perform classification
df["target_label"] = classify(list(zip(df["source"], df["log_message"])))
print("Dataframe:",df.to_dict())
# Save the modified file
output_file = "resources/output.csv"
df.to_csv(output_file, index=False)
print("File saved to output.csv")
return FileResponse(output_file, media_type='text/csv')
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
finally:
file.file.close()
# # Clean up if the file was saved
# if os.path.exists("output.csv"):
# os.remove("output.csv")
if __name__ == '__main__':
uvicorn.run("server:app", host="127.0.0.1", port=8000, reload=True) |