mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 15:48:29 +00:00
3f81383285
* Implement joao feedback * update colors for crew nodes * clean up * more linting clean up * round legend corners --------- Co-authored-by: João Moura <joaomdmoura@gmail.com>
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
def get_legend_items(colors):
|
|
return [
|
|
{"label": "Start Method", "color": colors["start"]},
|
|
{"label": "Method", "color": colors["method"]},
|
|
{
|
|
"label": "Crew Method",
|
|
"color": colors["bg"],
|
|
"border": colors["start"],
|
|
"dashed": False,
|
|
},
|
|
{
|
|
"label": "Router",
|
|
"color": colors["router"],
|
|
"border": colors["router_border"],
|
|
"dashed": True,
|
|
},
|
|
{"label": "Trigger", "color": colors["edge"], "dashed": False},
|
|
{"label": "AND Trigger", "color": colors["edge"], "dashed": True},
|
|
{
|
|
"label": "Router Trigger",
|
|
"color": colors["router_edge"],
|
|
"dashed": True,
|
|
},
|
|
]
|
|
|
|
|
|
def generate_legend_items_html(legend_items):
|
|
legend_items_html = ""
|
|
for item in legend_items:
|
|
if "border" in item:
|
|
style = "dashed" if item["dashed"] else "solid"
|
|
legend_items_html += f"""
|
|
<div class="legend-item">
|
|
<div class="legend-color-box" style="background-color: {item['color']}; border: 2px {style} {item['border']}; border-radius: 5px;"></div>
|
|
<div>{item['label']}</div>
|
|
</div>
|
|
"""
|
|
elif item.get("dashed") is not None:
|
|
style = "dashed" if item["dashed"] else "solid"
|
|
legend_items_html += f"""
|
|
<div class="legend-item">
|
|
<div class="legend-{style}" style="border-bottom: 2px {style} {item['color']}; border-radius: 5px;"></div>
|
|
<div>{item['label']}</div>
|
|
</div>
|
|
"""
|
|
else:
|
|
legend_items_html += f"""
|
|
<div class="legend-item">
|
|
<div class="legend-color-box" style="background-color: {item['color']}; border-radius: 5px;"></div>
|
|
<div>{item['label']}</div>
|
|
</div>
|
|
"""
|
|
return legend_items_html
|