Improve Streamlit sidebar collapse and processing state flow

app.py

@@ -15,6 +15,7 @@ if str(_PROJECT_ROOT) not in sys.path:
 
 import numpy as np
 import streamlit as st
+import streamlit.components.v1 as components
 
 from photo_editor.config import get_settings
 from photo_editor.images import dng_to_rgb
@@ -46,9 +47,66 @@ def _load_original_for_display(image_path: Path):
     return str(path)
 
 
+def _collapse_sidebar() -> None:
+    """Collapse Streamlit sidebar via whichever toggle exists in this version."""
+    components.html(
+        """
+<script>
+const doc = window.parent.document;
+
+function collapseIfOpen() {
+  // Newer Streamlit versions expose a dedicated collapse button when sidebar is open.
+  const closeBtn =
+    doc.querySelector('[data-testid="stSidebarCollapseButton"]') ||
+    doc.querySelector('button[aria-label="Close sidebar"]');
+  if (closeBtn) {
+    closeBtn.click();
+    return true;
+  }
+  return false;
+}
+
+// Try immediately, then briefly retry in case elements mount after rerun.
+if (!collapseIfOpen()) {
+  let tries = 0;
+  const interval = setInterval(() => {
+    tries += 1;
+    if (collapseIfOpen() || tries > 20) {
+      clearInterval(interval);
+    }
+  }, 100);
+}
+</script>
+""",
+        height=0,
+        width=0,
+    )
+
+
 
 def main() -> None:
     st.set_page_config(page_title="LumiGrade AI", page_icon="📷", layout="wide")
