Update app.py

app.py

@@ -20,45 +20,56 @@ st.write(
     """
 )
 
-# Create placeholders for webcam and captured image
-webcam_placeholder = st.empty()
-captured_image_placeholder = st.empty()
-
-# Create columns for buttons
-col1, col2 = st.columns(2)
-start_button = col1.button("Start Webcam")
-capture_button = col2.button("Capture Image", disabled=True)
-
-# Results area
-results_area = st.container()
-
 # Initialize session state variables if they don't exist
 if 'camera_on' not in st.session_state:
     st.session_state.camera_on = False
 if 'captured_image' not in st.session_state:
     st.session_state.captured_image = None
+if 'analyzed' not in st.session_state:
+    st.session_state.analyzed = False
 
 # Function to start webcam
 def start_webcam():
     st.session_state.camera_on = True
-    st.experimental_rerun()
+    st.session_state.analyzed = False
 
 # Function to capture image
 def capture_image():
     st.session_state.camera_on = False
-    st.experimental_rerun()
+    st.session_state.analyzed = False
+
+# Function to reset
+def reset_app():
+    st.session_state.camera_on = False
+    st.session_state.captured_image = None
+    st.session_state.analyzed = False
+
+# Create placeholders for webcam and captured image
+webcam_placeholder = st.empty()
+captured_image_placeholder = st.empty()
+
+# Create columns for buttons
+col1, col2 = st.columns(2)
+
+# Results area
+results_area = st.container()
 
-# Set button callbacks
-if start_button:
-    start_webcam()
-if capture_button:
-    capture_image()
+# Main app logic
+if not st.session_state.camera_on and st.session_state.captured_image is None:
+    # Initial state - just show the start button
+    if col1.button("Start Webcam"):
+        start_webcam()
+        st.rerun()
 
-# Main webcam logic
-if st.session_state.camera_on:
-    # Update the UI to show the webcam is active
-    capture_button = col2.button("Capture Image", key='capture_active', on_click=capture_image)
-    stop_button = col1.button("Stop Webcam", key='stop_webcam', on_click=lambda: setattr(st.session_state, 'camera_on', False))
+elif st.session_state.camera_on:
+    # Webcam is on - show the capture and stop buttons
+    if col1.button("Stop Webcam"):
+        st.session_state.camera_on = False
+        st.rerun()
+    
+    if col2.button("Capture Image"):
+        capture_image()
+        st.rerun()
     
     # Display webcam feed
     cap = cv2.VideoCapture(0)
@@ -78,21 +89,23 @@ if st.session_state.camera_on:
         else:
             st.error("Failed to capture image from webcam.")
         
-        # Small delay to allow UI updates
-        time.sleep(0.1)
-        
         # Release the camera
         cap.release()
         
         # Force a rerun to get another frame (creates a video effect)
         if st.session_state.camera_on:
             time.sleep(0.1)
-            st.experimental_rerun()
+            st.rerun()
 
-# If we have a captured image, display and analyze it
 elif st.session_state.captured_image is not None:
-    # Re-enable the start webcam button with a new key
-    start_again_button = col1.button("Restart Webcam", key='restart_webcam', on_click=start_webcam)
+    # We have a captured image - show it and analyze
+    if col1.button("Restart Webcam"):
+        start_webcam()
+        st.rerun()
+        
+    if col2.button("Reset"):
+        reset_app()
+        st.rerun()
     
     # Get the captured image
     image = st.session_state.captured_image
@@ -103,50 +116,51 @@ elif st.session_state.captured_image is not None:
     # Show the captured image
     captured_image_placeholder.image(image_rgb, caption="Captured Image", use_column_width=True)
     
-    # Analyze the image
-    with results_area:
-        with st.spinner("Analyzing your image with advanced AI models..."):
-            try:
-                results = DeepFace.analyze(
-                    image,
-                    actions=['age', 'gender'],
-                    detector_backend='retinaface',
-                    enforce_detection=True,
-                    align=True
-                )
-                
-                if isinstance(results, list):
-                    results = sorted(results, key=lambda x: x.get('face_confidence', 0), reverse=True)
-                    main_result = results[0]
-                else:
-                    main_result = results
-                
-                st.success("Analysis complete! Here's what we found:")
-                st.write("## Detailed Results")
-                
-                age = main_result['age']
-                st.write(f"**Predicted Age:** {age} years")
-                
-                gender = main_result['gender']
-                dominant_gender = gender if isinstance(gender, str) else max(gender, key=gender.get)
-                confidence = gender[dominant_gender] if isinstance(gender, dict) else None
+    # Analyze the image if not already analyzed
+    if not st.session_state.analyzed:
+        with results_area:
+            with st.spinner("Analyzing your image with advanced AI models..."):
+                try:
+                    results = DeepFace.analyze(
+                        image,
+                        actions=['age', 'gender'],
+                        detector_backend='retinaface',
+                        enforce_detection=True,
+                        align=True
+                    )
+                    
+                    if isinstance(results, list):
+                        results = sorted(results, key=lambda x: x.get('face_confidence', 0), reverse=True)
+                        main_result = results[0]
+                    else:
+                        main_result = results
+                    
+                    st.success("Analysis complete! Here's what we found:")
+                    st.write("## Detailed Results")
+                    
+                    age = main_result['age']
+                    st.write(f"**Predicted Age:** {age} years")
+                    
+                    gender = main_result['gender']
+                    dominant_gender = gender if isinstance(gender, str) else max(gender, key=gender.get)
+                    confidence = gender[dominant_gender] if isinstance(gender, dict) else None
+                    
+                    if confidence:
+                        st.write(f"**Predicted Gender:** {dominant_gender} ({confidence:.2f}% confidence)")
+                    else:
+                        st.write(f"**Predicted Gender:** {dominant_gender}")
+                    
+                    # Set analyzed flag to avoid re-analyzing on reruns
+                    st.session_state.analyzed = True
                 
-                if confidence:
-                    st.write(f"**Predicted Gender:** {dominant_gender} ({confidence:.2f}% confidence)")
-                else:
-                    st.write(f"**Predicted Gender:** {dominant_gender}")
-            
-            except Exception as e:
-                st.error(f"Analysis failed: {str(e)}")
-                st.info(
-                    "For best results, please try the following tips:\n"
-                    "- Ensure good lighting conditions\n"
-                    "- Position your face clearly in the frame\n"
-                    "- Move closer to the camera if needed"
-                )
-else:
-    # Initial state, just display start webcam button
-    st.info("Click 'Start Webcam' to begin")
+                except Exception as e:
+                    st.error(f"Analysis failed: {str(e)}")
+                    st.info(
+                        "For best results, please try the following tips:\n"
+                        "- Ensure good lighting conditions\n"
+                        "- Position your face clearly in the frame\n"
+                        "- Move closer to the camera if needed"
+                    )
 
 st.markdown("---")
 st.markdown(