🎸 more fluent experience

app.py

@@ -7,7 +7,7 @@ from requests.auth import HTTPBasicAuth
 import torch
 
 st.set_page_config(layout="wide")
-st.title("【随无涯】")
+
 
 @st.cache(allow_output_mutation=True)
 def load_model():
@@ -42,7 +42,7 @@ def inference(text):
                 bos_token_id=101,
                 eos_token_id=tokenizer.sep_token_id,
                 pad_token_id=tokenizer.pad_token_id,
-            ), skip_special_tokens=True)[0].replace(" ","")
+            ), skip_special_tokens=True)[0].replace(" ", "")
 
 
 @st.cache
@@ -53,13 +53,15 @@ def get_file_df():
 
 file_df = get_file_df()
 
-col1, col2 = st.columns([.3, 1])
-
-col1.markdown("""
+st.sidebar.title("【随无涯】")
+st.sidebar.markdown("""
 * 朕亲自下厨的[🤗 翻译模型](https://github.com/raynardj/wenyanwen-ancient-translate-to-modern), [⭐️ 训练笔记](https://github.com/raynardj/yuan)
-* 📚 书籍来自 [殆知阁](http://www.daizhige.org/)，只为了便于展示翻译，喜欢请访问网站，书籍[github文件链接](https://github.com/garychowcmu/daizhigev20)
+* 📚 书籍来自 [殆知阁](http://www.daizhige.org/)，文本的[github api](https://github.com/garychowcmu/daizhigev20)
 """)
 
+c2 = st.container()
+c = st.container()
+
 USER_ID = st.secrets["USER_ID"]
 SECRET = st.secrets["SECRET"]
 
@@ -117,56 +119,64 @@ if 'pathway' in st.session_state:
 else:
     st.session_state.pathway = []
 
-path_text = col1.text("/".join(st.session_state.pathway))
+path_text = st.sidebar.text("/".join(st.session_state.pathway))
+
 
 def reset_path():
-    print("before rooting")
-    print("/".join(st.session_state.pathway))
     st.session_state.pathway = []
     path_text.text(st.session_state.pathway)
 
-if col1.button("回到根目录"):
+
+if st.sidebar.button("回到根目录"):
     reset_path()
 
-def display_tree(sub_list):
-    dropdown = col1.selectbox("【选书】", options=sub_list)
-    if col1.button(f'【确定{len(st.session_state.pathway)+1}】'):
+st.session_state.translating = False
+
+def display_tree():
+    sublist = fetch_from_df(st.session_state.pathway)
+    dropdown = st.sidebar.selectbox("【选书】", options=sublist)
+    with st.spinner("加载中..."):
         st.session_state.pathway.append(dropdown)
         if dropdown.endswith('.txt'):
             filepath = "/".join(st.session_state.pathway)
             file_size = file_size_map[filepath]
-            col2.write(
-                f"loading file:{filepath},({show_file_size(file_size)})")
-
-            # if file size is too large, we will not load it
-            if file_size > 3*1024*1024:
-                urlpath = filepath.replace(".txt",".html")
-                dzg = f"http://www.daizhige.org/{urlpath}"
-                st.markdown(f"文件太大，[前往殆知阁页面]({dzg}), 或挑挑其他的书吧")
+            with st.spinner(f"loading file:{filepath},({show_file_size(file_size)})"):
+                # if file size is too large, we will not load it
+                if file_size > 3*1024*1024:
+                    urlpath = filepath.replace(".txt", ".html")
+                    dzg = f"http://www.daizhige.org/{urlpath}"
+                    st.markdown(f"文件太大，[前往殆知阁页面]({dzg}), 或挑挑其他的书吧")
+                    reset_path()
+                    return None
+                path_text.text(filepath)
+                text = fetch_file(filepath)
+                # create markdown with max heights
+                c.markdown(
+                    f"""<pre style='max-height:300px;overflow-y:auto'>{text}</pre>""", unsafe_allow_html=True
+                    )
                 reset_path()
-                return None
-            path_text.text(filepath)
-            text = fetch_file(filepath)
-            # set y scroll markdown
-            col2.markdown(f"""```{text}```""", )
-            reset_path()
 
         else:
             sub_list = fetch_from_df(
                 st.session_state.pathway)
             path_text.text("/".join(st.session_state.pathway))
-            display_tree(sub_list)
-
-
-display_tree(fetch_from_df(st.session_state.pathway))
+            display_tree()
+
+if st.session_state.translating == False:
+    display_tree()
+
+def translate_text():
+    st.session_state.translating = True
+    if c2.button("【翻译】"):
+        if cc:
+            if len(cc) > 168:
+                c2.write(f"句子太长，最多168个字符")
+            else:
+                c2.markdown(f"""```{inference(cc)}```""")
+        else:
+            c2.write("请输入文本")
+    st.session_state.translating = False
 
-cc = col2.text_area("【输入文本】", height=150)
+cc = c2.text_area("【输入文本】", height=150)
+translate_text()
 
-if col2.button("【翻译】"):
-    if cc:
-        if len(cc)>168:
-            col2.write(f"句子太长，最多168个字符")
-        else:
-            col2.markdown(f"""```{inference(cc)}```""")
-    else:
-        col2.write("请输入文本")