Update requirements and refactor interface components

- Upgrade package versions in requirements.txt
- Improve import and formatting in interface/components.py
- Minor code style and readability improvements

Signed-off-by: Unai Garay <[email protected]>

interface/components.py

@@ -1,11 +1,12 @@
 import streamlit as st
+
+from interface.draw_pipelines import get_pipeline_graph
 from interface.utils import (
-    get_pipelines,
-    extract_text_from_url,
     extract_text_from_file,
+    extract_text_from_url,
+    get_pipelines,
     reset_vars_data,
 )
-from interface.draw_pipelines import get_pipeline_graph
 
 
 def component_select_pipeline(container):
@@ -15,9 +16,11 @@ def component_select_pipeline(container):
             selected_pipeline = st.selectbox(
                 "Select pipeline",
                 pipeline_names,
-                index=pipeline_names.index("Keyword Search")
-                if "Keyword Search" in pipeline_names
-                else 0,
+                index=(
+                    pipeline_names.index("Keyword Search")
+                    if "Keyword Search" in pipeline_names
+                    else 0
+                ),
             )
             index_pipe = pipeline_names.index(selected_pipeline)
             st.write("---")
@@ -41,7 +44,10 @@ def component_select_pipeline(container):
                 != list(pipeline_func_parameters[index_pipe].values())
             ):
                 st.session_state["pipeline_func_parameters"] = pipeline_func_parameters
-                (search_pipeline, index_pipeline,) = pipeline_funcs[
+                (
+                    search_pipeline,
+                    index_pipeline,
+                ) = pipeline_funcs[
                     index_pipe
                 ](**pipeline_func_parameters[index_pipe])
                 st.session_state["pipeline"] = {
@@ -55,7 +61,10 @@ def component_select_pipeline(container):
             # Reload if Keyword Search is selected
             elif st.session_state["pipeline"]["name"] == "Keyword Search":
                 st.session_state["pipeline_func_parameters"] = pipeline_func_parameters
-                (search_pipeline, index_pipeline,) = pipeline_funcs[
+                (
+                    search_pipeline,
+                    index_pipeline,
+                ) = pipeline_funcs[
                     index_pipe
                 ](**pipeline_func_parameters[index_pipe])
                 st.session_state["pipeline"] = {
@@ -141,9 +150,9 @@ def component_file_input(container, doc_id):
                 file = st.file_uploader(
                     "Upload a .txt, .pdf, .csv, image file, audio file", key=doc_id
                 )
-                if file != None:
+                if file is not None:
                     extracted_text = extract_text_from_file(file)
-                    if extracted_text != None:
+                    if extracted_text is not None:
                         files.append({"text": extracted_text, "doc_id": doc_id})
                         doc_id += 1
                         st.markdown("---")

requirements.txt

@@ -1,14 +1,14 @@
-streamlit==1.12.2
-streamlit_option_menu==0.3.2
-farm-haystack==1.8.0
-black==22.8.0
-plotly==5.10.0
+streamlit==1.40.1
+streamlit_option_menu==0.4.0
+farm-haystack==1.26.4
+black==24.8.0
+plotly==5.24.1
 newspaper3k==0.2.8
-PyPDF2==2.10.7
-pytesseract==0.3.10
-soundfile==0.10.3.post1
-espnet
+PyPDF2==3.0.1
+pytesseract==0.3.13
+soundfile==0.13.1
+espnet==202412
 pydub==0.25.1
 espnet_model_zoo==0.1.7
-git+https://github.com/openai/whisper.git
-altair>=4.0.0
+openai-whisper==20240930
+altair==5.4.1