Update pages/type_text.py

pages/type_text.py

@@ -11,19 +11,15 @@ os.getenv("HF_TOKEN")
 
 for k, v in st.session_state.items():
     st.session_state[k] = v 
-    st.write("KKKK: ", k, "VVVVVV: ", v)
     
-selected_chapters_floatlist = list(st.session_state.items())[0][1]
-selected_chapters_intlist = [int(i) for i in selected_chapters_floatlist]
+#selected_chapters_floatlist = list(st.session_state.items())[0][1]
+#selected_chapters_intlist = [int(i) for i in selected_chapters_floatlist]
 #st.write("SELECTED CHAPTERS: ", selected_chapters_intlist)
 #selected_chapters_floatlist = list(st.session_state.items())[0][1]
 #selected_chapters_intlist = [int(i) for i in selected_chapters_floatlist]
-#selected_rows_list = list(st.session_state.items())[6][1]
-#for item in st.session_state.items():
-#    print("IIIIIIIIIIIII: ", item)
-
+selected_rows_list = list(st.session_state.items())[1][1]
 
-#st.write("SELECTED ROWS: ", selected_rows_list)
+st.write("SELECTED ROWS: ", selected_rows_list)
 
 def get_device_map() -> str:
     return 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -76,6 +72,13 @@ dict_allchaps = df_allchaps.to_dict(orient='index')
 st.write("VVVVVVVVVVV: ", dict_allchaps)
 for chapter in dict_allchaps.get("Chapter"): 
     st.write(chapter)
+
+
+
+
+
+
+    
 selected_rows_list = [] #dict_allchaps
 if len(selected_rows_list) == 0: 
     st.warning("Please select at least one chapter")