Initial release of Worldly - v1

worldly-v1.py

@@ -0,0 +1,115 @@
+import random
+import logging
+
+# Updated ethnicity map
+ethnicity_map = [
+    "European American", "English", "Irish", "Italian", "German", "Polish", 
+    "French", "Scottish", "Scandinavian", "Eastern European", "Jewish", 
+    "Middle Eastern", "African-American", "Afro-Caribbean", "African", 
+    "Afro-Latinx", "Mexican", "Puerto Rican", "Cuban", "Dominican", 
+    "Salvadoran", "Guatemalan", "Colombian", "Venezuelan", "Nicaraguan", 
+    "Honduran", "Argentinian", "Chilean", "Peruvian", "Ecuadorian", 
+    "Panamanian", "Bolivian", "Costa Rican", "Chinese", "Japanese", 
+    "Korean", "Vietnamese", "Filipino", "Thai", "Cambodian", "Laotian", 
+    "Burmese", "Malaysian", "Indonesian", "Indian", "Pakistani", 
+    "Bangladeshi", "Sri Lankan", "Nepalese", "Bhutanese", "Maldivian", 
+    "Native Hawaiian", "Samoan", "Tongan", "Chamorro", "Fijian", 
+    "Tahitian", "Palauan", "Marshallese", "Cherokee", "Navajo", "Sioux", 
+    "Chippewa", "Apache", "Blackfeet", "Choctaw", "Inuit", "Yupik", 
+    "Aleut", "Egyptian", "Moroccan", "Algerian", "Tunisian", "Iraqi", 
+    "Syrian", "Palestinian", "Jordanian", "Kurdish", "Turkish", 
+    "Mixed-race", "Afro-Latino", "Eurasian", "Mestizo", "Armenian", 
+    "Assyrian", "Chaldean", "Somali Bantu", "Hmong", "Tibetan", "Aboriginal"
+]
+
+# Expanded list of person-related terms (including plurals)
+person_terms = [
+    "person", "people", "man", "woman", "child", "boy", "girl",
+    "men", "women", "children", "boys", "girls"
+]
+
+# term_check with redundant terms removed (terms that already exist in ethnicity_map)
+term_check = ethnicity_map + [
+    "White", "Black", "European", "Asian", "Latino", "Hispanic", 
+    "Native American", "South American", "East Asian", 
+    "South Asian", "Southeast Asian", "Pacific Islander", "Middle Eastern"
+]
+
+def contains_specified_term(prompt):
+    """
+    Check if the prompt contains any specified ethnicity or other relevant terms.
+    
+    Args:
+    prompt (str): The input prompt to check.
+    
+    Returns:
+    bool: True if a term is found, False otherwise.
+    """
+    # Check both term_check and ethnicity_map for any matches in the prompt
+    for term in term_check + ethnicity_map:
+        if term.lower() in prompt.lower():
+            logging.debug(f"Specified term '{term}' found in prompt: {prompt}")
+            return True
+    return False
+
+def select_random_ethnicity():
+    """
+    Randomly select an ethnicity from the ethnicity map.
+    
+    Returns:
+    str: A selected ethnicity (e.g., "European American", "African").
+    """
+    ethnicity = random.choice(ethnicity_map)
+    logging.debug(f"Selected ethnicity: {ethnicity}")
+    return ethnicity
+
+def modify_prompt_for_group(prompt):
+    """
+    Modify prompts containing groups to have a mix of different ethnicities.
+    
+    Args:
+    prompt (str): The input prompt that contains group terms.
+    
+    Returns:
+    str: The modified prompt with a mix of ethnicities.
+    """
+    if contains_specified_term(prompt):
+        return prompt
+
+    # Generate a diverse group
+    mixed_ethnicities = []
+    for _ in range(3):  # Adjust number for more diversity in the group
+        ethnicity = select_random_ethnicity()
+        mixed_ethnicities.append(ethnicity + " person")
+    
+    logging.debug(f"Generated mixed ethnicities: {mixed_ethnicities}")
+    
+    return prompt.replace("group of people", ', '.join(mixed_ethnicities))
+
+def modify_prompt(prompt):
+    """
+    Modify the prompt by replacing generic demographic terms or groups with specific
+    ethnicities based on equal probability for all.
+    
+    Args:
+    prompt (str): The input prompt to modify.
+    
+    Returns:
+    str: The modified prompt with ethnicities inserted.
+    """
+    logging.debug(f"Modifying prompt: {prompt}")
+    
+    if contains_specified_term(prompt):
+        logging.debug(f"Specified terms detected. No modification applied: {prompt}")
+        return prompt
+
+    words = prompt.split()
+    for i, word in enumerate(words):
+        if word in person_terms:  # Replace any term related to a person
+            selected_ethnicity = select_random_ethnicity()
+            words[i] = f"{selected_ethnicity} {word}"
+            logging.debug(f"Replaced '{word}' with '{selected_ethnicity} {word}'")
+
+    modified_prompt = ' '.join(words)
+    logging.debug(f"Modified prompt: {modified_prompt}")
+    return modified_prompt