bertrend 0.3.18__tar.gz → 0.3.20__tar.gz

{bertrend-0.3.18 → bertrend-0.3.20}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: bertrend
-Version: 0.3.18
+Version: 0.3.20
 Summary: BERTrend analyses topic evolution over time using state-of-the-art transformer models: topic modelling and LLM
 License: LICENSE.md
 License-File: AUTHORS.txt
 License-File: LICENSE.md
-Keywords: topics,BERTopic,transformers,weak signals,LLM
+Keywords: topics,BERTopic,transformers,weak signals,LLM,trend analysis
 Author: Jérôme Picault
 Author-email: jerome.picault@rte-france.com
 Maintainer: Jérôme Picault
@@ -44,7 +44,7 @@ Requires-Dist: newspaper4k (>=0.9.3.1) ; extra == "apps"
 Requires-Dist: nltk (>=3.9.1)
 Requires-Dist: numpy (<2)
 Requires-Dist: openai (>=1.77.0)
-Requires-Dist: openai-agents[litellm] (>=0.2.5)
+Requires-Dist: openai-agents[litellm] (>=0.4.0)
 Requires-Dist: openpyxl (>=3.1.5)
 Requires-Dist: opentelemetry-exporter-otlp-proto-grpc (==1.25.0)
 Requires-Dist: pandas (>=2.2.2)
@@ -55,6 +55,7 @@ Requires-Dist: pygooglenews (>=0.1.3) ; extra == "apps"
 Requires-Dist: pylabeladjust (==0.1.13)
 Requires-Dist: pytest (>=8.4.0)
 Requires-Dist: pytest-asyncio
+Requires-Dist: python-dotenv (>=1.1.1)
 Requires-Dist: requests (>=2.32.3)
 Requires-Dist: scikit-learn (>=1.5.2)
 Requires-Dist: scipy (==1.12)
@@ -62,7 +63,7 @@ Requires-Dist: seaborn (>=0.13.2)
 Requires-Dist: sentence-transformers (==3.0.1)
 Requires-Dist: sentencepiece (>=0.2.0)
 Requires-Dist: starlette (>=0.46.2)
-Requires-Dist: streamlit (>=1.48)
+Requires-Dist: streamlit (>=1.50)
 Requires-Dist: streamlit-code-editor (>=0.1.22)
 Requires-Dist: supervisor (>=4.3.0)
 Requires-Dist: thefuzz (>=0.22.1)
@@ -150,15 +151,36 @@ https://github.com/user-attachments/assets/d79368d9-d4e0-4324-8a98-a888f0ab3b65
 
 ## Installation
 
