Add estimated time and animated status indicator for category extraction

app.py

@@ -850,7 +850,24 @@ with col_input:
                 else:
                     model_source = get_model_source(model)
 
-                    with st.spinner("Extracting categories..."):
+                    # Calculate estimated time based on input size
+                    num_items = len(input_data) if isinstance(input_data, list) else 1
+                    if input_type_selected == "pdf":
+                        # PDFs take longer - estimate ~5s per page
+                        total_pages = sum(count_pdf_pages(p) for p in (input_data if isinstance(input_data, list) else [input_data]))
+                        est_seconds = total_pages * 5
+                    elif input_type_selected == "image":
+                        # Images ~4s each
+                        est_seconds = num_items * 4
+                    else:
+                        # Text ~2s per item, but batched
+                        est_seconds = max(10, num_items * 0.5)
+
+                    est_time_str = f"{est_seconds:.0f}s" if est_seconds < 60 else f"{est_seconds/60:.1f}m"
+
+                    # Animated status indicator
+                    with st.status(f"Extracting categories (est. {est_time_str})...", expanded=True) as status:
+                        st.write("Analyzing your data to discover categories...")
                         start_time = time.time()
 
                         extract_kwargs = {
@@ -872,13 +889,15 @@ with col_input:
                             processing_time = time.time() - start_time
 
                             if categories:
+                                status.update(label=f"Extracted {len(categories)} categories in {processing_time:.1f}s", state="complete", expanded=False)
                                 st.session_state.extracted_categories = categories
                                 st.session_state.task_mode = "manual"
-                                st.success(f"Extracted {len(categories)} categories in {processing_time:.1f}s")
                                 st.rerun()
                             else:
-                                st.error("No categories were extracted")
+                                status.update(label="No categories extracted", state="error")
+                                st.error("No categories were extracted from the data")
                         except Exception as e:
+                            status.update(label="Extraction failed", state="error")
                             st.error(f"Error: {str(e)}")
 
     # Category inputs (shown for manual mode or after extraction)