Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import csv
 import zipfile
 import shutil
+import time
 from datetime import datetime
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
@@ -11,7 +12,6 @@ import gradio as gr
 from deepface import DeepFace
 import numpy as np
 from PIL import Image
-import time
 from pathlib import Path
 import pandas as pd
 
@@ -55,7 +55,7 @@ def validate_batch_number(batch_no):
     return True, ""
 
 def toggle_webcam_flow(batch_no):
-    """Control visibility of components based on flow"""
+    """Control visibility of components based on flow with added delay"""
     if batch_no.strip():
         # Validate batch number
         is_valid, validation_msg = validate_batch_number(batch_no)
@@ -64,15 +64,17 @@ def toggle_webcam_flow(batch_no):
                 batch_no_input: gr.Textbox(visible=True, value=""),
                 webcam: gr.Image(visible=False),
                 message: gr.Textbox(value=validation_msg, visible=True),
-                batch_no_state: ""
+                batch_no_state: "",
+                processing: gr.Textbox(visible=False)
             }
         
-        # Valid batch number - hide input, show webcam
+        # Show processing message before showing webcam
         return {
             batch_no_input: gr.Textbox(visible=False),
-            webcam: gr.Image(visible=True),
-            message: gr.Textbox(visible=False),
-            batch_no_state: batch_no
+            webcam: gr.Image(visible=False),
+            message: gr.Textbox(value=f"Processing batch {batch_no}...", visible=True),
+            batch_no_state: batch_no,
+            processing: gr.Textbox(value="Please wait...", visible=True)
         }
     else:
         # No batch number - show input, hide webcam
@@ -80,9 +82,19 @@ def toggle_webcam_flow(batch_no):
             batch_no_input: gr.Textbox(visible=True),
             webcam: gr.Image(visible=False),
             message: gr.Textbox(visible=False),
-            batch_no_state: ""
+            batch_no_state: "",
+            processing: gr.Textbox(visible=False)
         }
 
+def show_webcam_after_delay(batch_no):
+    """Show webcam after a short delay"""
+    time.sleep(3)  # 3 second delay
+    return {
+        webcam: gr.Image(visible=True),
+        message: gr.Textbox(visible=False),
+        processing: gr.Textbox(visible=False)
+    }
+
 def predict_emotion(batch_no: str, image):
     if not batch_no.strip():
         return {
@@ -92,7 +104,9 @@ def predict_emotion(batch_no: str, image):
             result_img_comp: gr.Image(visible=False),
             result_text_comp: gr.Textbox(visible=False),
             batch_no_input: gr.Textbox(visible=True),
-            webcam: gr.Image(visible=False)
+            webcam: gr.Image(visible=False),
+            done_btn: gr.Button(visible=False),
+            processing: gr.Textbox(visible=False)
         }
     
     # Validate batch number format
@@ -105,7 +119,9 @@ def predict_emotion(batch_no: str, image):
             result_img_comp: gr.Image(visible=False),
             result_text_comp: gr.Textbox(visible=False),
             batch_no_input: gr.Textbox(visible=True),
-            webcam: gr.Image(visible=False)
+            webcam: gr.Image(visible=False),
+            done_btn: gr.Button(visible=False),
+            processing: gr.Textbox(visible=False)
         }
     
     if image is None:
@@ -116,7 +132,9 @@ def predict_emotion(batch_no: str, image):
             result_img_comp: gr.Image(visible=False),
             result_text_comp: gr.Textbox(visible=False),
             batch_no_input: gr.Textbox(visible=False),
-            webcam: gr.Image(visible=True)
+            webcam: gr.Image(visible=True),
+            done_btn: gr.Button(visible=False),
+            processing: gr.Textbox(visible=False)
         }
     
     try:
@@ -148,7 +166,9 @@ def predict_emotion(batch_no: str, image):
                 result_img_comp: gr.Image(visible=False),
                 result_text_comp: gr.Textbox(visible=False),
                 batch_no_input: gr.Textbox(visible=True),
-                webcam: gr.Image(visible=False)
+                webcam: gr.Image(visible=False),
+                done_btn: gr.Button(visible=False),
+                processing: gr.Textbox(visible=False)
             }
 
         # Process the first face found
@@ -190,7 +210,9 @@ def predict_emotion(batch_no: str, image):
             result_img_comp: gr.Image(visible=True),
             result_text_comp: gr.Textbox(visible=True),
             batch_no_input: gr.Textbox(visible=True, value=""),