-### Environment Variables
-
-BERTrend requires the following environment variables to be set:
-
-- `BERTREND_BASE_DIR`: the home directory for BERTrend data, models, and logs
-- OpenAI variables (for LLM integration):
-  - `OPENAI_API_KEY`: the API key for your LLM deployment
-  - `OPENAI_ENDPOINT`: the endpoint for your LLM deployment (empty if using OpenAI services)
-  - `OPENAI_DEFAULT_MODEL_NAME`: the name of the default LLM model to use
+### Environment Setup (.env and .venv)
+
+BERTrend now supports configuration via a repository-level `.env` file. On import, the package attempts to load environment variables automatically using `python-dotenv`.
+
+Recommended steps:
+
+1. Create and fill your `.env` at the repo root (a template is provided):
+   - Copy `.env` and set your values (do not commit secrets).
+   - Common keys:
+     - `BERTREND_BASE_DIR`: base directory for BERTrend data/models/logs
+     - OpenAI/LLM:
+       - `OPENAI_API_KEY`
+       - `OPENAI_BASE_URL` (optional for OpenAI-compatible providers such as LiteLLM and Azure)
+       - `OPENAI_DEFAULT_MODEL`
+     - Providers (optional): `SEMANTIC_SCHOLAR_API_KEY`, `NEWSCATCHER_API_KEY`, `DBPEDIA_REST_API_URL`
+     - Embedding service security (optional): `BERTREND_SECRET_KEY`, `DEFAULT_RATE_LIMIT`, `DEFAULT_RATE_WINDOW`, `CLIENT_REGISTRY_FILE`
+     - Email (optional): `BERTREND_EMAIL_BACKEND`, `GMAIL_*`, `SMTP_*`, `SENDGRID_API_KEY`
+     - Misc: `CUDA_VISIBLE_DEVICES`
+2. Use a virtual environment and install dependencies inside it:
+   ```bash
+   python -m venv .venv
+   source .venv/bin/activate  # Windows: .venv\\Scripts\\activate
+   pip install -U pip
+   pip install .  # or pip install ".[apps]"
+   pip install python-dotenv  # ensures .env is auto-loaded
+   ```
+3. Notes
+   - If `python-dotenv` isn’t installed, `.env` won’t be auto-loaded; you can either install it or export variables via your shell/runner.
+   - Scheduled jobs created from the apps use the exact interpreter of your active environment (`sys.executable`), ensuring `.venv/bin/python` is used. Keep your venv path stable on the machine running cron.
+   - Cron entries source your shell profile and pass through any required variables. You can also rely on `.env` for most settings.
 
 ### Hardware and Software Requirements
 

{bertrend-0.3.18 → bertrend-0.3.20}/README.md

@@ -69,15 +69,36 @@ https://github.com/user-attachments/assets/d79368d9-d4e0-4324-8a98-a888f0ab3b65
 
 ## Installation
 
-### Environment Variables
-
-BERTrend requires the following environment variables to be set:
-
-- `BERTREND_BASE_DIR`: the home directory for BERTrend data, models, and logs
-- OpenAI variables (for LLM integration):
-  - `OPENAI_API_KEY`: the API key for your LLM deployment
-  - `OPENAI_ENDPOINT`: the endpoint for your LLM deployment (empty if using OpenAI services)
-  - `OPENAI_DEFAULT_MODEL_NAME`: the name of the default LLM model to use
+### Environment Setup (.env and .venv)
+
+BERTrend now supports configuration via a repository-level `.env` file. On import, the package attempts to load environment variables automatically using `python-dotenv`.
+
+Recommended steps:
+
+1. Create and fill your `.env` at the repo root (a template is provided):
+   - Copy `.env` and set your values (do not commit secrets).
+   - Common keys:
+     - `BERTREND_BASE_DIR`: base directory for BERTrend data/models/logs
+     - OpenAI/LLM:
+       - `OPENAI_API_KEY`
+       - `OPENAI_BASE_URL` (optional for OpenAI-compatible providers such as LiteLLM and Azure)
+       - `OPENAI_DEFAULT_MODEL`
+     - Providers (optional): `SEMANTIC_SCHOLAR_API_KEY`, `NEWSCATCHER_API_KEY`, `DBPEDIA_REST_API_URL`
+     - Embedding service security (optional): `BERTREND_SECRET_KEY`, `DEFAULT_RATE_LIMIT`, `DEFAULT_RATE_WINDOW`, `CLIENT_REGISTRY_FILE`
+     - Email (optional): `BERTREND_EMAIL_BACKEND`, `GMAIL_*`, `SMTP_*`, `SENDGRID_API_KEY`
+     - Misc: `CUDA_VISIBLE_DEVICES`
+2. Use a virtual environment and install dependencies inside it:
+   ```bash
+   python -m venv .venv
+   source .venv/bin/activate  # Windows: .venv\\Scripts\\activate
+   pip install -U pip
+   pip install .  # or pip install ".[apps]"
+   pip install python-dotenv  # ensures .env is auto-loaded
+   ```
+3. Notes
+   - If `python-dotenv` isn’t installed, `.env` won’t be auto-loaded; you can either install it or export variables via your shell/runner.
+   - Scheduled jobs created from the apps use the exact interpreter of your active environment (`sys.executable`), ensuring `.venv/bin/python` is used. Keep your venv path stable on the machine running cron.
+   - Cron entries source your shell profile and pass through any required variables. You can also rely on `.env` for most settings.
 
 ### Hardware and Software Requirements
 

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/BERTopicModel.py

