Add size and similarity score threshold input

app.py

@@ -134,13 +134,13 @@ def extract_elements(tgrl_text):
 # ************************* Bad Smells Detection *************************
 
 # ########### Long Elements ###########
-def get_long_elements(elements): # Using RegEx 
+def get_long_elements(elements, size_threshold): # Using RegEx 
 
   long_elements = []
 
   for key, value in elements.items():
     for i in range(0, len(elements[key])):
-        if len(re. findall(r'\w+', elements[key][i])) > 4:
+        if len(re. findall(r'\w+', elements[key][i])) > size_threshold:
             long_elements.append(elements[key][i])
   
   if long_elements:
@@ -332,7 +332,7 @@ def check_task_syntax(tasks):
 # ############################################
 
 # ########## Similarity ###########
-def get_similar_elements(elements_per_actor):
+def get_similar_elements(elements_per_actor, similarity_threshold):
 
   # Prepare sentence pair array
   sentence_pairs = []
@@ -348,7 +348,7 @@ def get_similar_elements(elements_per_actor):
 
   similar_elements = []
   for index, value in enumerate(sentence_pairs):
-    if semantic_similarity_scores[index] > 0.5:
+    if semantic_similarity_scores[index] > similarity_threshold:
       similar_elements.append(value)
       #semantic_similarity["pair_"+str(index+1)] = [value,semantic_similarity_scores[index]]
 
@@ -497,7 +497,7 @@ def check_contradiction(elements_per_actor):
 
 # ************************* User Interface *************************
 
-def identify_bad_smells(tgrl_file, selected_bad_smells):
+def identify_bad_smells(tgrl_file, selected_bad_smells, size_threshold, similarity_threshold):
 
   output = ""
 
@@ -506,7 +506,7 @@ def identify_bad_smells(tgrl_file, selected_bad_smells):
   elements, elements_per_actor, decomposed_elements = extract_elements(tgrl_text)
 
   if 'Size' in selected_bad_smells:
-    output = output + get_long_elements(elements) + "\n\n"
+    output = output + get_long_elements(elements, size_threshold) + "\n\n"
 
   if 'Complexity' in selected_bad_smells:
     output = output + get_complex_sentences(elements) + "\n\n"
@@ -527,7 +527,7 @@ def identify_bad_smells(tgrl_file, selected_bad_smells):
     output = output + check_task_syntax(elements['tasks']) + "\n\n"
 
   if 'Similar Elements' in selected_bad_smells:
-    output = output + get_similar_elements(elements_per_actor) + "\n\n"
+    output = output + get_similar_elements(elements_per_actor, similarity_threshold) + "\n\n"
 
   if 'Spelling Mistakes' in selected_bad_smells:
     output = output + check_spelling(elements) + "\n\n"
@@ -544,9 +544,11 @@ def identify_bad_smells(tgrl_file, selected_bad_smells):
 
 interface = gr.Interface(fn = identify_bad_smells, 
                          inputs = [gr.File(label="TGRL File"),
-                          gr.CheckboxGroup(["Size", "Complexity", "Punctuations", "Actors Syntax", "Goals Syntax", "Softgoals Syntax", "Tasks Syntax", "Similar Elements", "Spelling Mistakes", "Goal-Subgoal Mismatch", "Contradicting Elements"],
-                                           label="Which bad smells you want to detect?")],
-                         outputs = ["text"],
+                         gr.CheckboxGroup(["Size", "Complexity", "Punctuations", "Actors Syntax", "Goals Syntax", "Softgoals Syntax", "Tasks Syntax", "Similar Elements", "Spelling Mistakes", "Goal-Subgoal Mismatch", "Contradicting Elements"],
+                                           label="Which bad smells you want to detect?"),
+                                   gr.Slider(label= "Size threshold", value = 5, minimum = 2, maximum = 10, step = 1),
+                                   gr.Slider(label= "Similarity threshold", value = 0.6, minimum = 0, maximum = 1, step = 0.1)],
+                         outputs = [gr.Textbox(label= "Detected bad smells:")],
                          title = "TGRL Bad Smells Detection",
                          description = "Upload your .xgrl file and we will find the bad smells for you!",
                          theme = gr.themes.Soft())