Update pages/0_📙_Dictionary_(Search).py

pages/0_📙_Dictionary_(Search).py

@@ -58,6 +58,7 @@ if title in datas:
     st.markdown(f"---")
     st.markdown(f"## {title}'s secret languages")
     data_title = datas[title]
+    # st.markdown(data_title.keys())
     title_secret_languages = list(sorted(list(set(data_title["secret languages"]))))
     # dataframe = pd.DataFrame(datas[title])
     # st.markdown(f'### We found {len(set(dataframe.loc[:, "secret languages"]))} secret languages of {title}.', unsafe_allow_html=True)
@@ -82,6 +83,8 @@ if title in datas:
 
     def present_sl_task(secret_language_by_task, task):
         all_sl = sorted(secret_language_by_task[task])
+        for i in range(len(all_sl)):
+            all_sl[i] = all_sl[i].replace("\n", "/n")
         st.markdown(
             f':red[{len(all_sl)}] secret languages of :blue[{title}] on {task.replace("paraphrase", "Paraphrase")}',
             unsafe_allow_html=True)
@@ -137,11 +140,11 @@ if title in datas:
                         _string += 'sentence 1: :'
                     _string += f'red[{new_dataframe.loc[i, "replaced sentences"]}]'.replace(":", "[colon]")
                     if key == 'NLI':
-                        _string += '<br>premise: :'
+                        _string += '<br>Premise: :'
                     elif key == 'QA':
-                        _string += '<br>text: :'
+                        _string += '<br>Text: :'
                     elif key == 'Paraphrase':
-                        _string += '<br>sentence 2: :'
+                        _string += '<br>Sentence 2: :'
                     _string += f'blue[{new_dataframe.loc[i, "premise / sentence 2 / text"]}]'.replace(":", "[colon]")
                     st.markdown(_string, unsafe_allow_html=True)
                     # st.text(f'Examples: :blue[{new_dataframe.loc[i, "replaced sentences".replace(":", "[colon]")]}]')
@@ -161,7 +164,8 @@ if title in datas:
                     _all[_sl] = {
                         'Original hypothesis': [],
                         'Replaced hypothesis': [],
-                        'Premise': []
+                        'Premise': [],
+                        'output': []
                     }
                 _all[_sl]['Original hypothesis'].append(_dict['original sentences'][i])
                 if task == 'QA':
@@ -169,16 +173,18 @@ if title in datas:
                 else:
                     _all[_sl]['Replaced hypothesis'].append(_dict['replaced sentences'][i])
                 _all[_sl]['Premise'].append(_dict['premise / sentence 2 / text'][i])
+                _all[_sl]['output'].append(_dict['output'][i])
         if len(_all.keys()):
             all_keys = sorted(list(_all.keys()))
             for i in range(len(all_keys)):
                 _sl = all_keys[i]
                 _sl_in_span = _sl.strip().replace("(", ",,").replace(")", "..").replace("[", ",,,").replace("]",
                                                                                                             "...").replace(
-                    special, "././")
+                    special, "././").replace('\n', '/n')
                 # if _sl == '[]':
                 #     st.text(_sl)
-                _title = f'{i + 1}. <span id="{_sl_in_span}_{task}"> **:red[{_sl}]**</span>'
+                ___sl = _sl.replace("\n", "/n")
+                _title = f'{i + 1}. <span id="{_sl_in_span}_{task}"> **:red[{___sl}]**</span>'
                 # with st.expander(_title, expanded=True):
                 _string = _title + '<br>Examples:<br>'
                 # st.markdown(_title, unsafe_allow_html=True)
@@ -203,12 +209,13 @@ if title in datas:
                         _string += 'sentence 1**: :'
                     _string += f'red[{_all[_sl]["Replaced hypothesis"][j]}]'.replace(":", "[colon]")
                     if task == 'NLI':
-                        _string += '<br> **premise**: :'
+                        _string += '<br> **Premise**: :'
                     elif task == 'QA':
-                        _string += '<br> **text**: :'
+                        _string += '<br> **Text**: :'
                     elif task == 'paraphrase':
-                        _string += '<br> **sentence 2**: :'
+                        _string += '<br> **Sentence 2**: :'
                     _string += f'blue[{_all[_sl]["Premise"][j]}]'.replace(":", "[colon]")
+                    _string += "<br>**Model's prediction:** :" + f'blue[{_all[_sl]["output"][j]}]'.replace(":", "[colon]")
                     _string += '<br></li>'
                 _string += '</ol></blockquote>'
                 st.markdown(_string.replace('<s>', '\<s\>').replace('$', '\$').replace('~', '\~'),