@@ -177,7 +177,7 @@ class BERTopicModel:
         return OpenAI(
             client=OpenAI_Client(
                 api_key=LLM_CONFIG["api_key"],
-                endpoint=LLM_CONFIG["endpoint"],
+                base_url=LLM_CONFIG["base_url"],
                 model=LLM_CONFIG["model"],
             ).llm_client,
             model=LLM_CONFIG["model"],

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/__init__.py

@@ -5,6 +5,19 @@
 import os
 from pathlib import Path
 
+from loguru import logger
+
+# Auto-load .env if python-dotenv is available
+try:
+    from dotenv import load_dotenv  # type: ignore
+
+    if load_dotenv(override=True):
+        logger.info("Loaded .env file")
+    else:
+        logger.warning("Failed to load .env file")
+except Exception:
+    pass
+
 from bertrend.utils.config_utils import load_toml_config
 
 # Default config files path

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/config/services_default_config.toml

@@ -24,9 +24,9 @@ embedding_max_seq_length = 512
 # - local LLM deployed with VLLM or similar libraries
 # - the official OpenAI endpoint
 # - models deployed on Azure cloud
-api_key = "$AZURE_SE_WATTELSE_OPENAI_API_KEY_DEV"
-endpoint = "$AZURE_SE_WATTELSE_OPENAI_ENDPOINT_DEV"
-model = "$AZURE_SE_WATTELSE_OPENAI_DEFAULT_MODEL_NAME_DEV"
+api_key = "$OPENAI_API_KEY"
+base_url = "$OPENAI_BASE_URL"
+model = "$OPENAI_DEFAULT_MODEL"
 temperature = 0.1
 max_output_tokens = 2048
 system_prompt = "You are a helpful assistant, skilled in detailing topic evolution over time for the detection of emerging trends and signals."

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/demos_utils/data_loading_component.py

@@ -338,5 +338,5 @@ def display_data_loading_component():
         SessionStateManager.get_dataframe("time_filtered_df")[
             [TEXT_COLUMN, "timestamp"]
         ],
-        use_container_width=True,
+        width="stretch",
     )

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/demos_utils/icons.py

@@ -21,6 +21,7 @@ TOPIC_VISUALIZATION_ICON = ":material/monitoring:"
 TEMPORAL_VISUALIZATION_ICON = ":material/timelapse:"
 NEWSLETTER_ICON = ":material/newspaper:"
 ANALYSIS_ICON = ":material/query_stats:"
+TIMELINE_ICON = ":material/timeline:"
 DATA_LOADING_ICON = ":material/drive_folder_upload:"
 MODEL_TRAINING_ICON = ":material/cognition:"
 SERVER_STORAGE_ICON = ":material/database:"
@@ -35,6 +36,10 @@ DOWNLOAD_ICON = ":material/download:"
 EMAIL_ICON = ":material/mail:"
 RESTART_ICON = ":material/restart_alt:"
 PERSON_ADD_ICON = ":material/person_add:"
+PERSON_REMOVE_ICON = ":material/person_remove:"
+POPULARITY_ICON = ":material/thumbs_up_down:"
+SIGNAL_EVOLUTION_ICON = ":material/signal_cellular_alt:"
+CYCLE_ICON = ":material/cycle:"
 
 JSON_ICON = "🧾"
 PARQUET_ICON = "📦️"

bertrend-0.3.20/bertrend/demos/start_summarizer_app.sh

