Improve UI

app.py

@@ -85,7 +85,10 @@ if not compact_layout:
         """
     )
 
-generation_mode = st.radio("Mode", ["Standard", "Generation"], horizontal=True) == "Generation"
+generation_mode = st.radio(
+    "Mode", ["Standard mode", "Generation mode"],
+    horizontal=True, label_visibility="collapsed"
+) == "Generation mode"
 st.caption(
     "In standard mode, we analyze the model's predictions on the input text. "
     "In generation mode, we generate a continuation of the input text (prompt) "
@@ -120,29 +123,30 @@ max_tokens = int(MAX_MEM / (multiplier * window_len) - window_len)
 max_tokens = min(max_tokens, 4096)
 
 generate_kwargs = {}
-if generation_mode:
-    with st.expander("Generation options", expanded=False):
-        generate_kwargs["max_new_tokens"] = st.slider(
-            "Max. number of generated tokens",
-            min_value=8, max_value=min(1024, max_tokens), value=min(128, max_tokens)
-        )
-        col1, col2, col3, col4 = st.columns(4)
-        with col1:
-            generate_kwargs["temperature"] = st.number_input(
-                min_value=0.01, value=0.9, step=0.05, label="`temperature`"
-            )
-        with col2:
-            generate_kwargs["top_p"] = st.number_input(
-                min_value=0., value=0.95, max_value=1., step=0.05, label="`top_p`"
-            )
-        with col3:
-            generate_kwargs["typical_p"] = st.number_input(
-                min_value=0., value=1., max_value=1., step=0.05, label="`typical_p`"
-            )
-        with col4:
-            generate_kwargs["repetition_penalty"] = st.number_input(
-                min_value=1., value=1., step=0.05, label="`repetition_penalty`"
+with st.empty():
+    if generation_mode:
+        with st.expander("Generation options", expanded=False):
+            generate_kwargs["max_new_tokens"] = st.slider(
+                "Max. number of generated tokens",
+                min_value=8, max_value=min(1024, max_tokens), value=min(128, max_tokens)
             )
+            col1, col2, col3, col4 = st.columns(4)
+            with col1:
+                generate_kwargs["temperature"] = st.number_input(
+                    min_value=0.01, value=0.9, step=0.05, label="`temperature`"
+                )
+            with col2:
+                generate_kwargs["top_p"] = st.number_input(
+                    min_value=0., value=0.95, max_value=1., step=0.05, label="`top_p`"
+                )
+            with col3:
+                generate_kwargs["typical_p"] = st.number_input(
+                    min_value=0., value=1., max_value=1., step=0.05, label="`typical_p`"
+                )
+            with col4:
+                generate_kwargs["repetition_penalty"] = st.number_input(
+                    min_value=1., value=1., step=0.05, label="`repetition_penalty`"
+                )
 
 DEFAULT_TEXT = """
 We present context length probing, a novel explanation technique for causal
