def choose_document()

in obelics/visualization/web_document_and_filtering_visualization.py [0:0]

• 21 lines of code
• 3 McCabe index (conditional complexity)

    def choose_document(self):
        if self.dataset:
            st.header("Choose a document")
            if st.button("Select a random document"):
                dct_idx = random.randint(a=0, b=self.dataset.num_rows - 1)
            else:
                dct_idx = 0
            idx = st.number_input(
                f"Select a document among the first {self.dataset.num_rows} ones",
                min_value=0,
                max_value=self.dataset.num_rows - 1,
                value=dct_idx,
                step=1,
                help=f"Index between 0 and {self.dataset.num_rows-1}",
            )
            self.current_doc = self.dataset[idx]
            original_idx = self.current_doc["original_idx"]
            st.markdown(f"Original Document Id: {original_idx}")
            self.current_doc_original = self.full_dataset[original_idx]
        else:
            self.current_doc = None