Update src/streamlit_app.py

src/streamlit_app.py

@@ -818,4 +818,242 @@ def detector_page(squad, warnings_text=None):
                 color = "red" if bot_score > 50 else "green"
                 k1.markdown(
                     f"""<div class="stat-box">
-                        <div class="stat-num" style="color:{color}">{bot_score:.0f}%
+                        <div class="stat-num" style="color:{color}">{bot_score:.0f}%</div>
+                        <div class="stat-txt">AI-Likeness</div>
+                    </div>""",
+                    unsafe_allow_html=True
+                )
+                k2.markdown(
+                    f"""<div class="stat-box">
+                        <div class="stat-num">{grammar_score:.0f}%</div>
+                        <div class="stat-txt">Grammar Quality</div>
+                    </div>""",
+                    unsafe_allow_html=True
+                )
+                k3.markdown(
+                    f"""<div class="stat-box">
+                        <div class="stat-num">{mood_label}</div>
+                        <div class="stat-txt">Sentiment</div>
+                    </div>""",
+                    unsafe_allow_html=True
+                )
+
+                st.write("")
+                g1, g2, g3 = st.columns(3)
+                with g1:
+                    st.markdown("#### 📊 Scores")
+                    fig = breakdown_chart(res)
+                    st.pyplot(fig, use_container_width=True)
+                with g2:
+                    st.markdown("#### 📏 Sentence Lengths")
+                    fig2 = sentence_length_chart(stats)
+                    st.pyplot(fig2, use_container_width=True)
+                with g3:
+                    st.markdown("#### 🔤 Top Words")
+                    fig3 = word_freq_chart(stats)
+                    st.pyplot(fig3, use_container_width=True)
+
+                st.markdown("#### 💡 Verdict & Explanation")
+                if verdict_type == "error":
+                    st.error(verdict_text)
+                elif verdict_type == "warning":
+                    st.warning(verdict_text)
+                else:
+                    st.success(verdict_text)
+
+                reasons = explain_text(res, stats, strict_mode_saved)
+                for r in reasons:
+                    st.markdown(f"- {r}")
+
+                st.markdown(
+                    "<small>Note: These scores and explanations are signals, not absolute proof. "
+                    "Always combine them with your own judgement.</small>",
+                    unsafe_allow_html=True
+                )
+
+                # PDF report button
+                st.write("")
+                if HAVE_REPORTLAB:
+                    img_info_for_pdf = st.session_state.get("img_res_for_pdf", None)
+                    reverse_search_for_pdf = st.session_state.get("reverse_search_for_pdf", None)
+                    pdf_bytes = generate_pdf_report(
+                        platform_saved,
+                        review_text_saved,
+                        res,
+                        stats,
+                        img_info_for_pdf,
+                        reverse_search_for_pdf
+                    )
+                    st.download_button(
+                        "📄 Download PDF Report",
+                        data=pdf_bytes,
+                        file_name="review_validator_report.pdf",
+                        mime="application/pdf",
+                    )
+                else:
+                    st.info("PDF report requires reportlab. Add `reportlab` to requirements.txt to enable export.")
+
+    # --- IMAGE TAB ---
+    with tab2:
+        col_in, col_view = st.columns([1, 1])
+
+        with col_in:
+            st.markdown("#### Step 1: Provide Image")
+            method = st.radio(
+                "Input Method",
+                ["Paste URL", "Upload File"],
+                horizontal=True,
+                label_visibility="collapsed"
+            )
+
+            with st.form("image_form"):
+                img_file = None
+                img_url = None
+
+                if method == "Paste URL":
+                    img_url = st.text_input("Paste Image Link:")
+                else:
+                    img_file = st.file_uploader("Upload Image", type=['jpg', 'jpeg', 'png'])
+
+                strict_img = st.checkbox("Use Strict AI Mode for Images", value=True)
+                do_reverse_search = st.checkbox("🔍 Perform Reverse Image Search", value=True)
+                submitted = st.form_submit_button("Scan Image", type="primary")
+
+                if submitted:
+                    target_img = None
+                    if method == "Paste URL" and img_url:
+                        target_img = get_image_from_url(img_url)
+                    elif method == "Upload File" and img_file:
+                        try:
+                            target_img = Image.open(img_file).convert("RGB")
+                        except Exception:
+                            target_img = None
+
+                    if target_img is None:
+                        st.error("Could not read image. Try another link or file.")
+                    else:
+                        with st.spinner("Scanning image..."):
+                            data = check_image(target_img, squad)
+                            st.session_state['img_res'] = (data, strict_img)
+                            st.session_state['current_img'] = target_img
+                            # store a simplified version for PDF report
+                            st.session_state['img_res_for_pdf'] = data
+                            
+                            # Perform reverse image search if requested
+                            if do_reverse_search:
+                                with st.spinner("Performing reverse image search..."):
+                                    reverse_results = reverse_image_search(target_img)
+                                    st.session_state['reverse_search'] = reverse_results
+                                    st.session_state['reverse_search_for_pdf'] = reverse_results
+                            else:
+                                st.session_state['reverse_search'] = None
+                                st.session_state['reverse_search_for_pdf'] = None
+
+        with col_view:
+            if 'current_img' in st.session_state:
+                st.image(
+                    st.session_state['current_img'],
+                    use_column_width=True,
+                    caption="Analyzed Image"
+                )
+
+            if 'img_res' in st.session_state:
+                data, strict_img = st.session_state['img_res']
+                ai_score = data['ai_chance']
+                caption = data['caption']
+
+                st.markdown("#### Step 2: Analysis Results")
+
+                st.markdown(f"""
+                <div class="analysis-box">
+                    <strong>👁️ Visual Caption (approx):</strong><br>
+                    <em>{caption}</em>
+                </div>
+                """, unsafe_allow_html=True)
+
+                st.write("")
+
+                if strict_img:
+                    t_high = 90
+                    t_mid = 70
+                else:
+                    t_high = 80
+                    t_mid = 60
+
+                if ai_score >= t_high:
+                    st.error(f"🤖 Very likely AI-generated ({ai_score:.0f}% AI score)")
+                elif ai_score >= t_mid:
+                    st.warning(f"🤔 Suspicious / possibly AI ({ai_score:.0f}% AI score)")
+                else:
+                    st.success(f"📸 Likely real photo ({100 - ai_score:.0f}% real score)")
+
+                st.markdown("**Detector Details:**")
+                st.progress(ai_score / 100.0, text=f"AI probability: {ai_score:.1f}%")
+
+                with st.expander("See raw detector scores"):
+                    st.write(f"Image AI Score (model): {ai_score:.1f}%")
+
+                # Display reverse search results
+                if 'reverse_search' in st.session_state and st.session_state['reverse_search'] is not None:
+                    reverse_data = st.session_state['reverse_search']
+                    
+                    st.markdown("---")
+                    st.markdown("#### 🔍 Reverse Image Search Results")
+                    
+                    if reverse_data['success']:
+                        results = reverse_data['results']
+                        
+                        if results:
+                            st.markdown(f"""
+                            <div class="reverse-search-box">
+                                <strong>✅ Found {len(results)} matches online</strong><br>
+                                <small>This image appears on the following websites:</small>
+                            </div>
+                            """, unsafe_allow_html=True)
+                            
+                            for i, result in enumerate(results, 1):
+                                with st.container():
+                                    st.markdown(f"""
+                                    <div class="result-item">
+                                        <strong>{i}. {result['title']}</strong><br>
+                                        <small>📍 Source: {result['source']}</small><br>
+                                        <small>🔗 <a href="{result['link']}" target="_blank">View Original</a></small>
+                                    </div>
+                                    """, unsafe_allow_html=True)
+                        else:
+                            st.markdown("""
+                            <div class="reverse-search-box">
+                                <strong>🆕 No matches found online</strong><br>
+                                <small>This could mean the image is original/new, or not indexed by Google yet.</small>
+                            </div>
+                            """, unsafe_allow_html=True)
+                    else:
+                        st.error(f"❌ {reverse_data['error']}")
+                        if "SERPAPI_KEY" in reverse_data['error']:
+                            st.info("💡 To enable reverse image search:\n1. Sign up at https://serpapi.com/\n2. Add your API key to secrets or environment variables")
+
+# --- MAIN CONTROLLER ---
+def main():
+    inject_custom_css()
+
+    if 'page' not in st.session_state:
+        st.session_state['page'] = 'landing'
+
+    with st.spinner("Loading AI models (first run can take some time)..."):
+        squad, err = load_ai_squad()
+
+    if squad is None:
+        st.error(err or "Failed to load models.")
+        st.stop()
+
+    warnings_text = None
+    if err:
+        warnings_text = "Some features may be limited:<br>" + err.replace("\n", "<br>")
+
+    if st.session_state['page'] == 'landing':
+        landing_page()
+    else:
+        detector_page(squad, warnings_text=warnings_text)
+
+if __name__ == "__main__":
+    main()