Update app_pages/text_image_generation.py

app_pages/text_image_generation.py

@@ -10,6 +10,7 @@ from generator_function.image_function import (
 from util.session_state import current_uid
 
 def render_multimodel_ad_generator(prefix: str = "img_text"):
+    private_mode = st.session_state.get("private_mode_enabled", False)
     colA, colB = st.columns([1, 1])
 
     with colA:
@@ -29,6 +30,9 @@ def render_multimodel_ad_generator(prefix: str = "img_text"):
     with colB:
         prompt = st.text_area("Prompt", placeholder="Describe the image you want to generate...", height=160, key=f"{prefix}_prompt")
 
+    if private_mode:
+        st.info("Private mode is on. Generations stay in-session only and will not reach the AI Library or external storage.")
+
     if st.button("Generate Images", type="primary", key=f"{prefix}_go"):
         uid = current_uid()
         if not uid:
@@ -42,4 +46,5 @@ def render_multimodel_ad_generator(prefix: str = "img_text"):
             num_images=num_images,
             category=(category.strip() or None),
             platform=platform or None,
+            private_mode=private_mode,
         )