Spaces:

facebook
/

omnisealbench

Running

App Files Files Community

Mark Duppenthaler commited on about 18 hours ago

Commit

3a8329c

1 Parent(s): 2ebc2c3

message size and alias updates

Browse files

Files changed (8) hide show

frontend/dist/assets/{index-DNwK_9E1.js → index-DiLkGNtv.js} +0 -0
frontend/dist/index.html +1 -1
frontend/src/Descriptions.ts +8 -0
frontend/src/components/DatasetSelector.tsx +1 -1
frontend/src/components/Examples.tsx +2 -1
frontend/src/components/LeaderboardChart.tsx +10 -3
frontend/src/components/ModelFilter.tsx +2 -1
frontend/src/components/ModelInfoIcon.tsx +2 -0

frontend/dist/assets/{index-DNwK_9E1.js → index-DiLkGNtv.js} RENAMED Viewed

The diff for this file is too large to render. See raw diff

frontend/dist/index.html CHANGED Viewed

@@ -5,7 +5,7 @@
     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>🥇 Omni Seal Bench Watermarking Leaderboard</title>
-    <script type="module" crossorigin src="/assets/index-DNwK_9E1.js"></script>
     <link rel="stylesheet" crossorigin href="/assets/index-stFRue7K.css">
   </head>
   <body>

     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>🥇 Omni Seal Bench Watermarking Leaderboard</title>
+    <script type="module" crossorigin src="/assets/index-DiLkGNtv.js"></script>
     <link rel="stylesheet" crossorigin href="/assets/index-stFRue7K.css">
   </head>
   <body>

frontend/src/Descriptions.ts CHANGED Viewed

@@ -43,6 +43,10 @@ class Descriptions {
     return this.descriptions[name]?.link
   }
   getModelDescription(name: string): string | undefined {
     return this.modelDescriptions[name]?.description
   }
@@ -59,6 +63,10 @@ class Descriptions {
     return this.modelDescriptions[name]?.github_link
   }
   // Metric description methods
   getMetricDescription(name: string): string | undefined {
     return this.metricDescriptions[name]?.description

     return this.descriptions[name]?.link
   }
+  getModelAlias(name: string): string | undefined {
+    return this.modelDescriptions[name]?.alias
+  }
   getModelDescription(name: string): string | undefined {
     return this.modelDescriptions[name]?.description
   }
     return this.modelDescriptions[name]?.github_link
   }
+  getModelMessageSize(name: string): number | undefined {
+    return this.modelDescriptions[name]?.message_size
+  }
   // Metric description methods
   getMetricDescription(name: string): string | undefined {
     return this.metricDescriptions[name]?.description

frontend/src/components/DatasetSelector.tsx CHANGED Viewed

@@ -26,7 +26,7 @@ const DatasetSelector: React.FC<DatasetSelectorProps> = ({
                 checked={selectedDatasetName === datasetName}
                 onChange={() => onDatasetNameChange(datasetName)}
               />
-              <span className="text-sm">{datasetName}</span>
               <DatasetInfoIcon datasetName={datasetName} />
             </label>
           ))}

                 checked={selectedDatasetName === datasetName}
                 onChange={() => onDatasetNameChange(datasetName)}
               />
+              <span className="text-sm ml-2">{datasetName}</span>
               <DatasetInfoIcon datasetName={datasetName} />
             </label>
           ))}

frontend/src/components/Examples.tsx CHANGED Viewed