+
+    if "sidebar_collapsed" not in st.session_state:
+        st.session_state["sidebar_collapsed"] = False
+    if "run_pipeline_requested" not in st.session_state:
+        st.session_state["run_pipeline_requested"] = False
+    if "selected_image_path" not in st.session_state:
+        st.session_state["selected_image_path"] = ""
+    if "is_processing" not in st.session_state:
+        st.session_state["is_processing"] = False
+    if "refresh_after_run" not in st.session_state:
+        st.session_state["refresh_after_run"] = False
+
+    collapse_css = ""
+    if st.session_state["sidebar_collapsed"]:
+        collapse_css = """
+/* Force-hide sidebar after run is triggered; avoids version-specific JS toggles. */
+[data-testid="stSidebar"] {
+  display: none !important;
+}
+"""
+
     st.markdown(
         """
 <style>
@@ -120,6 +178,9 @@ h1 {
 [data-testid="stSidebar"] [data-testid="stSidebarContent"] {
   padding-top: 0 !important;
 }
+"""
+        + collapse_css
+        + """
 </style>
 """,
         unsafe_allow_html=True,
@@ -127,6 +188,10 @@ h1 {
 
     st.title("📷 LumiGrade AI")
     st.caption("Upload an image to get expert-informed edit recommendations and an instant enhanced result.")
+    if st.session_state["sidebar_collapsed"]:
+        if st.button("⚙️ Show Inputs", key="show_inputs_btn", disabled=st.session_state["is_processing"]):
+            st.session_state["sidebar_collapsed"] = False
+            st.rerun()
 
     # Config check
     s = get_settings()
@@ -142,7 +207,7 @@ h1 {
     # control and wire it to this flag.
     use_editing_api = False
 
-    image_path = None
+    image_path = Path(st.session_state["selected_image_path"]) if st.session_state["selected_image_path"] else None
 
     with st.sidebar:
         # Reliable spacing so only the Pipeline Inputs card moves down.
@@ -152,7 +217,7 @@ h1 {
             uploaded = st.file_uploader(
                 "Upload JPEG, PNG, DNG, HEIC, or HEIF",
                 type=["jpg", "jpeg", "png", "dng", "heic", "heif"],
-                help="You can upload DNG/HEIC directly, or use a local path below.",
+                help="Upload JPEG/PNG/DNG/HEIC/HEIF to run the edit recommendation pipeline.",
             )
             if uploaded is not None:
                 suffix = Path(uploaded.name).suffix.lower()
@@ -168,25 +233,30 @@ h1 {
                     target = _STREAMLIT_INPUT_JPG_PATH
                 target.write_bytes(uploaded.getvalue())
                 image_path = target
-            else:
-                path_str = st.text_input(
-                    "Or enter path to image (e.g. for DNG)",
-                    placeholder="/path/to/image.dng or image.jpg",
-                )
-                path_str = (path_str or "").strip()
-                if path_str:
-                    p = Path(path_str)
-                    if p.exists():
-                        image_path = p
-                    else:
-                        st.warning("File not found. Use a full path that exists, or upload a file.")
-
-            run_clicked = st.button("▶ Run Pipeline", type="primary", use_container_width=True)
+                st.session_state["selected_image_path"] = str(target)
+
+            run_clicked = st.button(
+                "✨ Generate Edit Recommendations",
+                type="primary",
+                use_container_width=True,
+                disabled=st.session_state["is_processing"],
+            )
             status = st.empty()
             if image_path is None:
                 status.info("Provide an image to run.")
 
-    if run_clicked and image_path is not None:
+            if run_clicked and image_path is not None:
+                # Mark busy before rerun so any control rendered on next pass is disabled.
+                st.session_state["is_processing"] = True
+                st.session_state["sidebar_collapsed"] = True
+                st.session_state["run_pipeline_requested"] = True
+                st.rerun()
+    should_run_pipeline = st.session_state.pop("run_pipeline_requested", False)
+    if should_run_pipeline and st.session_state["selected_image_path"]:
+        image_path = Path(st.session_state["selected_image_path"])
+    if should_run_pipeline and image_path is not None:
+        st.session_state["is_processing"] = True
+        _collapse_sidebar()
         loading_box = st.empty()
 
         def _render_loading(current_stage: str, state: str = "running") -> None:
@@ -264,6 +334,19 @@ h1 {
             st.session_state.pop("pipeline_input_path", None)
             st.session_state.pop("pipeline_current_params", None)
             _render_loading("consulting", "failed")
+        finally:
+            st.session_state["is_processing"] = False
+            # Button states are computed at render time; rerun once so controls
+            # immediately reflect processing completion (re-enable Show Inputs).
+            st.session_state["refresh_after_run"] = True
+
+    if st.session_state.get("refresh_after_run"):
+        st.session_state["refresh_after_run"] = False
+        st.rerun()
+
+    display_input_path = image_path
+    if display_input_path is None and st.session_state.get("pipeline_input_path"):
+        display_input_path = Path(st.session_state["pipeline_input_path"])
 
     with st.container(border=True):
         st.subheader("Results Dashboard")
@@ -271,9 +354,9 @@ h1 {
 
         # Show result if available
         if (
-            image_path is not None
+            display_input_path is not None
             and st.session_state.get("pipeline_result")
-            and st.session_state.get("pipeline_input_path") == str(image_path)
+            and st.session_state.get("pipeline_input_path") == str(display_input_path)
         ):
             result = st.session_state["pipeline_result"]
             out_path = st.session_state["pipeline_output_path"]
@@ -331,9 +414,9 @@ h1 {
 
     # Keep this full-width and at the bottom, per request.
     if (
-        image_path is not None
+        display_input_path is not None
         and st.session_state.get("pipeline_result")
-        and st.session_state.get("pipeline_input_path") == str(image_path)
+        and st.session_state.get("pipeline_input_path") == str(display_input_path)
     ):
         result = st.session_state["pipeline_result"]
         out_path = st.session_state["pipeline_output_path"]
@@ -342,7 +425,7 @@ h1 {
             st.subheader("Original vs Result")
             col_orig, col_result = st.columns(2)
             with col_orig:
-                st.image(_load_original_for_display(image_path), caption="Original", use_container_width=True)
+                st.image(_load_original_for_display(display_input_path), caption="Original", use_container_width=True)
             with col_result:
                 st.image(str(out_path), caption="Edited", use_container_width=True)