Update functions.py

functions.py

@@ -121,12 +121,50 @@ def preprocess_text(text):
     return text
 
 
-# def keyword_extraction(text):
-#     try:
-#         extractor = keyword_extractor()
-#         result = extractor(text)
-#     except Exception as e:
-#         # Optionally, log the error: print(f"Error processing text: {e}")
-#         result = None
-
-#     return result
+
+def generate_variants(keyword):
+    # Split the keyword into individual words
+    words = keyword.split()
+    
+    # Original keyword
+    original = keyword
+    # Convert the keyword to all uppercase letters
+    all_upper = keyword.upper()
+    # Convert the keyword to all lowercase letters
+    all_lower = keyword.lower()
+    # Concatenate words with each word capitalized (no spaces)
+    no_space_title = ''.join(word.capitalize() for word in words)
+    # Concatenate words in all uppercase (no spaces)
+    no_space_upper = ''.join(word.upper() for word in words)
+    # Concatenate words in all lowercase (no spaces)
+    no_space_lower = ''.join(word.lower() for word in words)
+    # Create a string with only the first letter of each word (e.g., MHW)
+    initials = ''.join(word[0].upper() for word in words)
+    
+    # Return all variants as a list
+    return [original, all_upper, all_lower, no_space_title, no_space_upper, no_space_lower, initials]
+
+
+
+# Function to check if a cell contains any excluded keywords
+def contains_excluded_keywords(cell, excluded_keywords):
+    if isinstance(cell, np.ndarray):
+        # Join array elements into a single string and search for keywords
+        cell_str = ' '.join(map(str, cell))
+        return any(keyword in cell_str for keyword in excluded_keywords)
+    elif isinstance(cell, str):
+        return any(keyword in cell for keyword in excluded_keywords)
+    return 
+
+
+
+# Function to extract terms from a cell
+def extract_terms(cell):
+    if isinstance(cell, np.ndarray):
+        # Convert each element to a string and strip whitespace
+        return [str(item).strip() for item in cell if str(item).strip()]
+    elif isinstance(cell, str):
+        # Split the string by commas and strip whitespace from each term
+        return [term.strip() for term in cell.split(',') if term.strip()]
+    else:
+        return []