Update app.py

app.py

@@ -13,7 +13,7 @@ st.title("将棋解説文の自動生成")
 df = pd.read_csv("./dataset10.csv")
 num = st.text_input("0から9の数字を入力")
 
-if  num in list(range(10)):
+if  num in [str(x) for x in list(range(10))]:
     df = df.iloc[int(num)]
     st.write(df["game_type"],df["precedence_name"],df["follower_name"])
     sfen = df["sfen"].split("\n")
@@ -27,6 +27,11 @@ if  num in list(range(10)):
         else:
             te.append(sfen[x])
     board = cshogi.Board()
-    st.markdown(board.to_svg(), unsafe_allow_html=True)
+    st.markdown(board.to_svg(),unsafe_allow_html=True)
+    s = st.selectbox(options=te)
+    idx = te.index(s)
+    board = cshogi.Board(sfen=te_sf[s])
+    st.markdown(board.to_svg(),unsafe_allow_html=True)
+