Update src/app.py

src/app.py

@@ -22,6 +22,10 @@ st.set_page_config(
     layout="wide"
 )
 
+# Initialize session state for the camera image if it doesn't exist
+if 'captured_image_bytes' not in st.session_state:
+    st.session_state.captured_image_bytes = None
+
 # Title and description
 st.title("🤖 Empathetic AI Assistant")
 st.markdown("""
@@ -40,18 +44,26 @@ col1, col2 = st.columns([1, 1])
 
 with col1:
     st.subheader("📸 1. Capture Your Expression")
-    camera_image = st.camera_input("Take a snapshot")
     
-    # --- START: ADDED DEBUG BLOCK ---
-    # This block will re-run every time the app updates
-    if camera_image is None:
-        st.warning("DEBUG: 'camera_image' is currently None. Waiting for user to take photo...")
-    else:
-        st.success("DEBUG: Image captured!")
-        st.write(f"DEBUG: 'camera_image' has data. Type: {type(camera_image)}")
-        st.write(f"DEBUG: Image size: {camera_image.size} bytes")
-    # --- END: ADDED DEBUG BLOCK ---
+    # Use a key to give the camera_input a unique identity
+    # The output of the camera_input is stored in 'camera_input_data'
+    camera_input_data = st.camera_input("Take a snapshot", key="my_camera_input")
+
+    # If new data comes from camera_input, update session state
+    if camera_input_data is not None and camera_input_data != st.session_state.captured_image_bytes:
+        st.session_state.captured_image_bytes = camera_input_data
+        st.experimental_rerun() # Rerun to display the stored image immediately
     
+    # Display the stored image if available
+    if st.session_state.captured_image_bytes:
+        st.image(st.session_state.captured_image_bytes, caption="Captured Image")
+        st.success("DEBUG: Image captured and stored in session state!")
+        if st.button("Clear Photo", key="clear_camera_btn"):
+            st.session_state.captured_image_bytes = None
+            st.experimental_rerun() # Rerun to clear the display
+    else:
+        st.warning("DEBUG: 'captured_image_bytes' is currently None. Waiting for user to take photo...")
+
 with col2:
     st.subheader("💭 2. Your Query")
     user_query = st.text_area(
@@ -71,18 +83,19 @@ st.divider()
 # Main action button
 if st.button("🧠 Analyze My Emotion & Answer", type="primary", use_container_width=True):
     
-    # Validation
-    if not camera_image:
+    # Validation - now check against session state
+    if st.session_state.captured_image_bytes is None: # <--- CHECK SESSION STATE HERE
         st.error("❌ Please take a snapshot using the camera first!")
     elif not audio_bytes:
         st.error("❌ Please record your voice first!")
     elif not user_query.strip():
         st.error("❌ Please enter your query!")
     else:
-        # Step 1: Process camera image
+        # Step 1: Process camera image from session state
         with st.spinner("📸 Processing facial expression..."):
             try:
-                file_bytes = np.asarray(bytearray(camera_image.read()), dtype=np.uint8)
+                # Read from session state
+                file_bytes = np.asarray(bytearray(st.session_state.captured_image_bytes.read()), dtype=np.uint8)
                 image = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
                 
                 with tempfile.NamedTemporaryFile(delete=True, suffix=".jpg") as temp_img:
@@ -155,7 +168,7 @@ if st.button("🧠 Analyze My Emotion & Answer", type="primary", use_container_w
 with st.sidebar:
     st.header("ℹ️ How to Use")
     st.markdown("""
-    1. **Take a snapshot**
+    1. **Take a snapshot** (the image will stay until you clear it)
     2. **Type your query**
     3. **Click the mic** to record (click again to stop)
     4. **Click the 'Analyze' button**