update interface

app.py

@@ -282,7 +282,6 @@ def fuse_captions(cap1: str, cap2: str, qwen_tok, qwen_mod) -> str:
             if fused.lower().startswith(prefix.lower()):
                 fused = fused[len(prefix):].strip()
 
-        # Safety — trim to last complete sentence if still cut off
         if fused and not fused.endswith((".", "!", "?")):
             last_stop = max(
                 fused.rfind("."),
@@ -477,17 +476,23 @@ if uploaded_file is not None:
     with col_run:
         if st.button("Generate Caption", type="primary", use_container_width=True):
 
-            with st.spinner("Loading local models (first run takes 3-4 min)..."):
+            st.info(" First time? Models are downloading and loading — this takes 3 to 4 minutes only once. After that, every run will be much faster!")
+            st.warning(" Please do not refresh or close the page while models are loading.")
+
+            with st.spinner("⏳ Loading Florence-2, BLIP and Qwen models... Please wait patiently!"):
                 (
                     florence_proc, florence_mod,
                     blip_proc, blip_itm,
                     qwen_tok, qwen_mod
                 ) = load_local_models()
 
+            st.success(" All models loaded successfully! Starting the pipeline now...")
+
             progress = st.progress(0)
             status   = st.empty()
 
             status.info("Step 1/6: Generating captions with Florence-2-Large...")
+            st.info(" Florence-2 is reading your image carefully — generating 5 diverse captions. This takes about 20–30 seconds, please hold on!")
             captions = generate_captions_florence(
                 input_image, florence_proc, florence_mod
             )
@@ -536,6 +541,7 @@ if uploaded_file is not None:
                 st.info(f"2. {best_2}")
 
             status.info("Step 6/6: Fusing captions with Qwen2.5-1.5B...")
+            st.success(" Best captions selected! Now crafting your final detailed caption — please hold on for about 30–60 seconds, it is definitely on the way!")
             final = fuse_captions(best_1, best_2, qwen_tok, qwen_mod)
             progress.progress(100)
             status.success("Pipeline complete!")