Update app.py
Browse files
app.py
CHANGED
|
@@ -19,26 +19,22 @@ from human_text_detect import detect_human_text
|
|
| 19 |
app = Flask(__name__)
|
| 20 |
CORS(app)
|
| 21 |
|
|
|
|
|
|
|
| 22 |
@app.route('/')
|
| 23 |
def index():
|
| 24 |
return 'Hello'
|
| 25 |
|
| 26 |
-
|
| 27 |
-
|
| 28 |
|
| 29 |
-
# Get data
|
| 30 |
-
print('Get data')
|
| 31 |
-
data = request.get_json()
|
| 32 |
-
text = data.get('text')
|
| 33 |
-
model_name = data.get('model')
|
| 34 |
-
topic = data.get('topic')
|
| 35 |
-
|
| 36 |
# Validate data
|
| 37 |
print('Validate data')
|
| 38 |
answer = validate_data(text, model_name, topic)
|
| 39 |
if answer != '':
|
| 40 |
-
|
| 41 |
-
|
|
|
|
| 42 |
topic = check_topic(topic)
|
| 43 |
hcRelativeToThreshold, df_sentences = detect_human_text(model_name, topic, text)
|
| 44 |
message = 'Edits found in the text' if hcRelativeToThreshold >= 0 else 'We couldn\'t find edits in the text'
|
|
@@ -52,7 +48,40 @@ def check_text():
|
|
| 52 |
}
|
| 53 |
for _, row in df_sentences.iterrows()
|
| 54 |
]
|
| 55 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
|
| 57 |
def validate_data(text, model_name, topic):
|
| 58 |
if text is None or text == '':
|
|
|
|
| 19 |
app = Flask(__name__)
|
| 20 |
CORS(app)
|
| 21 |
|
| 22 |
+
task_results = {}
|
| 23 |
+
|
| 24 |
@app.route('/')
|
| 25 |
def index():
|
| 26 |
return 'Hello'
|
| 27 |
|
| 28 |
+
def process_analysis(task_id, text, model_name, topic):
|
| 29 |
+
print(f"Processing task: {task_id}")
|
| 30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
# Validate data
|
| 32 |
print('Validate data')
|
| 33 |
answer = validate_data(text, model_name, topic)
|
| 34 |
if answer != '':
|
| 35 |
+
task_results[task_id] = {'error': answer}
|
| 36 |
+
return
|
| 37 |
+
|
| 38 |
topic = check_topic(topic)
|
| 39 |
hcRelativeToThreshold, df_sentences = detect_human_text(model_name, topic, text)
|
| 40 |
message = 'Edits found in the text' if hcRelativeToThreshold >= 0 else 'We couldn\'t find edits in the text'
|
|
|
|
| 48 |
}
|
| 49 |
for _, row in df_sentences.iterrows()
|
| 50 |
]
|
| 51 |
+
|
| 52 |
+
# Store the result
|
| 53 |
+
task_results[task_id] = {'message': message, 'hcRelativeToThreshold': hcRelativeToThreshold, 'sentences': sentences}
|
| 54 |
+
|
| 55 |
+
@app.route('/detectHumanInAIText/checkText', methods=['POST'])
|
| 56 |
+
def check_text():
|
| 57 |
+
# Get data
|
| 58 |
+
print('Get data')
|
| 59 |
+
data = request.get_json()
|
| 60 |
+
text = data.get('text')
|
| 61 |
+
model_name = data.get('model')
|
| 62 |
+
topic = data.get('topic')
|
| 63 |
+
|
| 64 |
+
# Generate a unique taskId
|
| 65 |
+
task_id = str(uuid.uuid4())
|
| 66 |
+
|
| 67 |
+
# Start processing in a separate thread
|
| 68 |
+
thread = threading.Thread(target=process_analysis, args=(task_id, text, model_name, topic))
|
| 69 |
+
thread.start()
|
| 70 |
+
|
| 71 |
+
# Return taskId immediately
|
| 72 |
+
return jsonify({'taskId': task_id}), 202
|
| 73 |
+
|
| 74 |
+
@app.route('/detectHumanInAIText/getAnalyzeResults', methods=['GET'])
|
| 75 |
+
def get_results():
|
| 76 |
+
task_id = request.args.get('taskId')
|
| 77 |
+
|
| 78 |
+
if not task_id:
|
| 79 |
+
return jsonify({'error': 'Missing taskId parameter'}), 400
|
| 80 |
+
|
| 81 |
+
if task_id not in task_results:
|
| 82 |
+
return jsonify({'status': 'pending'}), 202
|
| 83 |
+
|
| 84 |
+
return jsonify(task_results.pop(task_id)), 200
|
| 85 |
|
| 86 |
def validate_data(text, model_name, topic):
|
| 87 |
if text is None or text == '':
|