Update app.py

app.py

@@ -248,23 +248,23 @@ def check_softgoal_syntax(softgoals):
 # ############################################
 
 # ########## Incorrect Task Syntax ###########
-def find_non_VPs(sentences):
+def find_NPs(sentences):
 
-  pipeline = TokenClassificationPipeline(model=pos_model, tokenizer=pos_tokenizer)
+  pipeline = TokenClassificationPipeline(model=model, tokenizer=tokenizer)
 
   outputs = pipeline(sentences)
 
-  Non_VPs = []
-
+  NPs = []
+    
   for idx, output in enumerate(outputs):
     if not output[0]['entity'].startswith('V'):
-      Non_VPs.append(sentences[idx])
+      NPs.append(sentences[idx])
     
-  return Non_VPs
+  return NPs
 
 def check_task_syntax(tasks):
 
-  incorrect_task_syntax = find_non_VPs(tasks)
+  incorrect_task_syntax = find_NPs(tasks)
 
   if incorrect_task_syntax:
     incorrect_task_syntax = "\n".join(incorrect_task_syntax)
@@ -273,6 +273,18 @@ def check_task_syntax(tasks):
     return "All tasks are syntactically correct."
 # ############################################
 
+# ########## Incorrect Resource Syntax ###########
+def check_resource_syntax(resources):
+
+  incorrect_resource_syntax = find_non_NPs(resources)
+
+  if incorrect_resource_syntax:
+    incorrect_resource_syntax = "\n".join(incorrect_resource_syntax)
+    return "Incorrect Resources Syntax:\n" + incorrect_resource_syntax
+  else:
+    return "All resources are syntactically correct."
+# ############################################
+
 # ########## Similarity ###########
 def get_similar_elements(elements_per_actor, similarity_threshold):
 
@@ -467,6 +479,9 @@ def identify_bad_smells(tgrl_file, selected_bad_smells, size_threshold, similari
   
   if 'Tasks Syntax' in selected_bad_smells:
     output = output + check_task_syntax(elements['tasks']) + "\n\n"
+      
+  if 'Resources Syntax' in selected_bad_smells:
+    output = output + check_resource_syntax(elements['resources']) + "\n\n"
 
   if 'Similar Elements' in selected_bad_smells:
     output = output + get_similar_elements(elements_per_actor, similarity_threshold) + "\n\n"
@@ -480,13 +495,12 @@ def identify_bad_smells(tgrl_file, selected_bad_smells, size_threshold, similari
   if 'Contradicting Elements' in selected_bad_smells:
     output = output + check_contradiction(elements_per_actor) + "\n\n"
     
-    
   return output
 
 
 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"],
+                         gr.CheckboxGroup(["Size", "Complexity", "Punctuations", "Actors Syntax", "Goals Syntax", "Softgoals Syntax", "Tasks Syntax", "Resources 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.9, minimum = 0, maximum = 1, step = 0.1)],