-            webcam: gr.Image(visible=False)
+            webcam: gr.Image(visible=False),
+            done_btn: gr.Button(visible=True),
+            processing: gr.Textbox(visible=False)
         }
     
     except Exception as e:
@@ -201,9 +223,23 @@ def predict_emotion(batch_no: str, image):
             result_img_comp: gr.Image(visible=False),
             result_text_comp: gr.Textbox(visible=False),
             batch_no_input: gr.Textbox(visible=True, value=""),
-            webcam: gr.Image(visible=False)
+            webcam: gr.Image(visible=False),
+            done_btn: gr.Button(visible=False),
+            processing: gr.Textbox(visible=False)
         }
 
+def complete_session():
+    """Reset interface after user is done viewing results"""
+    return {
+        batch_no_input: gr.Textbox(visible=True, value=""),
+        webcam: gr.Image(visible=False),
+        result_img_comp: gr.Image(visible=False),
+        result_text_comp: gr.Textbox(visible=False),
+        done_btn: gr.Button(visible=False),
+        message: gr.Textbox(value="Session completed. Next user may enter batch number.", visible=True),
+        processing: gr.Textbox(visible=False)
+    }
+
 def get_image_gallery(emotion, image_type):
     """Get image gallery for selected emotion and type"""
     if emotion == "All Emotions":
@@ -444,14 +480,22 @@ with gr.Blocks(title="Emotion Capture", css="""
     .gradio-container { max-width: 800px !important }
     .message { color: red; font-weight: bold; }
     .gallery { grid-template-columns: repeat(auto-fill, minmax(200px, 1fr)); }
+    .done-btn { margin-top: 20px; }
+    .processing { 
+        font-size: 1.2em; 
+        color: #0066cc; 
+        font-weight: bold;
+        text-align: center;
+        margin: 20px 0;
+    }
 """) as capture_interface:
     
     gr.Markdown("""
     # Emotion Capture Interface
-    1. Enter your batch number (numbers only)
-    2. Capture your face (webcam will appear)
+    1. Enter your batch number (numbers only) and wait for confirmation
+    2. Capture your face (webcam will appear after validation)
     3. View your emotion analysis results
-    4. Interface will reset automatically for next user
+    4. Click "Done" when finished to reset for next user
     """)
     
     # Add a state to store batch number between callbacks
@@ -465,6 +509,14 @@ with gr.Blocks(title="Emotion Capture", css="""
         )
         message = gr.Textbox(label="", visible=False, interactive=False, elem_classes="message")
     
+    # Processing indicator (hidden by default)
+    processing = gr.Textbox(
+        label="", 
+        visible=False, 
+        interactive=False, 
+        elem_classes="processing"
+    )
+    
     # Webcam component starts hidden
     webcam = gr.Image(
         sources=["webcam"],
@@ -485,11 +537,23 @@ with gr.Blocks(title="Emotion Capture", css="""
     result_img_comp = gr.Image(visible=False)
     result_text_comp = gr.Textbox(visible=False)
     
-    # Show webcam when valid batch number is entered
+    # Done button (starts hidden)
+    done_btn = gr.Button(
+        "Done", 
+        visible=False, 
+        elem_classes="done-btn",
+        variant="primary"
+    )
+    
+    # Show webcam when valid batch number is entered (with delay)
     batch_no_input.change(
         toggle_webcam_flow,
         inputs=batch_no_input,
-        outputs=[batch_no_input, webcam, message, batch_no_state]
+        outputs=[batch_no_input, webcam, message, batch_no_state, processing]
+    ).then(
+        show_webcam_after_delay,
+        inputs=batch_no_state,
+        outputs=[webcam, message, processing]
     )
     
     # Process when webcam captures an image
@@ -503,7 +567,23 @@ with gr.Blocks(title="Emotion Capture", css="""
             result_img_comp,
             result_text_comp,
             batch_no_input,
-            webcam
+            webcam,
+            done_btn,
+            processing
+        ]
+    )
+    
+    # Handle session completion
+    done_btn.click(
+        complete_session,
+        outputs=[
+            batch_no_input,
+            webcam,
+            result_img_comp,
+            result_text_comp,
+            done_btn,
+            message,
+            processing
         ]
     )
 
@@ -703,7 +783,8 @@ with gr.Blocks(title="Data Management") as data_interface:
 # Combine interfaces
 demo = gr.TabbedInterface(
     [capture_interface, data_interface],
-    ["Emotion Capture", "Data Management"]
+    ["Emotion Capture", "Data Management"],
+    title="Emotion Analysis System"
 )
 
 if __name__ == "__main__":