@@ -111,6 +111,7 @@ const Examples = ({ fileType }: ExamplesProps) => {
           <div className="grid grid-cols-2 md:grid-cols-3 lg:grid-cols-4 gap-1 max-h-48 overflow-y-auto pr-2">
             {Object.keys(examples).map((model) => {
               const fullName = descriptions.current.getModelFullName(model) || model
               return (
                 <div key={model} className="flex items-center gap-2 text-sm relative group">
                   <label className="flex items-center gap-2 flex-grow">
@@ -123,7 +124,7 @@ const Examples = ({ fileType }: ExamplesProps) => {
                     />
                     <div className="flex items-center">
                       <span className="truncate" title={fullName}>
-                        {model}
                       </span>
                       <ModelInfoIcon modelName={model} />
                     </div>

           <div className="grid grid-cols-2 md:grid-cols-3 lg:grid-cols-4 gap-1 max-h-48 overflow-y-auto pr-2">
             {Object.keys(examples).map((model) => {
               const fullName = descriptions.current.getModelFullName(model) || model
+              const modelAlias = descriptions.current.getModelAlias(model) || model
               return (
                 <div key={model} className="flex items-center gap-2 text-sm relative group">
                   <label className="flex items-center gap-2 flex-grow">
                     />
                     <div className="flex items-center">
                       <span className="truncate" title={fullName}>
+                        {modelAlias}
                       </span>
                       <ModelInfoIcon modelName={model} />
                     </div>

frontend/src/components/LeaderboardChart.tsx CHANGED Viewed

@@ -14,6 +14,7 @@ import API from '../API'
 import LoadingSpinner from './LoadingSpinner'
 import { ContentType, TooltipProps } from 'recharts/types/component/Tooltip'
 import { NameType, ValueType } from 'recharts/types/component/DefaultTooltipContent'
 interface LeaderboardChartProps {
   dataset: string
@@ -82,6 +83,10 @@ const AttackSelector = ({
   )
 }
 const CustomTooltip = (
   props: TooltipProps<ValueType, NameType> & {
     dataset: string
@@ -115,9 +120,11 @@ const CustomTooltip = (
         </div>
         {payload.map((entry: any, i: number) => {
           const model = entry.name
           const handleClick = () => {
             navigate(
-              `${route}?model=${encodeURIComponent(model)}&attack=${encodeURIComponent(selectedAttack || '')}&strength=${strength}`
             )
           }
@@ -285,11 +292,11 @@ const LeaderboardChart = ({ dataset, selectedModels, datasetType }: LeaderboardC
                     return [...models].map((model, index) => {
                       return (
                         <Line
-                          key={model as string}
                           type="monotone"
                           dataKey={selectedMetric as string} // Ensure selectedMetric is a string
                           data={sortedChartData.filter((row) => row.model === model)}
-                          name={model as string}
                           stroke={colors[index % colors.length]}
                           dot={false}
                         />

 import LoadingSpinner from './LoadingSpinner'
 import { ContentType, TooltipProps } from 'recharts/types/component/Tooltip'
 import { NameType, ValueType } from 'recharts/types/component/DefaultTooltipContent'
+import Descriptions from '../Descriptions'
 interface LeaderboardChartProps {
   dataset: string
   )
 }
+function getModelAlias(model: string): string {
+  return Descriptions.getInstance().getModelAlias(model) || model
+}
 const CustomTooltip = (
   props: TooltipProps<ValueType, NameType> & {
     dataset: string
         </div>
         {payload.map((entry: any, i: number) => {
           const model = entry.name
+          const originalModel = entry.payload.model
           const handleClick = () => {
             navigate(
+              `${route}?model=${encodeURIComponent(originalModel)}&attack=${encodeURIComponent(selectedAttack || '')}&strength=${strength}`
             )
           }
                     return [...models].map((model, index) => {
                       return (
                         <Line
+                          key={getModelAlias(model as string)}
                           type="monotone"
                           dataKey={selectedMetric as string} // Ensure selectedMetric is a string
                           data={sortedChartData.filter((row) => row.model === model)}
+                          name={getModelAlias(model as string)}
                           stroke={colors[index % colors.length]}
                           dot={false}
                         />

frontend/src/components/ModelFilter.tsx CHANGED Viewed

@@ -38,6 +38,7 @@ const ModelFilter: React.FC<ModelFilterProps> = ({ models, selectedModels, setSe
             const description = descriptions.current.getModelDescription(model)
             const paperUrl = descriptions.current.getModelPaperUrl(model)
             const githubUrl = descriptions.current.getModelGithubUrl(model)
             return (
               <div key={model} className="flex items-center gap-2 text-sm relative group">
                 <label className="flex items-center gap-2 flex-grow">
@@ -49,7 +50,7 @@ const ModelFilter: React.FC<ModelFilterProps> = ({ models, selectedModels, setSe
                   />
                   <div className="flex items-center">
                     <span className="truncate" title={fullName}>
-                      {model}
                     </span>
                     <ModelInfoIcon modelName={model} />
                   </div>

             const description = descriptions.current.getModelDescription(model)
             const paperUrl = descriptions.current.getModelPaperUrl(model)
             const githubUrl = descriptions.current.getModelGithubUrl(model)
+            const modelAlias = descriptions.current.getModelAlias(model)
             return (
               <div key={model} className="flex items-center gap-2 text-sm relative group">
                 <label className="flex items-center gap-2 flex-grow">
                   />
                   <div className="flex items-center">
                     <span className="truncate" title={fullName}>
+                      {modelAlias || model}
                     </span>
                     <ModelInfoIcon modelName={model} />
                   </div>

frontend/src/components/ModelInfoIcon.tsx CHANGED Viewed

@@ -21,6 +21,7 @@ const ModelInfoIcon: React.FC<ModelInfoIconProps> = ({ modelName }) => {
   const description = descriptions.current.getModelDescription(modelName)
   const paperUrl = descriptions.current.getModelPaperUrl(modelName)
   const githubUrl = descriptions.current.getModelGithubUrl(modelName)
   return (
     <>
@@ -51,6 +52,7 @@ const ModelInfoIcon: React.FC<ModelInfoIconProps> = ({ modelName }) => {
           <div className="mb-2 whitespace-pre-line">
             {description || 'No description available.'}
           </div>
           <div className="flex space-x-4">
             {paperUrl && (
               <div>

   const description = descriptions.current.getModelDescription(modelName)
   const paperUrl = descriptions.current.getModelPaperUrl(modelName)
   const githubUrl = descriptions.current.getModelGithubUrl(modelName)
+  const messageSize = descriptions.current.getModelMessageSize(modelName)
   return (
     <>
           <div className="mb-2 whitespace-pre-line">
             {description || 'No description available.'}
           </div>
+          {messageSize !== undefined && <div className="mb-2">Message Size: {messageSize}</div>}
           <div className="flex space-x-4">
             {paperUrl && (
               <div>