Update app.py

app.py

@@ -62,6 +62,13 @@ if "pending_prompt" not in st.session_state:
 tab1, tab2 = st.tabs(["💬 Chat Assistant", "🖼️ Visual Reference Search"])
 
 # ------------------ Tab 1 ------------------
+
+# ✅ Ensure required keys exist
+if "image_url" not in st.session_state:
+    st.session_state.image_url = None
+if "image_updated" not in st.session_state:
+    st.session_state.image_updated = False
+
 with tab1:
     with st.sidebar:
         st.header("🔧 Tools")
@@ -160,7 +167,7 @@ with tab1:
                 st.image(img, caption="📄 Referenced Image", use_container_width=True)
             except Exception as e:
                 st.error(f"🖼️ Failed to load image: {e}")
-                
+
 # ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
     ASSISTANT_ID = "asst_9v09zgizdcuuhNdcFQpRo9RO"