@@ -0,0 +1,9 @@
+#
+# Copyright (c) 2024, RTE (https://www.rte-france.com)
+# See AUTHORS.txt
+# SPDX-License-Identifier: MPL-2.0
+# This file is part of BERTrend.
+#
+
+# Starts the Summarizer app (a comparison of various method of text summarization)
+CUDA_VISIBLE_DEVICES=0 OPENAI_API_KEY=$OPENAI_API_KEY OPENAI_BASE_URL=$OPENAI_BASE_URL OPENAI_DEFAULT_MODEL=$OPENAI_DEFAULT_MODEL streamlit run --theme.primaryColor royalblue summarization/summarizer_app.py

bertrend-0.3.20/bertrend/demos/streamlit_components/input_with_pills_component.py

@@ -0,0 +1,101 @@
+import streamlit as st
+from typing import Callable
+
+from bertrend.demos.demos_utils.icons import PERSON_ADD_ICON
+
+
+def input_with_pills(
+    label: str,
+    key: str,
+    placeholder: str = "",
+    validate_fn: Callable[[str], bool] | None = None,
+    label_visibility: str = "hidden",
+    help: str = None,
+    value: list[str] = None,
+) -> list[str] | None:
+    """
+    A simple Streamlit component that auto-adds valid input to pills.
+    Dialog-safe version that doesn't use st.rerun()
+    """
+
+    # Define all keys
+    all_items_key = f"{key}_all_items"
+    input_key = f"{key}_input"
+    pills_widget_key = f"{key}_pills_widget"
+    selection_tracking_key = f"{key}_selection_tracking"
+
+    if all_items_key not in st.session_state:
+        initial_value = value if value is not None else []
+        st.session_state[all_items_key] = initial_value
+        st.session_state[selection_tracking_key] = initial_value[:]
+
+    if input_key not in st.session_state:
+        st.session_state[input_key] = ""
+
+    col1, col2 = st.columns([3, 3])
+
+    with col1:
+        with st.form(key=f"{key}_form", clear_on_submit=True, border=False):
+            current_input = st.text_input(
+                label,
+                placeholder=placeholder,
+                key=input_key,
+                label_visibility=label_visibility,
+                help=help,
+            )
+            submitted = st.form_submit_button(PERSON_ADD_ICON)
+
+            if submitted and current_input.strip():
+                is_valid = True
+                error_msg = ""
+                stripped_input = current_input.strip()
+
+                if validate_fn and not validate_fn(stripped_input):
+                    is_valid = False
+                    error_msg = "Invalid format"
+                elif stripped_input in st.session_state[all_items_key]:
+                    is_valid = False
+                    error_msg = "Already exists"
+
+                if is_valid:
+                    # Add to the list of all items
+                    st.session_state[all_items_key].append(stripped_input)
+
+                    # Update tracking selection
+                    st.session_state[selection_tracking_key].append(stripped_input)
+
+                    # Update pills widget selection
+                    st.session_state[pills_widget_key] = st.session_state[
+                        selection_tracking_key
+                    ][:]
+                else:
+                    st.error(error_msg)
+
+    with col2:
+        # Display pills
+        if st.session_state[all_items_key]:
+            st.markdown("<div style='margin-top: 8px;'></div>", unsafe_allow_html=True)
+
+            # Initialize pills widget if not exists
+            if pills_widget_key not in st.session_state:
+                st.session_state[pills_widget_key] = st.session_state[
+                    selection_tracking_key
+                ][:]
+
+            def on_pills_change():
+                # Sync the widget selection to our tracking key
+                st.session_state[selection_tracking_key] = st.session_state[
+                    pills_widget_key
+                ][:]
+
+            selected = st.pills(
+                "Items",
+                st.session_state[all_items_key],
+                selection_mode="multi",
+                key=pills_widget_key,
+                label_visibility="collapsed",
+                on_change=on_pills_change,
+            )
+
+    st.session_state[key] = st.session_state[selection_tracking_key]
+    return st.session_state[key]

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/summarization/summarizer_app.py

