add file

main.py

@@ -1,5 +1,6 @@
 from flask import Flask, request
 from app import judge
+from train.add_data import add
 from flask_cors import CORS
 
 app = Flask(__name__)
@@ -21,4 +22,5 @@ def returnText(text):
 def check():
     comment = request.json['comment']
     result = judge(comment)
+    add(comment, result[0], result[1], result[2], result[3], result[4], result[5])
     return result

train/add_data.py

@@ -0,0 +1,7 @@
+import pandas as pd
+
+def add(comment, toxic,s_toxic, obscene, threat, insult, identity_hate):
+    df = pd.read_csv('train.csv')
+    size = len(df.index)
+    df.loc[size] = [size,comment,toxic,s_toxic,obscene,threat,insult,identity_hate]
+    df.to_csv('train.csv')