Update app.py

app.py

@@ -28,18 +28,18 @@ task_results = {}
 def index():
     return 'Hello'
 
-def process_analysis(task_id, text, model_name, threshold):
+def process_analysis(task_id, text, model_name, topic, threshold):
     print(f"Processing task: {task_id}")
 
     # Validate data
     print('Validate data')
-    answer = validate_data(text, model_name)
+    answer = validate_data(text, model_name, topic)
     if answer != '':
         task_results[task_id] = {'status': 'error', 'error': answer}
         return
 
     # topic = check_topic(topic)
-    hcRelativeToThreshold, hc, df_sentences = detect_human_text(model_name, threshold, text)
+    hcRelativeToThreshold, hc, df_sentences = detect_human_text(model_name, topic, threshold, text)
 
     sentences = [
         {
@@ -67,14 +67,14 @@ def check_text():
     data = request.get_json()
     text = data.get('text')
     model_name = data.get('model')
-    # topic = data.get('topic')
+    topic = data.get('topic')
     threshold = data.get('threshold')
 
     # Generate a unique taskId
     task_id = str(uuid.uuid4())
 
     # Start processing in a separate thread
-    thread = threading.Thread(target=process_analysis, args=(task_id, text, model_name, threshold))
+    thread = threading.Thread(target=process_analysis, args=(task_id, text, model_name, topic, threshold))
     thread.start()
 
     # Return taskId immediately
@@ -92,21 +92,21 @@ def get_results():
     
     return jsonify(task_results.pop(task_id)), 200
 
-def validate_data(text, model_name, topic=""):
+def validate_data(text, model_name, topic):
     if text is None or text == '':
         return 'Text is missing'
 
     if model_name is None or model_name == '':
         return 'Model name is missing'
 
-    # if topic is None or topic == '':
-    #     return 'Topic is missing'
+    if topic is None or topic == '':
+        return 'Topic is missing'
     
     if model_name not in ['GPT2XL', 'PHI2']:
         return f'Model {model_name} not supported'
     
-    # if check_topic(topic) == None:
-    #     return f'Topic {topic} not supported'
+    if check_topic(topic) == None:
+        return f'Topic {topic} not supported'
     
     return ''