@@ -3,8 +3,8 @@
 #  SPDX-License-Identifier: MPL-2.0
 #  This file is part of BERTrend.
 import os
-
 import streamlit as st
+from dotenv import load_dotenv
 from loguru import logger
 
 from bertrend.demos.demos_utils.icons import WARNING_ICON
@@ -15,6 +15,9 @@ from bertrend.services.summary.extractive_summarizer import (
     ExtractiveSummarizer,
 )
 
+load_dotenv(override=True)
+
+
 SUMMARIZER_OPTIONS_MAPPER = {
     "GPTSummarizer": GPTSummarizer,
     "EnhancedExtractiveSummarizer": EnhancedExtractiveSummarizer,
@@ -39,7 +42,7 @@ def get_summarizer(summary_model):
     kwargs = {}
     if summary_model == "GPTSummarizer":
         kwargs["api_key"] = st.session_state.openai_api_key
-        kwargs["endpoint"] = st.session_state.openai_endpoint
+        kwargs["base_url"] = st.session_state.openai_base_url
 
     summarizer_class = SUMMARIZER_OPTIONS_MAPPER[summary_model]
     return summarizer_class(**kwargs)
@@ -63,6 +66,7 @@ def app():
         )
         / 100
     )
+
     st.text_input(
         "Openai API key",
         key="openai_api_key",
@@ -70,12 +74,12 @@ def app():
         type="password",
     )
     st.text_input(
-        "Openai endpoint", key="openai_endpoint", value=os.getenv("OPENAI_ENDPOINT")
+        "Openai base URL", key="openai_base_url", value=os.getenv("OPENAI_BASE_URL")
     )
     st.text_input(
         "Openai model name",
         key="openai_model_name",
-        value=os.getenv("OPENAI_DEFAULT_MODEL_NAME"),
+        value=os.getenv("OPENAI_DEFAULT_MODEL"),
     )
 
     col1, col2 = st.columns(2)

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/topic_analysis/demo_pages/explore_topics.py

@@ -87,7 +87,7 @@ def display_sidebar():
                 button_title += f" :red[+{new_docs_number}]"
             st.button(
                 button_title,
-                use_container_width=True,
+                width="stretch",
                 on_click=set_topic_selection,
                 args=(topic_number,),
             )
@@ -119,7 +119,7 @@ def plot_topic_over_time():
                     st.session_state["topic_model"],
                 ),
                 config=PLOTLY_BUTTON_SAVE_CONFIG,
-                use_container_width=True,
+                width="stretch",
             )
         else:
             st.plotly_chart(
@@ -132,7 +132,7 @@ def plot_topic_over_time():
                     ].max(),
                 ),
                 config=PLOTLY_BUTTON_SAVE_CONFIG,
-                use_container_width=True,
+                width="stretch",
             )
 
 
@@ -203,7 +203,7 @@ def display_source_distribution(
         showlegend=False, height=600, width=500, margin=dict(t=0, b=0, l=0, r=0)
     )
 
-    st.plotly_chart(fig, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True)
+    st.plotly_chart(fig, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch")
 
 
 def get_website_name(url):

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/topic_analysis/demo_pages/newsletters_generation.py

@@ -158,7 +158,7 @@ def main():
         generate_newsletter_clicked = st.button(
             translate("generate_newsletter_button"),
             type="primary",
-            use_container_width=True,
+            width="stretch",
         )
 
     # Generate newsletters when button is clicked

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/topic_analysis/demo_pages/temporal_visualizations.py

@@ -340,7 +340,7 @@ def display_topic_evolution_dataframe():
             st.session_state.temptopic.final_df[columns_present].sort_values(
                 by=["Topic", "Timestamp"], ascending=[True, True]
             ),
-            use_container_width=True,
+            width="stretch",
         )
 
 
