def initialize_all_session_state()

in gemini/sample-apps/accelerating_product_innovation/app/pages_utils/setup.py [0:0]

• 78 lines of code
• 7 McCabe index (conditional complexity)

def initialize_all_session_state(reinitialize: bool = False) -> None:
    """Initializes all the session states used in the app.

    Args:
        reinitialize (optional, bool):
            Indicated if the session state is being reinitialized
            or being initialized for the first time.
            (This value is important to indicate that the value of
            the selected project has been updated. If it is set to false
            then no modification is made to the session state).

    Returns:
        None
    """
    # Get lists of projects in the application.
    project_list_blob = bucket.blob("project_list.txt")
    project_list = json.loads(project_list_blob.download_as_string())

    # Initialize default values for the session state.
    session_state_defaults: dict[str, Any] = {
        "product_categories": project_list,
        "new_product_category_added": None,
        "previous_product_category": None,
        "text_edit_prompt": None,
        "headers": {"Content-Type": "application/json"},
        "update_text_btn": None,
        "uploaded_files": None,
        "rag_search_term": None,
        "rag_answers_gen": False,
        "rag_answer": None,
        "rag_answer_references": None,
        "insights_suggestion": None,
        "insights_placeholder": "",
        "suggestion_first_time": 1,
        "processed_data_list": [],
        "query_vectors": [],
        "embeddings_df": None,
        "temp_suggestions": None,
        "assorted_prod_title": None,
        "assorted_prod_content": [],
        "email_gen": False,
        "create_product": False,
        "modifying": False,
        "custom_prompt": "",
        "feature_suggestions": None,
        "selected_titles": [],
        "saved_titles": [],
        "selected_prompt": None,
        "product_gen_image": None,
        "features_generated": False,
        "generated_points": None,
        "content_generated": False,
        "product_content": None,
        "image_to_edit": -1,
        "generate_images": False,
        "image_prompt": None,
        "image_file_prefix": "uploaded_image",
        "email_image": None,
        "email_prompt": "High SPF",
        "num_drafts": None,
        "email_text": None,
        "generated_image": None,
        "mask_image": None,
        "edit_suggestion": False,
        "suggested_images": None,
        "uploaded_img": False,
        "start_editing": False,
        "text_to_edit": None,
        "content_edited": None,
        "row": None,
        "edited_content": None,
        "col": None,
        "generated_response": None,
        "draft_elements": None,
        "chosen_titles": [],
        "buffer": None,
        "save_edited_image": None,
        "email_files": [],
        "image_edit_col": None,
        "image_edit_row": None,
        "bg_editing": False,
    }

    for key, value in session_state_defaults.items():
        if (
            reinitialize is False and key not in st.session_state
        ) or reinitialize is True:
            st.session_state[key] = value

    if "product_category" not in st.session_state:
        st.session_state.product_category = st.session_state.product_categories[0]

    st.session_state.generation_config = generative_models.GenerationConfig(
        max_output_tokens=8192,
        temperature=0.001,
        top_p=1,
    )