Update src/app.py

src/app.py

@@ -173,6 +173,7 @@ def load_model(from_pretrained, base_model, cache_dir, device):
 # Result Feedback Module Import
 # -----------------
 from feedback import FeedbackManager
+from stats import StatsManager
 
 # Initialize Feedback Manager with HF dataset
 # make sure HF_TOKEN is set to visit private repository
@@ -183,6 +184,16 @@ feedback_manager = FeedbackManager(
     local_backup=False if os.environ.get('SPACE_ID') else True # 保留本地备份
 )
 
+@st.cache_resource
+def get_stats_manager():
+    return StatsManager(
+        dataset_repo_id=FEEDBACK_DATASET_ID,
+        hf_token=os.environ.get('HF_TOKEN'),
+        local_backup=False if os.environ.get('SPACE_ID') else True,
+    )
+
+stats_manager = get_stats_manager()
+
 # -----------------
 # Configuration
 # -----------------
@@ -226,6 +237,13 @@ if 'feedback_given' not in st.session_state:
     st.session_state.feedback_given = False
 # ========================================
 
+# ----- Visit Counter -----
+# session_state resets on F5 / new tab, so this runs exactly once per browser session
+if 'visit_counted' not in st.session_state:
+    st.session_state.visit_counted = True
+    stats_manager.increment_visit()
+# -------------------------
+
 # -----------------
 # Streamlit Layout
 # -----------------
@@ -319,6 +337,12 @@ if detect_clicked:
                 'elapsed_time': elapsed_time
             }
 
+            # Count detection once per unique detect action
+            _det_key = f'det_counted_{hash(text_input[:80])}'
+            if _det_key not in st.session_state:
+                st.session_state[_det_key] = True
+                stats_manager.increment_detection()
+
             st.info(
                 f"""
                 **Conclusion**: 
@@ -362,58 +386,74 @@ if detect_clicked:
                     """
                 )
 
-            # ========== 🆕 Feedback buttons (moved here for better UX) ==========
-            st.markdown("**📝 Result Feedback**: Does this detection result meet your expectations?")
-            
-            current_text = text_input
-            current_domain = selected_domain
-            current_statistics = crit
-            current_pvalue = p_value
-            feedback_col1, feedback_col2 = st.columns(2)
-            
-            with feedback_col1:
-                if st.button("✅ Expected", use_container_width=True, type="secondary", key=f"expected_btn_{hash(text_input[:50])}"):
-                    try:
-                        success, message = feedback_manager.save_feedback(
-                            current_text,
-                            current_domain,
-                            current_statistics,
-                            current_pvalue,
-                            'expected'
-                        )
-                        if success:
-                            st.success("✅ Thank you for your feedback!")
-                            st.caption(f"💾 {message}")
-                        else:
-                            st.error(f"Failed to save feedback: {message}")
-                    except Exception as e:
-                        st.error(f"Failed to save feedback: {str(e)}")
-                        import traceback
-                        st.code(traceback.format_exc())
-            
-            with feedback_col2:
-                if st.button("❌ Unexpected", use_container_width=True, type="secondary", key=f"unexpected_btn_{hash(text_input[:50])}"):
-                    try:
-                        success, message = feedback_manager.save_feedback(
-                            current_text,
-                            current_domain,
-                            current_statistics,
-                            current_pvalue,
-                            'unexpected'
-                        )
-                        if success:
-                            st.warning("❌ Feedback recorded! This will help us improve.")
-                            st.caption(f"💾 {message}")
-                        else:
-                            st.error(f"Failed to save feedback: {message}")
-                    except Exception as e:
-                        st.error(f"Failed to save feedback: {str(e)}")
-                        import traceback
-                        st.code(traceback.format_exc())
-            
-            if st.session_state.feedback_given:
-                st.success("✅ Feedback submitted successfully!")
-            # ============================================
+            # ========== Feedback UI ==========
+            _fb_key = f'feedback_given_{hash(text_input[:50])}'
+            if _fb_key not in st.session_state:
+                st.session_state[_fb_key] = False
+
+            st.markdown(
+                """
+                <style>
+                .fb-header { display: flex; align-items: center; gap: 0.4rem; margin-bottom: 0.3rem; }
+                .privacy-tip {
+                    position: relative; display: inline-block;
+                    cursor: help; color: #9ca3af; font-size: 0.9rem;
+                }
+                .privacy-tip .tip-text {
+                    visibility: hidden; opacity: 0;
+                    width: 240px; background-color: #374151; color: #f9fafb;
+                    text-align: left; border-radius: 6px;
+                    padding: 0.5rem 0.7rem; font-size: 0.78rem; line-height: 1.4;
+                    position: absolute; z-index: 100;
+                    bottom: 130%; left: 50%; transform: translateX(-50%);
+                    transition: opacity 0.25s ease; pointer-events: none;
+                }
+                .privacy-tip:hover .tip-text { visibility: visible; opacity: 1; }
+                </style>
+                <div class="fb-header">
+                  <strong>📝 Result Feedback</strong>: Does this detection result meet your expectations?
+                  <span class="privacy-tip">🔒
+                    <span class="tip-text">🔒 Your feedback is stored privately and will never be shared with third parties. It is used solely to improve detection accuracy.</span>
+                  </span>
+                </div>
+                """,
+                unsafe_allow_html=True
+            )
+
+            if not st.session_state[_fb_key]:
+                feedback_col1, feedback_col2 = st.columns(2)
+                with feedback_col1:
+                    if st.button("✅ Expected", use_container_width=True, type="secondary",
+                                 key=f"expected_btn_{hash(text_input[:50])}"):
+                        try:
+                            fb_success, fb_message = feedback_manager.save_feedback(
+                                text_input, selected_domain, crit, p_value, 'expected'
+                            )
+                            if fb_success:
+                                st.session_state[_fb_key] = True
+                                st.toast("Thank you for your feedback!", icon="✅")
+                            else:
+                                st.error(f"Failed to save feedback: {fb_message}")
+                        except Exception as e:
+                            st.error(f"Failed to save feedback: {str(e)}")
+
+                with feedback_col2:
+                    if st.button("❌ Unexpected", use_container_width=True, type="secondary",
+                                 key=f"unexpected_btn_{hash(text_input[:50])}"):
+                        try:
+                            fb_success, fb_message = feedback_manager.save_feedback(
+                                text_input, selected_domain, crit, p_value, 'unexpected'
+                            )
+                            if fb_success:
+                                st.session_state[_fb_key] = True
+                                st.toast("Feedback recorded! This will help us improve.", icon="📝")
+                            else:
+                                st.error(f"Failed to save feedback: {fb_message}")
+                        except Exception as e:
+                            st.error(f"Failed to save feedback: {str(e)}")
+            else:
+                st.caption("✅ Feedback submitted — thank you!")
+            # =============================================
 
             # Show detailed results
             with result_placeholder:
@@ -442,6 +482,32 @@ if detect_clicked:
 #         language="bibtex"
 #     )
 
+# -----------------
+# Statistics Chip (fixed top-right)
+# -----------------
+st.markdown(
+    f"""
+    <style>
+    .stats-chip {{
+        position: fixed;
+        top: 3.6rem;
+        right: 1rem;
+        display: flex;
+        align-items: center;
+        gap: 0.35rem;
+        font-size: 0.78rem;
+        color: #9ca3af;
+        z-index: 999;
+        pointer-events: none;
+    }}
+    </style>
+    <div class="stats-chip">
+      <span>{stats_manager.visit_count:,} visits</span>
+    </div>
+    """,
+    unsafe_allow_html=True
+)
+
 # -----------------
 # Footer
 # -----------------