@@ -154,11 +158,14 @@ and offer some initial analyses and insights, including the potential for studyi
 dependencies.
 """.replace("\n", " ").strip()
 
-text = st.text_area(
-    f"Prompt" if generation_mode else "Input text (≤\u2009{max_tokens} tokens)",
-    st.session_state.get("input_text", DEFAULT_TEXT),
-    key="input_text",
-)
+with st.expander(
+    f"Prompt" if generation_mode else f"Input text (≤\u2009{max_tokens} tokens)", expanded=True
+):
+    text = st.text_area(
+        "Input text",
+        st.session_state.get("input_text", DEFAULT_TEXT),
+        key="input_text", label_visibility="collapsed"
+    )
 
 inputs = tokenizer([text])
 [input_ids] = inputs["input_ids"]
@@ -316,6 +323,9 @@ def run_context_length_probing(
 if not generation_mode:
     run_context_length_probing = st.cache_data(run_context_length_probing, show_spinner=False)
 
+if generation_mode:
+    st.button("Rerun", type="primary")
+
 output_ids, scores = run_context_length_probing(
     _model=model,
     _tokenizer=tokenizer,

highlighted_text/build/asset-manifest.json

@@ -1,20 +1,20 @@
 {
   "files": {
-    "main.css": "./static/css/main.a84f16ea.chunk.css",
-    "main.js": "./static/js/main.5f2b5265.chunk.js",
-    "main.js.map": "./static/js/main.5f2b5265.chunk.js.map",
+    "main.css": "./static/css/main.1ea17946.chunk.css",
+    "main.js": "./static/js/main.34cd70dc.chunk.js",
+    "main.js.map": "./static/js/main.34cd70dc.chunk.js.map",
     "runtime-main.js": "./static/js/runtime-main.e6b0ae4c.js",
     "runtime-main.js.map": "./static/js/runtime-main.e6b0ae4c.js.map",
     "static/js/2.ce130e37.chunk.js": "./static/js/2.ce130e37.chunk.js",
     "static/js/2.ce130e37.chunk.js.map": "./static/js/2.ce130e37.chunk.js.map",
     "index.html": "./index.html",
-    "static/css/main.a84f16ea.chunk.css.map": "./static/css/main.a84f16ea.chunk.css.map",
+    "static/css/main.1ea17946.chunk.css.map": "./static/css/main.1ea17946.chunk.css.map",
     "static/js/2.ce130e37.chunk.js.LICENSE.txt": "./static/js/2.ce130e37.chunk.js.LICENSE.txt"
   },
   "entrypoints": [
     "static/js/runtime-main.e6b0ae4c.js",
     "static/js/2.ce130e37.chunk.js",
-    "static/css/main.a84f16ea.chunk.css",
-    "static/js/main.5f2b5265.chunk.js"
+    "static/css/main.1ea17946.chunk.css",
+    "static/js/main.34cd70dc.chunk.js"
   ]
 }

highlighted_text/build/index.html

@@ -1 +1 @@
-<!doctype html><html lang="en"><head><title>Highlighted text component</title><meta charset="UTF-8"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="description" content="Highlighted text"/><link href="./static/css/main.a84f16ea.chunk.css" rel="stylesheet"></head><body><div id="root"></div><script>!function(e){function t(t){for(var n,l,a=t[0],p=t[1],i=t[2],c=0,s=[];c<a.length;c++)l=a[c],Object.prototype.hasOwnProperty.call(o,l)&&o[l]&&s.push(o[l][0]),o[l]=0;for(n in p)Object.prototype.hasOwnProperty.call(p,n)&&(e[n]=p[n]);for(f&&f(t);s.length;)s.shift()();return u.push.apply(u,i||[]),r()}function r(){for(var e,t=0;t<u.length;t++){for(var r=u[t],n=!0,a=1;a<r.length;a++){var p=r[a];0!==o[p]&&(n=!1)}n&&(u.splice(t--,1),e=l(l.s=r[0]))}return e}var n={},o={1:0},u=[];function l(t){if(n[t])return n[t].exports;var r=n[t]={i:t,l:!1,exports:{}};return e[t].call(r.exports,r,r.exports,l),r.l=!0,r.exports}l.m=e,l.c=n,l.d=function(e,t,r){l.o(e,t)||Object.defineProperty(e,t,{enumerable:!0,get:r})},l.r=function(e){"undefined"!=typeof Symbol&&Symbol.toStringTag&&Object.defineProperty(e,Symbol.toStringTag,{value:"Module"}),Object.defineProperty(e,"__esModule",{value:!0})},l.t=function(e,t){if(1&t&&(e=l(e)),8&t)return e;if(4&t&&"object"==typeof e&&e&&e.__esModule)return e;var r=Object.create(null);if(l.r(r),Object.defineProperty(r,"default",{enumerable:!0,value:e}),2&t&&"string"!=typeof e)for(var n in e)l.d(r,n,function(t){return e[t]}.bind(null,n));return r},l.n=function(e){var t=e&&e.__esModule?function(){return e.default}:function(){return e};return l.d(t,"a",t),t},l.o=function(e,t){return Object.prototype.hasOwnProperty.call(e,t)},l.p="./";var a=this.webpackJsonpstreamlit_component_template=this.webpackJsonpstreamlit_component_template||[],p=a.push.bind(a);a.push=t,a=a.slice();for(var i=0;i<a.length;i++)t(a[i]);var f=p;r()}([])</script><script src="./static/js/2.ce130e37.chunk.js"></script><script src="./static/js/main.5f2b5265.chunk.js"></script></body></html>
+<!doctype html><html lang="en"><head><title>Highlighted text component</title><meta charset="UTF-8"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="description" content="Highlighted text"/><link href="./static/css/main.1ea17946.chunk.css" rel="stylesheet"></head><body><div id="root"></div><script>!function(e){function t(t){for(var n,l,a=t[0],p=t[1],i=t[2],c=0,s=[];c<a.length;c++)l=a[c],Object.prototype.hasOwnProperty.call(o,l)&&o[l]&&s.push(o[l][0]),o[l]=0;for(n in p)Object.prototype.hasOwnProperty.call(p,n)&&(e[n]=p[n]);for(f&&f(t);s.length;)s.shift()();return u.push.apply(u,i||[]),r()}function r(){for(var e,t=0;t<u.length;t++){for(var r=u[t],n=!0,a=1;a<r.length;a++){var p=r[a];0!==o[p]&&(n=!1)}n&&(u.splice(t--,1),e=l(l.s=r[0]))}return e}var n={},o={1:0},u=[];function l(t){if(n[t])return n[t].exports;var r=n[t]={i:t,l:!1,exports:{}};return e[t].call(r.exports,r,r.exports,l),r.l=!0,r.exports}l.m=e,l.c=n,l.d=function(e,t,r){l.o(e,t)||Object.defineProperty(e,t,{enumerable:!0,get:r})},l.r=function(e){"undefined"!=typeof Symbol&&Symbol.toStringTag&&Object.defineProperty(e,Symbol.toStringTag,{value:"Module"}),Object.defineProperty(e,"__esModule",{value:!0})},l.t=function(e,t){if(1&t&&(e=l(e)),8&t)return e;if(4&t&&"object"==typeof e&&e&&e.__esModule)return e;var r=Object.create(null);if(l.r(r),Object.defineProperty(r,"default",{enumerable:!0,value:e}),2&t&&"string"!=typeof e)for(var n in e)l.d(r,n,function(t){return e[t]}.bind(null,n));return r},l.n=function(e){var t=e&&e.__esModule?function(){return e.default}:function(){return e};return l.d(t,"a",t),t},l.o=function(e,t){return Object.prototype.hasOwnProperty.call(e,t)},l.p="./";var a=this.webpackJsonpstreamlit_component_template=this.webpackJsonpstreamlit_component_template||[],p=a.push.bind(a);a.push=t,a=a.slice();for(var i=0;i<a.length;i++)t(a[i]);var f=p;r()}([])</script><script src="./static/js/2.ce130e37.chunk.js"></script><script src="./static/js/main.34cd70dc.chunk.js"></script></body></html>

highlighted_text/build/static/css/main.1ea17946.chunk.css

@@ -0,0 +1,2 @@
+body{padding:0;margin:0;font-family:"Source Sans Pro",sans-serif;font-size:1rem;line-height:1.4}.container{border:1px solid #d2d2d2;border-radius:5px;padding:12px 16px}.highlighted-text,.status-bar{color:#000;background-color:#fff}.highlighted-text{background-color:#fefefe;border:1px solid rgba(49,51,63,.2);border-radius:5px;padding:5px;cursor:pointer}.highlighted-text .token.prefix~.token:not(.prefix){color:#2563eb}.highlighted-text .token.active{outline:1px solid #444}.status-bar{min-height:1.4em;padding-bottom:12px}.status-bar .token{border:1px solid #aaa;border-radius:2px;background:#f5f5f5;padding:1px;margin:1px 2px 1px 1px}
+/*# sourceMappingURL=main.1ea17946.chunk.css.map */

highlighted_text/build/static/css/main.1ea17946.chunk.css.map

@@ -0,0 +1 @@
+{"version":3,"sources":["webpack://src/index.scss"],"names":[],"mappings":"AAAA,KACI,SAAA,CACA,QAAA,CACA,wCAAA,CACA,cAAA,CACA,eAAA,CAGJ,WACI,wBAAA,CACA,iBAAA,CACA,iBAAA,CAGJ,8BACI,UAAA,CACA,qBAAA,CAGJ,kBACI,wBAAA,CACA,kCAAA,CACA,iBAAA,CACA,WAAA,CACA,cAAA,CAEA,oDACI,aAAA,CAGJ,gCACI,sBAAA,CAIR,YACI,gBAAA,CACA,mBAAA,CAEA,mBACI,qBAAA,CACA,iBAAA,CACA,kBAAA,CACA,WAAA,CAEA,sBAAA","file":"main.1ea17946.chunk.css","sourcesContent":["body {\n    padding: 0;\n    margin: 0;\n    font-family: \"Source Sans Pro\", sans-serif;\n    font-size: 1rem;\n    line-height: 1.4;\n}\n\n.container {\n    border: 1px solid #d2d2d2;\n    border-radius: 5px;\n    padding: 12px 16px 12px 16px;\n}\n\n.highlighted-text, .status-bar {\n    color: black;\n    background-color: white;\n}\n\n.highlighted-text {\n    background-color: #fefefe;\n    border: 1px solid #31333f33;\n    border-radius: 5px;\n    padding: 5px;\n    cursor: pointer;\n\n    .token.prefix ~ .token:not(.prefix) {\n        color: #2563eb;\n    }\n\n    .token.active {\n        outline: 1px solid #444;\n    }\n}\n\n.status-bar {\n    min-height: 1.4em;\n    padding-bottom: 12px;\n\n    .token {\n        border: 1px solid #aaa;\n        border-radius: 2px;\n        background: #f5f5f5;\n        padding: 1px;\n        margin: 1px;\n        margin-right: 2px;\n    }\n}\n"]}

highlighted_text/build/static/css/main.a84f16ea.chunk.css

@@ -1,2 +0,0 @@
-body{padding:0;margin:0;font-family:"Source Sans Pro",sans-serif;font-size:1rem;line-height:1.4}.highlighted-text,.status-bar{color:#000;background-color:#fff}.highlighted-text{border:1px solid #d2d2d2;border-radius:0 0 5px 5px;padding:4px;cursor:pointer}.highlighted-text .token.prefix~.token:not(.prefix){color:#2563eb}.highlighted-text .token.active{outline:1px solid #444}.status-bar{min-height:1.4em;border:1px solid #d2d2d2;border-bottom:none;border-radius:5px 5px 0 0;padding:3px 4px}.status-bar .token{border:1px solid #aaa;border-radius:2px;background:#f5f5f5;padding:1px;margin:1px 2px 1px 1px}
-/*# sourceMappingURL=main.a84f16ea.chunk.css.map */

highlighted_text/build/static/css/main.a84f16ea.chunk.css.map

@@ -1 +0,0 @@
-{"version":3,"sources":["webpack://src/index.scss"],"names":[],"mappings":"AAAA,KACI,SAAA,CACA,QAAA,CACA,wCAAA,CACA,cAAA,CACA,eAAA,CAGJ,8BACI,UAAA,CACA,qBAAA,CAGJ,kBACI,wBAAA,CACA,yBAAA,CACA,WAAA,CACA,cAAA,CAEA,oDACI,aAAA,CAGJ,gCACI,sBAAA,CAIR,YACI,gBAAA,CAEA,wBAAA,CAAA,kBAAA,CACA,yBAAA,CACA,eAAA,CAEA,mBACI,qBAAA,CACA,iBAAA,CACA,kBAAA,CACA,WAAA,CAEA,sBAAA","file":"main.a84f16ea.chunk.css","sourcesContent":["body {\n    padding: 0;\n    margin: 0;\n    font-family: \"Source Sans Pro\", sans-serif;\n    font-size: 1rem;\n    line-height: 1.4;\n}\n\n.highlighted-text, .status-bar {\n    color: black;\n    background-color: white;\n}\n\n.highlighted-text {\n    border: 1px solid #d2d2d2;\n    border-radius: 0 0 5px 5px;\n    padding: 4px;\n    cursor: pointer;\n\n    .token.prefix ~ .token:not(.prefix) {\n        color: #2563eb;\n    }\n\n    .token.active {\n        outline: 1px solid #444;\n    }\n}\n\n.status-bar {\n    min-height: 1.4em;\n    border: 1px solid #d2d2d2;\n    border-bottom: none;\n    border-radius: 5px 5px 0 0;\n    padding: 3px 4px;\n\n    .token {\n        border: 1px solid #aaa;\n        border-radius: 2px;\n        background: #f5f5f5;\n        padding: 1px;\n        margin: 1px;\n        margin-right: 2px;\n    }\n}\n"]}

highlighted_text/build/static/js/main.34cd70dc.chunk.js

@@ -0,0 +1,2 @@
+(this.webpackJsonpstreamlit_component_template=this.webpackJsonpstreamlit_component_template||[]).push([[0],{27:function(t,e,a){},28:function(t,e,a){"use strict";a.r(e);var n=a(7),s=a.n(n),r=a(18),c=a.n(r),i=a(4),o=a(0),l=a(1),h=a(2),d=a(3),j=a(16),u=a(6),x=function(t){Object(h.a)(a,t);var e=Object(d.a)(a);function a(){var t;Object(o.a)(this,a);for(var n=arguments.length,s=new Array(n),r=0;r<n;r++)s[r]=arguments[r];return(t=e.call.apply(e,[this].concat(s))).state={activeIndex:null,hoverIndex:null,isFrozen:!1},t}return Object(l.a)(a,[{key:"render",value:function(){var t=this,e=this.props.args.tokens,a=this.getScores(),n=this.props.args.prefix_len,s="highlighted-text";this.state.isFrozen&&(s+=" frozen");var r=function(){t.setState({isFrozen:!1})};return Object(u.jsxs)("div",{className:"container",children:[Object(u.jsxs)("div",{className:"status-bar",children:[Object(u.jsxs)("span",{className:this.state.isFrozen?"":" d-none",children:[Object(u.jsx)("i",{className:"fa fa-lock"})," "]},"lock-icon"),null!=this.state.activeIndex?Object(u.jsxs)(u.Fragment,{children:[Object(u.jsx)("strong",{children:"index:"},"index-label")," ",Object(u.jsxs)("span",{children:[this.state.activeIndex," "]},"index")]}):Object(u.jsx)(u.Fragment,{})]},"status-bar"),Object(u.jsx)("div",{className:s,onClick:r,children:e.map((function(e,s){var c="token";t.state&&t.state.activeIndex==s&&(c+=" active"),s<n&&(c+=" prefix");var i={backgroundColor:a[s]>0?"rgba(32, 255, 32, ".concat(a[s],")"):"rgba(255, 32, 32, ".concat(-a[s],")")};return Object(u.jsx)("span",{className:c,style:i,onMouseOver:function(){t.state.isFrozen||t.setState({activeIndex:s}),t.setState({hoverIndex:s})},onClick:r,children:e},s)}))},"text")]})}},{key:"getScores",value:function(){var t=this.props.args.tokens;if(!this.state||null==this.state.activeIndex||this.state.activeIndex<1)return t.map((function(){return 0}));var e=this.props.args.scores,a=this.state.activeIndex-1,n=Math.min(Math.max(0,a),e[a].length),s=e[a].slice(0,n);s.reverse();var r=[].concat(Object(i.a)(Array(Math.max(0,a-s.length)).fill(0)),Object(i.a)(s.map((function(t){return void 0==t||isNaN(t)?0:t}))));return r=[].concat(Object(i.a)(r),Object(i.a)(Array(t.length-r.length).fill(0)))}}]),a}(j.a),b=Object(j.b)(x);a(27);c.a.render(Object(u.jsx)(s.a.StrictMode,{children:Object(u.jsx)(b,{})}),document.getElementById("root"))}},[[28,1,2]]]);
+//# sourceMappingURL=main.34cd70dc.chunk.js.map

highlighted_text/build/static/js/main.34cd70dc.chunk.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["HighlightedText.tsx","index.tsx"],"names":["HighlightedText","_StreamlitComponentBa","_inherits","_super","_createSuper","_this","_classCallCheck","_len","arguments","length","args","Array","_key","call","apply","concat","state","activeIndex","hoverIndex","isFrozen","_createClass","key","value","_this2","tokens","this","props","scores","getScores","prefixLength","className","onClick","setState","_jsxs","children","_jsx","_Fragment","map","t","i","style","backgroundColor","onMouseOver","allScores","hi","Math","min","max","row","slice","reverse","result","_toConsumableArray","fill","x","undefined","isNaN","StreamlitComponentBase","withStreamlitConnection","ReactDOM","render","React","StrictMode","document","getElementById"],"mappings":"gQAeMA,EAAe,SAAAC,GAAAC,YAAAF,EAAAC,GAAA,IAAAE,EAAAC,YAAAJ,GAAA,SAAAA,IAAA,IAAAK,EAAAC,YAAA,KAAAN,GAAA,QAAAO,EAAAC,UAAAC,OAAAC,EAAA,IAAAC,MAAAJ,GAAAK,EAAA,EAAAA,EAAAL,EAAAK,IAAAF,EAAAE,GAAAJ,UAAAI,GACqD,OADrDP,EAAAF,EAAAU,KAAAC,MAAAX,EAAA,OAAAY,OAAAL,KACVM,MAAQ,CAACC,YAAa,KAAMC,WAAY,KAAMC,UAAU,GAAMd,CAAC,CA6ErE,OA7EoEe,YAAApB,EAAA,EAAAqB,IAAA,SAAAC,MAErE,WAAU,IAADC,EAAA,KACCC,EAAmBC,KAAKC,MAAMhB,KAAa,OAC3CiB,EAAmBF,KAAKG,YACxBC,EAAuBJ,KAAKC,MAAMhB,KAAiB,WAErDoB,EAAY,mBACZL,KAAKT,MAAMG,WACXW,GAAa,WAGjB,IAAMC,EAAU,WACZR,EAAKS,SAAS,CAAEb,UAAU,GAC9B,EAEA,OAAOc,eAAA,OAAKH,UAAU,YAAWI,SAAA,CAC7BD,eAAA,OAAKH,UAAU,aAAYI,SAAA,CACvBD,eAAA,QAAMH,UAAWL,KAAKT,MAAMG,SAAW,GAAK,UAAUe,SAAA,CAAiBC,cAAA,KAAGL,UAAU,eAAiB,MAA1C,aAE7B,MAA1BL,KAAKT,MAAMC,YACXgB,eAAAG,WAAA,CAAAF,SAAA,CACIC,cAAA,UAAAD,SAA0B,UAAd,eAA6B,IAACD,eAAA,QAAAC,SAAA,CAAmBT,KAAKT,MAAMC,YAAY,MAAhC,YAEtDkB,cAAAC,WAAA,MAPsB,cAUhCD,cAAA,OAAKL,UAAWA,EAAWC,QAASA,EAAQG,SAEpCV,EAAOa,KAAI,SAACC,EAAWC,GACnB,IAAIT,EAAY,QACZP,EAAKP,OACDO,EAAKP,MAAMC,aAAesB,IAC1BT,GAAa,WAGjBS,EAAIV,IACJC,GAAa,WAEjB,IAAMU,EAAQ,CACVC,gBACId,EAAOY,GAAK,EAAC,qBAAAxB,OACcY,EAAOY,GAAE,0BAAAxB,QACRY,EAAOY,GAAE,MAS7C,OAAOJ,cAAA,QAAcL,UAAWA,EAAWU,MAAOA,EAC9CE,YAPgB,WACXnB,EAAKP,MAAMG,UACZI,EAAKS,SAAS,CAAEf,YAAasB,IAEjChB,EAAKS,SAAS,CAAEd,WAAYqB,GAChC,EAE8BR,QAASA,EAAQG,SAAEI,GAD/BC,EAEtB,KA3ByC,UA+BzD,GAAC,CAAAlB,IAAA,YAAAC,MAED,WACI,IAAME,EAASC,KAAKC,MAAMhB,KAAa,OACvC,IAAKe,KAAKT,OAAmC,MAA1BS,KAAKT,MAAMC,aAAuBQ,KAAKT,MAAMC,YAAc,EAC1E,OAAOO,EAAOa,KAAI,kBAAM,CAAC,IAE7B,IAAMM,EAAwBlB,KAAKC,MAAMhB,KAAa,OAEhD6B,EAAId,KAAKT,MAAMC,YAAc,EAC7B2B,EAAKC,KAAKC,IAAID,KAAKE,IAAI,EAAGR,GAAII,EAAUJ,GAAG9B,QAC3CuC,EAAML,EAAUJ,GAAGU,MAAM,EAAGL,GAClCI,EAAIE,UACJ,IAAIC,EAAM,GAAApC,OAAAqC,YACHzC,MAAMkC,KAAKE,IAAI,EAAGR,EAAIS,EAAIvC,SAAS4C,KAAK,IAAED,YAC1CJ,EAAIX,KAAI,SAACiB,GAAC,YAAUC,GAALD,GAAkBE,MAAMF,GAAK,EAAIA,CAAC,MAGxD,OADAH,EAAM,GAAApC,OAAAqC,YAAOD,GAAMC,YAAKzC,MAAMa,EAAOf,OAAS0C,EAAO1C,QAAQ4C,KAAK,IAEtE,KAACrD,CAAA,CA9EgB,CAASyD,KAiFfC,cAAwB1D,G,MC3FvC2D,IAASC,OACPzB,cAAC0B,IAAMC,WAAU,CAAA5B,SACfC,cAACnC,EAAe,MAElB+D,SAASC,eAAe,Q","file":"static/js/main.34cd70dc.chunk.js","sourcesContent":["import {\n    StreamlitComponentBase,\n    withStreamlitConnection,\n} from \"streamlit-component-lib\";\n\ntype HighlightedTextState = {\n    activeIndex: number | null,\n    hoverIndex: number | null,\n    isFrozen: boolean\n};\n\n/**\n * This is a React-based component template. The `render()` function is called\n * automatically when your component should be re-rendered.\n */\nclass HighlightedText extends StreamlitComponentBase<HighlightedTextState> {\n    public state = {activeIndex: null, hoverIndex: null, isFrozen: false};\n\n    render() {\n        const tokens: string[] = this.props.args[\"tokens\"];\n        const scores: number[] = this.getScores();\n        const prefixLength: number = this.props.args[\"prefix_len\"];\n\n        let className = \"highlighted-text\";\n        if (this.state.isFrozen) {\n            className += \" frozen\";\n        }\n\n        const onClick = () => {\n            this.setState({ isFrozen: false });\n        };\n\n        return <div className=\"container\">\n            <div className=\"status-bar\" key=\"status-bar\">\n                <span className={this.state.isFrozen ? \"\" : \" d-none\"} key=\"lock-icon\"><i className=\"fa fa-lock\"></i> </span>\n                {\n                    this.state.activeIndex != null ?\n                    <>\n                        <strong key=\"index-label\">index:</strong> <span key=\"index\">{this.state.activeIndex} </span>\n                    </>\n                    : <></>\n                }\n            </div>\n            <div className={className} onClick={onClick} key=\"text\">\n                {\n                    tokens.map((t: string, i: number) => {\n                        let className = \"token\";\n                        if (this.state) {\n                            if (this.state.activeIndex == i) {\n                                className += \" active\";\n                            }\n                        }\n                        if (i < prefixLength) {\n                            className += \" prefix\";\n                        }\n                        const style = {\n                            backgroundColor:\n                                scores[i] > 0\n                                    ? `rgba(32, 255, 32, ${scores[i]})`\n                                    : `rgba(255, 32, 32, ${-scores[i]})`\n                        };\n\n                        const onMouseOver = () => {\n                            if (!this.state.isFrozen) {\n                                this.setState({ activeIndex: i });\n                            }\n                            this.setState({ hoverIndex: i });\n                        };\n                        return <span key={i} className={className} style={style}\n                            onMouseOver={onMouseOver} onClick={onClick}>{t}</span>;\n                    })\n                }\n            </div>\n        </div>;\n    }\n\n    private getScores() {\n        const tokens = this.props.args[\"tokens\"];\n        if (!this.state || this.state.activeIndex == null || this.state.activeIndex < 1) {\n            return tokens.map(() => 0);\n        }\n        const allScores: number[][] = this.props.args[\"scores\"];\n\n        const i = this.state.activeIndex - 1;\n        const hi = Math.min(Math.max(0, i), allScores[i].length);\n        const row = allScores[i].slice(0, hi);\n        row.reverse();\n        let result = [\n            ...Array(Math.max(0, i - row.length)).fill(0),\n            ...row.map((x) => x == undefined || isNaN(x) ? 0 : x)\n        ];\n        result = [...result, ...Array(tokens.length - result.length).fill(0)];\n        return result;\n    }\n}\n\nexport default withStreamlitConnection(HighlightedText);\n","import React from \"react\";\nimport ReactDOM from \"react-dom\";\nimport HighlightedText from \"./HighlightedText\";\nimport \"./index.scss\";\n\nReactDOM.render(\n  <React.StrictMode>\n    <HighlightedText />\n  </React.StrictMode>,\n  document.getElementById(\"root\")\n)\n"],"sourceRoot":""}

highlighted_text/build/static/js/main.5f2b5265.chunk.js

@@ -1,2 +0,0 @@
-(this.webpackJsonpstreamlit_component_template=this.webpackJsonpstreamlit_component_template||[]).push([[0],{27:function(t,e,n){},28:function(t,e,n){"use strict";n.r(e);var a=n(7),s=n.n(a),r=n(18),c=n.n(r),i=n(4),o=n(0),l=n(1),h=n(2),d=n(3),j=n(16),u=n(6),x=function(t){Object(h.a)(n,t);var e=Object(d.a)(n);function n(){var t;Object(o.a)(this,n);for(var a=arguments.length,s=new Array(a),r=0;r<a;r++)s[r]=arguments[r];return(t=e.call.apply(e,[this].concat(s))).state={activeIndex:null,hoverIndex:null,isFrozen:!1},t}return Object(l.a)(n,[{key:"render",value:function(){var t=this,e=this.props.args.tokens,n=this.getScores(),a=this.props.args.prefix_len,s="highlighted-text";this.state.isFrozen&&(s+=" frozen");var r=function(){t.setState({isFrozen:!1})};return Object(u.jsxs)(u.Fragment,{children:[Object(u.jsxs)("div",{className:"status-bar",children:[Object(u.jsxs)("span",{className:this.state.isFrozen?"":" d-none",children:[Object(u.jsx)("i",{className:"fa fa-lock"})," "]},"lock-icon"),null!=this.state.activeIndex?Object(u.jsxs)(u.Fragment,{children:[Object(u.jsx)("strong",{children:"index:"},"index-label")," ",Object(u.jsxs)("span",{children:[this.state.activeIndex," "]},"index")]}):Object(u.jsx)(u.Fragment,{})]},"status-bar"),Object(u.jsx)("div",{className:s,onClick:r,children:e.map((function(e,s){var c="token";t.state&&t.state.activeIndex==s&&(c+=" active"),s<a&&(c+=" prefix");var i={backgroundColor:n[s]>0?"rgba(32, 255, 32, ".concat(n[s],")"):"rgba(255, 32, 32, ".concat(-n[s],")")};return Object(u.jsx)("span",{className:c,style:i,onMouseOver:function(){t.state.isFrozen||t.setState({activeIndex:s}),t.setState({hoverIndex:s})},onClick:r,children:e},s)}))},"text")]})}},{key:"getScores",value:function(){var t=this.props.args.tokens;if(!this.state||null==this.state.activeIndex||this.state.activeIndex<1)return t.map((function(){return 0}));var e=this.props.args.scores,n=this.state.activeIndex-1,a=Math.min(Math.max(0,n),e[n].length),s=e[n].slice(0,a);s.reverse();var r=[].concat(Object(i.a)(Array(Math.max(0,n-s.length)).fill(0)),Object(i.a)(s.map((function(t){return void 0==t||isNaN(t)?0:t}))));return r=[].concat(Object(i.a)(r),Object(i.a)(Array(t.length-r.length).fill(0)))}}]),n}(j.a),b=Object(j.b)(x);n(27);c.a.render(Object(u.jsx)(s.a.StrictMode,{children:Object(u.jsx)(b,{})}),document.getElementById("root"))}},[[28,1,2]]]);
-//# sourceMappingURL=main.5f2b5265.chunk.js.map

highlighted_text/build/static/js/main.5f2b5265.chunk.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["HighlightedText.tsx","index.tsx"],"names":["HighlightedText","_StreamlitComponentBa","_inherits","_super","_createSuper","_this","_classCallCheck","_len","arguments","length","args","Array","_key","call","apply","concat","state","activeIndex","hoverIndex","isFrozen","_createClass","key","value","_this2","tokens","this","props","scores","getScores","prefixLength","className","onClick","setState","_jsxs","_Fragment","children","_jsx","map","t","i","style","backgroundColor","onMouseOver","allScores","hi","Math","min","max","row","slice","reverse","result","_toConsumableArray","fill","x","undefined","isNaN","StreamlitComponentBase","withStreamlitConnection","ReactDOM","render","React","StrictMode","document","getElementById"],"mappings":"gQAeMA,EAAe,SAAAC,GAAAC,YAAAF,EAAAC,GAAA,IAAAE,EAAAC,YAAAJ,GAAA,SAAAA,IAAA,IAAAK,EAAAC,YAAA,KAAAN,GAAA,QAAAO,EAAAC,UAAAC,OAAAC,EAAA,IAAAC,MAAAJ,GAAAK,EAAA,EAAAA,EAAAL,EAAAK,IAAAF,EAAAE,GAAAJ,UAAAI,GACqD,OADrDP,EAAAF,EAAAU,KAAAC,MAAAX,EAAA,OAAAY,OAAAL,KACVM,MAAQ,CAACC,YAAa,KAAMC,WAAY,KAAMC,UAAU,GAAMd,CAAC,CA6ErE,OA7EoEe,YAAApB,EAAA,EAAAqB,IAAA,SAAAC,MAErE,WAAU,IAADC,EAAA,KACCC,EAAmBC,KAAKC,MAAMhB,KAAa,OAC3CiB,EAAmBF,KAAKG,YACxBC,EAAeJ,KAAKC,MAAMhB,KAAiB,WAE7CoB,EAAY,mBACZL,KAAKT,MAAMG,WACXW,GAAa,WAGjB,IAAMC,EAAU,WACZR,EAAKS,SAAS,CAAEb,UAAU,GAC9B,EAEA,OAAOc,eAAAC,WAAA,CAAAC,SAAA,CACHF,eAAA,OAAKH,UAAU,aAAYK,SAAA,CACvBF,eAAA,QAAMH,UAAWL,KAAKT,MAAMG,SAAW,GAAK,UAAUgB,SAAA,CAAiBC,cAAA,KAAGN,UAAU,eAAiB,MAA1C,aAE7B,MAA1BL,KAAKT,MAAMC,YACXgB,eAAAC,WAAA,CAAAC,SAAA,CACIC,cAAA,UAAAD,SAA0B,UAAd,eAA6B,IAACF,eAAA,QAAAE,SAAA,CAAmBV,KAAKT,MAAMC,YAAY,MAAhC,YAEtDmB,cAAAF,WAAA,MAPsB,cAUhCE,cAAA,OAAKN,UAAWA,EAAWC,QAASA,EAAQI,SAEpCX,EAAOa,KAAI,SAACC,EAAWC,GACnB,IAAIT,EAAY,QACZP,EAAKP,OACDO,EAAKP,MAAMC,aAAesB,IAC1BT,GAAa,WAGjBS,EAAIV,IACJC,GAAa,WAEjB,IAAMU,EAAQ,CACVC,gBACId,EAAOY,GAAK,EAAC,qBAAAxB,OACcY,EAAOY,GAAE,0BAAAxB,QACRY,EAAOY,GAAE,MAS7C,OAAOH,cAAA,QAAcN,UAAWA,EAAWU,MAAOA,EAC9CE,YAPgB,WACXnB,EAAKP,MAAMG,UACZI,EAAKS,SAAS,CAAEf,YAAasB,IAEjChB,EAAKS,SAAS,CAAEd,WAAYqB,GAChC,EAE8BR,QAASA,EAAQI,SAAEG,GAD/BC,EAEtB,KA3ByC,UA+BzD,GAAC,CAAAlB,IAAA,YAAAC,MAED,WACI,IAAME,EAASC,KAAKC,MAAMhB,KAAa,OACvC,IAAKe,KAAKT,OAAmC,MAA1BS,KAAKT,MAAMC,aAAuBQ,KAAKT,MAAMC,YAAc,EAC1E,OAAOO,EAAOa,KAAI,kBAAM,CAAC,IAE7B,IAAMM,EAAwBlB,KAAKC,MAAMhB,KAAa,OAEhD6B,EAAId,KAAKT,MAAMC,YAAc,EAC7B2B,EAAKC,KAAKC,IAAID,KAAKE,IAAI,EAAGR,GAAII,EAAUJ,GAAG9B,QAC3CuC,EAAML,EAAUJ,GAAGU,MAAM,EAAGL,GAClCI,EAAIE,UACJ,IAAIC,EAAM,GAAApC,OAAAqC,YACHzC,MAAMkC,KAAKE,IAAI,EAAGR,EAAIS,EAAIvC,SAAS4C,KAAK,IAAED,YAC1CJ,EAAIX,KAAI,SAACiB,GAAC,YAAUC,GAALD,GAAkBE,MAAMF,GAAK,EAAIA,CAAC,MAGxD,OADAH,EAAM,GAAApC,OAAAqC,YAAOD,GAAMC,YAAKzC,MAAMa,EAAOf,OAAS0C,EAAO1C,QAAQ4C,KAAK,IAEtE,KAACrD,CAAA,CA9EgB,CAASyD,KAiFfC,cAAwB1D,G,MC3FvC2D,IAASC,OACPxB,cAACyB,IAAMC,WAAU,CAAA3B,SACfC,cAACpC,EAAe,MAElB+D,SAASC,eAAe,Q","file":"static/js/main.5f2b5265.chunk.js","sourcesContent":["import {\n    StreamlitComponentBase,\n    withStreamlitConnection,\n} from \"streamlit-component-lib\";\n\ntype HighlightedTextState = {\n    activeIndex: number | null,\n    hoverIndex: number | null,\n    isFrozen: boolean\n};\n\n/**\n * This is a React-based component template. The `render()` function is called\n * automatically when your component should be re-rendered.\n */\nclass HighlightedText extends StreamlitComponentBase<HighlightedTextState> {\n    public state = {activeIndex: null, hoverIndex: null, isFrozen: false};\n\n    render() {\n        const tokens: string[] = this.props.args[\"tokens\"];\n        const scores: number[] = this.getScores();\n        const prefixLength = this.props.args[\"prefix_len\"];\n\n        let className = \"highlighted-text\";\n        if (this.state.isFrozen) {\n            className += \" frozen\";\n        }\n\n        const onClick = () => {\n            this.setState({ isFrozen: false });\n        };\n\n        return <>\n            <div className=\"status-bar\" key=\"status-bar\">\n                <span className={this.state.isFrozen ? \"\" : \" d-none\"} key=\"lock-icon\"><i className=\"fa fa-lock\"></i> </span>\n                {\n                    this.state.activeIndex != null ?\n                    <>\n                        <strong key=\"index-label\">index:</strong> <span key=\"index\">{this.state.activeIndex} </span>\n                    </>\n                    : <></>\n                }\n            </div>\n            <div className={className} onClick={onClick} key=\"text\">\n                {\n                    tokens.map((t: string, i: number) => {\n                        let className = \"token\";\n                        if (this.state) {\n                            if (this.state.activeIndex == i) {\n                                className += \" active\";\n                            }\n                        }\n                        if (i < prefixLength) {\n                            className += \" prefix\";\n                        }\n                        const style = {\n                            backgroundColor:\n                                scores[i] > 0\n                                    ? `rgba(32, 255, 32, ${scores[i]})`\n                                    : `rgba(255, 32, 32, ${-scores[i]})`\n                        };\n\n                        const onMouseOver = () => {\n                            if (!this.state.isFrozen) {\n                                this.setState({ activeIndex: i });\n                            }\n                            this.setState({ hoverIndex: i });\n                        };\n                        return <span key={i} className={className} style={style}\n                            onMouseOver={onMouseOver} onClick={onClick}>{t}</span>;\n                    })\n                }\n            </div>\n        </>;\n    }\n\n    private getScores() {\n        const tokens = this.props.args[\"tokens\"];\n        if (!this.state || this.state.activeIndex == null || this.state.activeIndex < 1) {\n            return tokens.map(() => 0);\n        }\n        const allScores: number[][] = this.props.args[\"scores\"];\n\n        const i = this.state.activeIndex - 1;\n        const hi = Math.min(Math.max(0, i), allScores[i].length);\n        const row = allScores[i].slice(0, hi);\n        row.reverse();\n        let result = [\n            ...Array(Math.max(0, i - row.length)).fill(0),\n            ...row.map((x) => x == undefined || isNaN(x) ? 0 : x)\n        ];\n        result = [...result, ...Array(tokens.length - result.length).fill(0)];\n        return result;\n    }\n}\n\nexport default withStreamlitConnection(HighlightedText);\n","import React from \"react\";\nimport ReactDOM from \"react-dom\";\nimport HighlightedText from \"./HighlightedText\";\nimport \"./index.scss\";\n\nReactDOM.render(\n  <React.StrictMode>\n    <HighlightedText />\n  </React.StrictMode>,\n  document.getElementById(\"root\")\n)\n"],"sourceRoot":""}

highlighted_text/src/HighlightedText.tsx

@@ -30,7 +30,7 @@ class HighlightedText extends StreamlitComponentBase<HighlightedTextState> {
             this.setState({ isFrozen: false });
         };
 
-        return <>
+        return <div className="container">
             <div className="status-bar" key="status-bar">
                 <span className={this.state.isFrozen ? "" : " d-none"} key="lock-icon"><i className="fa fa-lock"></i> </span>
                 {
@@ -71,7 +71,7 @@ class HighlightedText extends StreamlitComponentBase<HighlightedTextState> {
                     })
                 }
             </div>
-        </>;
+        </div>;
     }
 
     private getScores() {

highlighted_text/src/index.scss

@@ -6,15 +6,22 @@ body {
     line-height: 1.4;
 }
 
+.container {
+    border: 1px solid #d2d2d2;
+    border-radius: 5px;
+    padding: 12px 16px 12px 16px;
+}
+
 .highlighted-text, .status-bar {
     color: black;
     background-color: white;
 }
 
 .highlighted-text {
-    border: 1px solid #d2d2d2;
-    border-radius: 0 0 5px 5px;
-    padding: 4px;
+    background-color: #fefefe;
+    border: 1px solid #31333f33;
+    border-radius: 5px;
+    padding: 5px;
     cursor: pointer;
 
     .token.prefix ~ .token:not(.prefix) {
@@ -28,10 +35,7 @@ body {
 
 .status-bar {
     min-height: 1.4em;
-    border: 1px solid #d2d2d2;
-    border-bottom: none;
-    border-radius: 5px 5px 0 0;
-    padding: 3px 4px;
+    padding-bottom: 12px;
 
     .token {
         border: 1px solid #aaa;