Update app.py

app.py

@@ -18,12 +18,16 @@ import gradio as gr
 from deepface import DeepFace
 import cv2
 
+# ---------------------------
 # Device
+# ---------------------------
 if not torch.cuda.is_available():
     os.environ["CUDA_VISIBLE_DEVICES"] = ""
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# Model
+# ---------------------------
+# Load ResNet50
+# ---------------------------
 weights = ResNet50_Weights.DEFAULT
 model = resnet50(weights=weights).to(device)
 model.eval()
@@ -64,7 +68,9 @@ def get_dominant_color(image,num_colors=5):
     hex_color = f"#{dominant_color[0]:02x}{dominant_color[1]:02x}{dominant_color[2]:02x}"
     return dominant_color, hex_color
 
+# ---------------------------
 # Core analysis
+# ---------------------------
 def classify_zip_and_analyze_color(zip_file):
     results = []
     images_dict = {}  # store images for preview
@@ -199,15 +205,21 @@ def classify_zip_and_analyze_color(zip_file):
 
     return df, list(images_dict.keys()), images_dict, out_xlsx, plot1_img, plot2_img, plot3_img, plot4_img
 
-# Show selected image preview
-def show_preview(filename, images_dict):
-    return images_dict.get(filename, None)
+# ---------------------------
+# Preview callback
+# ---------------------------
+def show_preview(selected_file, images_state):
+    if images_state is None or selected_file is None:
+        return None
+    return images_state.get(selected_file, None)
 
 # ---------------------------
 # Gradio interface
 # ---------------------------
 with gr.Blocks() as demo:
     uploaded_zip = gr.File(label="Upload ZIP of images", file_types=[".zip"])
+    analyze_btn = gr.Button("Run Analysis")  # Run button just after upload
+
     output_df = gr.Dataframe(headers=["Filename","Top 3 Predictions","Confidence","Dominant Color","Basic Color","Face Info"])
     image_dropdown = gr.Dropdown(label="Select image to preview")
     image_preview = gr.Image(label="Image Preview")
@@ -219,8 +231,6 @@ with gr.Blocks() as demo:
     plot3 = gr.Image(label="Gender Distribution")
     plot4 = gr.Image(label="Age Distribution by Gender")
 
-    analyze_btn = gr.Button("Run Analysis")
-
     def run_analysis(zip_file):
         df, filenames, images_dict, out_xlsx, p1, p2, p3, p4 = classify_zip_and_analyze_color(zip_file)
         return df, filenames, images_dict, out_xlsx, p1, p2, p3, p4