Update app.py

app.py

@@ -39,8 +39,7 @@ def process_analysis(task_id, text, model_name, topic):
         return
 
     topic = check_topic(topic)
-    hcRelativeToThreshold, df_sentences = detect_human_text(model_name, topic, text)
-    message = 'Edits found in the text' if hcRelativeToThreshold >= 0 else 'We couldn\'t find edits in the text'
+    hcRelativeToThreshold, hc, df_sentences = detect_human_text(model_name, topic, text)
 
     sentences = [
         {
@@ -51,9 +50,15 @@ def process_analysis(task_id, text, model_name, topic):
         }
         for _, row in df_sentences.iterrows()
     ]
+
+    foundPotentialEdits = [sentence for sentence in sentences if sentence['pvalue'] < 0.05]
+    
+    message = f'We found {len(foundPotentialEdits)} potential edits in the text' if len(foundPotentialEdits) > 0 else 'We couldn\'t find edits in the text'
+
+    
     
     # Store the result
-    task_results[task_id] = {'status': 'completed', 'message': message, 'hcRelativeToThreshold': hcRelativeToThreshold, 'sentences': sentences}
+    task_results[task_id] = {'status': 'completed', 'message': message, 'hcRelativeToThreshold': hcRelativeToThreshold, 'HC': hc, 'sentences': sentences}
 
 @app.route('/detectHumanInAIText/checkText', methods=['POST'])
 def check_text():