@@ -349,14 +349,14 @@ def display_topic_info_dataframe():
     with st.expander(translate("topic_info_dataframe")):
         st.dataframe(
             st.session_state.temptopic.topic_model.get_topic_info(),
-            use_container_width=True,
+            width="stretch",
         )
 
 
 def display_documents_per_date_dataframe():
     """Display the Documents per Date Dataframe."""
     with st.expander(translate("documents_per_date_dataframe")):
-        st.dataframe(st.session_state.aggregated_df, use_container_width=True)
+        st.dataframe(st.session_state.aggregated_df, width="stretch")
 
 
 def display_temptopic_visualizations():
@@ -423,7 +423,7 @@ def display_topic_evolution(topics_to_show):
         color_palette=color_palette,
     )
     st.plotly_chart(
-        fig_topic_evolution, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True
+        fig_topic_evolution, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch"
     )
 
     st.divider()
@@ -447,7 +447,7 @@ def display_overall_topic_stability(topics_to_show):
     st.plotly_chart(
         fig_overall_stability,
         config=PLOTLY_BUTTON_SAVE_CONFIG,
-        use_container_width=True,
+        width="stretch",
     )
 
     st.divider()
@@ -463,7 +463,7 @@ def display_temporal_stability_metrics(topics_to_show):
         topics_to_show=topics_to_show,
     )
     st.plotly_chart(
-        fig_topic_stability, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True
+        fig_topic_stability, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch"
     )
 
     fig_representation_stability = plot_temporal_stability_metrics(
@@ -474,7 +474,7 @@ def display_temporal_stability_metrics(topics_to_show):
     st.plotly_chart(
         fig_representation_stability,
         config=PLOTLY_BUTTON_SAVE_CONFIG,
-        use_container_width=True,
+        width="stretch",
     )
 
 
@@ -511,7 +511,7 @@ def display_topics_popularity():
                         st.session_state["topic_model"],
                     ),
                     config=PLOTLY_BUTTON_SAVE_CONFIG,
-                    use_container_width=True,
+                    width="stretch",
                 )
 
 

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/topic_analysis/demo_pages/topic_visualizations.py

@@ -231,7 +231,7 @@ def main():
             st.plotly_chart(
                 overall_results_plot,
                 config=PLOTLY_BUTTON_SAVE_CONFIG,
-                use_container_width=True,
+                width="stretch",
             )
         else:
             st.error(translate("overall_results_display_error"), icon=ERROR_ICON)
@@ -242,7 +242,7 @@ def main():
         with st.spinner(translate("topics_treemap_computation")):
             treemap_plot = create_treemap()
             st.plotly_chart(
-                treemap_plot, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True
+                treemap_plot, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch"
             )
 
     # Data Map
@@ -263,7 +263,7 @@ def main():
                     data=get_binary_file_downloader_html(save_path),
                     file_name="datamapplot.html",
                     mime="text/html",
-                    use_container_width=True,
+                    width="stretch",
                     type="secondary",
                 )
             else:

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/topic_analysis/demo_pages/training_page.py

@@ -70,7 +70,7 @@ def data_distribution(df: pd.DataFrame):
             value="1M",
         )
         fig = plot_docs_repartition_over_time(df, freq)
-        st.plotly_chart(fig, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True)
+        st.plotly_chart(fig, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch")
 
 
 def save_model_interface():

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/weak_signals/app.py

@@ -195,7 +195,7 @@ def training_page():
                 selected_docs[
                     ["timestamp", TEXT_COLUMN, "document_id", "source", "url"]
                 ],
-                use_container_width=True,
+                width="stretch",
             )
         else:
             st.warning(translate("no_data_warning"), icon=WARNING_ICON)
