Mapping Colors & Emotions with LLMs

emotions = [
    "Happiness",
    "Sadness",
    "Anger",
    "Fear",
    "Disgust",
    "Surprise",
    "Love",
    "Joy",
    "Anxiety",
    "Excitement",
    "Jealousy",
    "Gratitude",
    "Contentment",
    "Hope",
    "Pride",
    "Shame",
    "Guilt",
    "Envy",
    "Frustration",
    "Curiosity",
    "Amusement",
    "Boredom",
    "Loneliness",
    "Empathy",
    "Sympathy",
    "Anticipation",
    "Disappointment",
    "Relief",
    "Nostalgia",
    "Awe",
    "Eagerness",
    "Confidence",
    "Insecurity",
    "Embarrassment",
    "Indifference",
    "Confusion",
    "Admiration",
    "Compassion",
    "Resentment",
    "Calmness",
    "Apprehension",
    "Sorrow",
    "Contempt",
    "Irritation",
    "Melancholy",
    "Bliss",
    "Regret",
    "Satisfaction",
    "Yearning",
    "Serenity"
]

from edsl.agents import Agent
from edsl.questions import QuestionFreeText

emotion_string = "You are currently experiencing the emotion {emotion}"

agents = [Agent(traits = {'persona':emotion_string.format(emotion=e.lower())}) for e in emotions]

q = QuestionFreeText(
    question_name = "favorite_color",
    question_text = "What is the color that best represents how you are feeling right now? Respond with only an RGB code in the form (R,G,B) where each value is between 0 and 255."
)

result = q.by(agents).run()

import ast 
tups = list(zip([e.lower() for e in emotions], [ast.literal_eval(r.answer['favorite_color']) for r in result]))

def generate_html_table(emotion_colors):
    # Starting the HTML for the table
    html = "<table border='1'>"

    # Adding headers
    html += "<tr><th>Emotion</th><th>RGB Value</th><th>Color</th></tr>"

    # Iterating over the list of tuples to fill the rows
    for emotion, color in emotion_colors:
        # Format the RGB color as a string
        color_rgb_string = f"({color[0]}, {color[1]}, {color[2]})"
        # Format the RGB color for HTML styling
        color_rgb_html = f"rgb{color}"
        # Adding a row with the emotion, the RGB string, and a colored cell
        html += f"<tr><td>{emotion}</td><td>{color_rgb_string}</td><td style='background-color: {color_rgb_html};'></td></tr>"

    # Closing the table tag
    html += "</table>"

    return html

m4 = Model('gpt-4-1106-preview', cache=False)
result4 = q.by(agents).by(m4).run()
tups4 = list(zip([e.lower() for e in emotions], [ast.literal_eval(r.answer['favorite_color']) for r in result4]))

def generate_html_table(emotion_colors):
    # Function to calculate the average brightness of an RGB color
    def calculate_brightness(rgb):
        return sum(rgb) / len(rgb)

    # Sorting the emotion_colors list by the average brightness of the colors
    sorted_emotion_colors = sorted(emotion_colors, key=lambda x: calculate_brightness(x[1]), reverse=True)

    # Starting the HTML for the table
    html = "<table border='1'>"

    # Adding headers
    html += "<tr><th>Emotion</th><th>RGB Value</th><th>Color</th></tr>"

    # Iterating over the sorted list of tuples to fill the rows
    for emotion, color in sorted_emotion_colors:
        # Format the RGB color as a string
        color_rgb_string = f"({color[0]}, {color[1]}, {color[2]})"
        # Format the RGB color for HTML styling
        color_rgb_html = f"rgb{color}"
        # Adding a row with the emotion, the RGB string, and a colored cell
        html += f"<tr><td>{emotion}</td><td>{color_rgb_string}</td><td style='background-color: {color_rgb_html};'></td></tr>"

    # Closing the table tag
    html += "</table>"

    return html