import { LogDtoSourceEnum } from "@montelo/browser-client";
import { Pencil } from "lucide-react";

export const SourceToIconMap: Record<LogDtoSourceEnum, JSX.Element> = {
  [LogDtoSourceEnum.Manual]: (
    <div className="h-6 w-6 rounded bg-slate-400 p-1 dark:bg-slate-600">
      <Pencil className="h-full w-full" />
    </div>
  ),
  [LogDtoSourceEnum.Openai]: (
    <div className="h-6 w-6 rounded bg-green-700 p-1">
      <img src="/openai.svg" alt="Openai Logo" className="h-full w-full" />
    </div>
  ),
  [LogDtoSourceEnum.Mistral]: (
    <div className="h-6 w-6 rounded bg-orange-400 p-1">
      <img src="/mistral.png" alt="Mistral Logo" className="h-full w-full" />
    </div>
  ),
  [LogDtoSourceEnum.Anthropic]: (
    <div className="h-6 w-6 rounded bg-[#c6a083] p-1">
      <img src="/anthropic.png" alt="Anthropic Logo" className="h-full w-full" />
    </div>
  ),
  [LogDtoSourceEnum.Cohere]: (
    <div className="h-6 w-6 rounded bg-orange-500 p-1">
      <img src="/openai.svg" alt="Openai Logo" className="h-full w-full" />
    </div>
  ),
};
