Update src/app.py

src/app.py

@@ -6,6 +6,8 @@ from pathlib import Path
 # -----------------
 APP_DIR = Path(__file__).parent.resolve()
 
+account_name = 'mamba413'
+
 # -----------------
 # Fix Streamlit Permission Issues
 # -----------------
@@ -34,13 +36,75 @@ import streamlit as st
 from FineTune.model import ComputeStat
 import time
 
+st.markdown(
+    """
+    <style>
+    /* Text area & text input */
+    textarea, input[type="text"] {
+        background-color: #f8fafc !important;
+        border: 1px solid #e5e7eb !important;
+        color: #111827 !important;
+    }
+
+    textarea::placeholder {
+        color: #9ca3af !important;
+    }
+
+    /* Selectbox */
+    div[data-testid="stSelectbox"] > div {
+        background-color: #f8fafc !important;
+        border: 1px solid #e5e7eb !important;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+
+st.markdown(
+    """
+    <style>
+    /* Detect button */
+    div.stButton > button[kind="primary"] {
+        background-color: #fdae6b;
+        border: white;
+        color: black;
+        font-weight: 600;
+        height: 4.3rem;
+
+        font-size: 1.1rem;
+
+        display: flex;
+        align-items: center;
+        justify-content: center;
+        gap: 0.55rem;
+    }
+
+    /* Icon inside Detect button */
+    div.stButton > button[kind="primary"] span {
+        font-size: 1.25rem;
+        line-height: 1;
+    }
+
+    div.stButton > button[kind="primary"]:hover {
+        background-color: #fd8d3c;
+        border-color: white;
+    }
+
+    div.stButton > button[kind="primary"]:active {
+        background-color: #fd8d3c;
+        border-color: white;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+
 # -----------------
 # Page Configuration
 # -----------------
 st.set_page_config(
-    page_title="AdaDetectGPT",
-    page_icon="🔍",
-    layout="wide"
+    page_title="DetectGPTPro",
+    page_icon="🕵️",
 )
 
 # -----------------
@@ -113,7 +177,7 @@ from feedback import FeedbackManager
 # Initialize Feedback Manager with HF dataset
 # 请将 'your-username/your-dataset-name' 替换为您的实际 HF 数据集仓库 ID
 # 确保在环境变量中设置了 HF_TOKEN 以访问私有数据集
-FEEDBACK_DATASET_ID = os.environ.get('FEEDBACK_DATASET_ID', 'mamba413/user-feedback')
+FEEDBACK_DATASET_ID = os.environ.get('FEEDBACK_DATASET_ID', f'{account_name}/user-feedback')
 feedback_manager = FeedbackManager(
     dataset_repo_id=FEEDBACK_DATASET_ID,
     hf_token=os.environ.get('HF_TOKEN'),
@@ -166,17 +230,15 @@ if 'feedback_given' not in st.session_state:
 # -----------------
 # Streamlit Layout
 # -----------------
-_, col0, _ = st.columns((1, 5, 1))
-with col0:
-    st.markdown(
-        "<h1 style='text-align: center; color: #0072C3;'>AdaDetectGPT: Adaptive LLM's Texts Detection</h1>",
-        unsafe_allow_html=True,
-    )
-
 st.markdown(
-    """Pasted the text to be detected below and click the 'Detect' button to get the p-value. Use a better option may improve detection."""
+    "<h1 style='text-align: center;'> 🕵️ Adaptive LLM-texts Detector</h1>",
+    unsafe_allow_html=True,
 )
 
+# st.markdown(
+#     """Pasted the text to be detected below and click the 'Detect' button to get the p-value. Use a better option may improve detection."""
+# )
+
 # Display model loading status
 if not model_loaded:
     st.error(f"❌ Failed to load model: {error_message}")
@@ -186,40 +248,59 @@ if not model_loaded:
 # Main Interface
 # -----------------
 # --- Two columns: Input text & button | Result displays ---
-col1, col2 = st.columns((1, 1))
-
-with col1:
-    text_input = st.text_area(
-        label="",
-        placeholder="Paste your text to be detected here",
-        help="Typically, providing text with a longer content would get a more reliable result.",
-        height=200,
-    )
-    
-    detect_clicked = st.button("Detect", type="primary", use_container_width=True)
+text_input = st.text_area(
+    label="",
+    placeholder="Paste your text to be detected here. Typically, providing text with a longer content would get a more reliable result.",
+    height=240,
+)
 
-    selected_domain = st.selectbox(
-        label="⚙️ Domain (Optional)",
-        options=DOMAINS,
-        index=0,  # Default to General
-        help="💡 **Tip:** Select the domain that best matches your text for improving detection accuracy. Default is 'General' that means consider all domains."
-    )
+subcol11, subcol12, subcol13 = st.columns((1, 1, 1))
 
+selected_domain = subcol11.selectbox(
+    label="💡 Domain that matches your text",
+    options=DOMAINS,
+    index=0,  # Default to General
+    # label_visibility="collapsed",
+    # label_visibility="hidden",
+)
+
+detect_clicked = subcol12.button("🔍 Detect", type="primary", use_container_width=True)
+
+selected_level = subcol13.slider(
+    label="Significance (α)",
+    min_value=0.01,
+    max_value=0.2,
+    value=0.05,
+    step=0.005,
+    # label_visibility="collapsed",
+)
+
+col2, col3, col4 = st.columns((1, 1, 2))
 with col2:
     statistics_ph = st.empty()
     statistics_ph.text_input(
-        label="Statistics",
+        label="Statistic",
         value="",
+        placeholder="",
         disabled=True,
-        help="Statistics will appear here after clicking the Detect button.",
     )
-    
+
+with col3:
     pvalue_ph = st.empty()
     pvalue_ph.text_input(
         label="p-value",
         value="",
+        placeholder="",
+        disabled=True,
+    )
+
+with col4:
+    conclusion_ph = st.empty()
+    conclusion_ph.text_input(
+        label="Conclusion",
+        value="",
+        placeholder="",
         disabled=True,
-        help="p-value will appear here after clicking the Detect button.",
     )
 
 # -----------------
@@ -267,83 +348,88 @@ if detect_clicked:
             }
 
             # Update score displays
-            with col2:
-                statistics_ph.text_input(
-                    label="Statistics",
-                    value=f"{crit:.6f}",
-                    disabled=True,
-                    help="Detection statistics will appear here after clicking Detect.",
-                )
-                
-                pvalue_ph.text_input(
-                    label="p-value",
-                    value=f"{p_value:.6f}",
-                    disabled=True,
-                    help="p-value will appear here after clicking Detect.",
-                )
-
-                st.info(
-                    """
-                    **📊 p-value:**
-                    - **Lower p-value** (closer to 0) indicates text is **more likely AI-generated**
-                    - **Higher p-value** (closer to 1) indicates text is **more likely human-written**
-                    - Generally, p-value < 0.05 suggests the text may be LLM-generated
-                    """,
-                    icon="💡"
-                )
-
-                # ========== 🆕 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!")
-                # ============================================
+            statistics_ph.text_input(
+                label="Statistics",
+                value=f"{crit:.6f}",
+                disabled=True,
+                help="Detection statistics will appear here after clicking Detect.",
+            )
+            
+            pvalue_ph.text_input(
+                label="p-value",
+                value=f"{p_value:.6f}",
+                disabled=True,
+                help="p-value will appear here after clicking Detect.",
+            )
+
+            conclusion_ph.text_input(
+                label="Conclusion",
+                value="Reject H0: Text is likely LLM-generated." if p_value < selected_level else "Fail to Reject H0: Text is likely human-written.",
+                disabled=True,
+                help="Conclusion will appear here after clicking Detect.",
+            )
+
+            st.info(
+                """
+                **📊 p-value:**
+                - **Lower p-value** (closer to 0) indicates text is **more likely AI-generated**
+                - **Higher p-value** (closer to 1) indicates text is **more likely human-written**
+                """,
+                icon="💡"
+            )
+
+            # ========== 🆕 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!")
+            # ============================================
 
             # Show detailed results
             with result_placeholder:
@@ -388,14 +474,14 @@ st.markdown(
         background-color: white;
         color: gray;
         text-align: center;
-        padding: 10px;
+        padding: 1px;
         border-top: 1px solid #e0e0e0;
         z-index: 999;
     }
     
     /* Add padding to main content to prevent overlap with fixed footer */
     .main .block-container {
-        padding-bottom: 60px;
+        padding-bottom: 1px;
     }
     </style>
     <div class='footer'>