@@ -285,13 +285,13 @@ def analysis_page():
             st.plotly_chart(
                 plot_num_topics(topic_models),
                 config=PLOTLY_BUTTON_SAVE_CONFIG,
-                use_container_width=True,
+                width="stretch",
             )
             # Size of Outlier Topic for each topic model
             st.plotly_chart(
                 plot_size_outliers(topic_models),
                 config=PLOTLY_BUTTON_SAVE_CONFIG,
-                use_container_width=True,
+                width="stretch",
             )
 
         display_topics_per_timestamp(topic_models)
@@ -335,7 +335,7 @@ def analysis_page():
                 st.plotly_chart(
                     fig_trend,
                     config=PLOTLY_BUTTON_SAVE_CONFIG,
-                    use_container_width=True,
+                    width="stretch",
                 )
 
                 # Display the dataframe with zeroshot topics information
@@ -352,7 +352,7 @@ def analysis_page():
                     for timestamp, data in weak_signal_trend.items()
                 ]
                 zeroshot_topics_df = pd.DataFrame(zeroshot_topics_data)
-                st.dataframe(zeroshot_topics_df, use_container_width=True)
+                st.dataframe(zeroshot_topics_df, width="stretch")
 
                 # Save the zeroshot topics data to a JSON file
                 json_file_path = ZEROSHOT_TOPICS_DATA_DIR
@@ -462,7 +462,7 @@ def main():
         # State Management
         st.subheader(translate("state_management"))
 
-        if st.button(translate("restore_previous_run"), use_container_width=True):
+        if st.button(translate("restore_previous_run"), width="stretch"):
             restore_state()
             try:
                 SessionStateManager.set("bertrend", BERTrend.restore_model())
@@ -470,10 +470,10 @@ def main():
             except Exception:
                 st.warning(translate("no_models_warning"), icon=WARNING_ICON)
 
-        if st.button(translate("purge_cache"), use_container_width=True):
+        if st.button(translate("purge_cache"), width="stretch"):
             purge_cache()
 
-        if st.button(translate("clear_session_state"), use_container_width=True):
+        if st.button(translate("clear_session_state"), width="stretch"):
             SessionStateManager.clear()
 
         # BERTopic Hyperparameters

{bertrend-0.3.18 → bertrend-0.3.20}/bertrend/demos/weak_signals/visualizations_utils.py

@@ -75,7 +75,7 @@ def display_sankey_diagram(all_merge_histories_df: pd.DataFrame) -> None:
 
         # Display the diagram using Streamlit in an expander
         st.plotly_chart(
-            sankey_diagram, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True
+            sankey_diagram, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch"
         )
 
 
@@ -227,7 +227,7 @@ def display_popularity_evolution():
         q1,
         q3,
     )
-    st.plotly_chart(fig, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True)
+    st.plotly_chart(fig, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch")
 
 
 def display_signal_types():
@@ -309,7 +309,7 @@ def display_newly_emerged_topics(all_new_topics_df: pd.DataFrame) -> None:
             ].sort_values(by=["Timestamp", "Document_Count"], ascending=[True, False])
         )
         st.plotly_chart(
-            fig_new_topics, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True
+            fig_new_topics, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch"
         )
 
 
@@ -330,15 +330,15 @@ def display_topics_per_timestamp(topic_models: dict[pd.Timestamp, BERTopic]) ->
 
         fig = plot_topics_for_model(selected_model)
 
-        st.plotly_chart(fig, config=PLOTLY_BUTTON_SAVE_CONFIG, use_container_width=True)
+        st.plotly_chart(fig, config=PLOTLY_BUTTON_SAVE_CONFIG, width="stretch")
 
         st.dataframe(
             selected_model.doc_info_df[
                 ["Paragraph", "document_id", "Topic", "Representation", "source"]
             ],
-            use_container_width=True,
+            width="stretch",
         )
-        st.dataframe(selected_model.topic_info_df, use_container_width=True)
+        st.dataframe(selected_model.topic_info_df, width="stretch")
 
 
 def display_signal_analysis(topic_number: int):