Update app.py

app.py

@@ -5,7 +5,7 @@ from semviqa.ser.qatc_model import QATCForQuestionAnswering
 from semviqa.tvc.model import ClaimModelForClassification
 from semviqa.ser.ser_eval import extract_evidence_tfidf_qatc
 from semviqa.tvc.tvc_eval import classify_claim
-import io
+import time  # Thêm thư viện time để đo thời gian inference
 
 # Load models with caching
 @st.cache_resource()
@@ -179,66 +179,92 @@ with st.container():
             st.markdown("<h3>Verification Result</h3>", unsafe_allow_html=True)
             if verify_button:
                 # Placeholder for displaying result/loading
-                result_placeholder = st.empty()
-                result_placeholder.markdown("<em>Verifying...</em>")
+                with st.spinner("Verifying..."):  # Thêm spinner khi đang xử lý
+                    start_time = time.time()  # Bắt đầu đo thời gian inference
 
-                with torch.no_grad():
-                    # Extract evidence and classify the claim
-                    evidence = extract_evidence_tfidf_qatc(
-                        claim, context, model_qatc, tokenizer_qatc,
-                        "cuda" if torch.cuda.is_available() else "cpu",
-                        confidence_threshold=tfidf_threshold,
-                        length_ratio_threshold=length_ratio_threshold
-                    )
-                    verdict = "NEI"
-                    details = ""
-                    prob3class, pred_tc = classify_claim(
-                        claim, evidence, model_tc, tokenizer_tc,
-                        "cuda" if torch.cuda.is_available() else "cpu"
-                    )
-                    if pred_tc != 0:
-                        prob2class, pred_bc = classify_claim(
-                            claim, evidence, model_bc, tokenizer_bc,
+                    with torch.no_grad():
+                        # Extract evidence
+                        evidence_start_time = time.time()
+                        evidence = extract_evidence_tfidf_qatc(
+                            claim, context, model_qatc, tokenizer_qatc,
+                            "cuda" if torch.cuda.is_available() else "cpu",
+                            confidence_threshold=tfidf_threshold,
+                            length_ratio_threshold=length_ratio_threshold
+                        )
+                        evidence_time = time.time() - evidence_start_time
+
+                        # Hiển thị evidence trước
+                        st.markdown(f"""
+                            <div class='result-box'>
+                                <p><strong>Evidence:</strong> {evidence}</p>
+                                <p><strong>Evidence Inference Time:</strong> {evidence_time:.2f} seconds</p>
+                            </div>
+                        """, unsafe_allow_html=True)
+
+                        # Classify the claim
+                        verdict_start_time = time.time()
+                        verdict = "NEI"
+                        details = ""
+                        prob3class, pred_tc = classify_claim(
+                            claim, evidence, model_tc, tokenizer_tc,
                             "cuda" if torch.cuda.is_available() else "cpu"
                         )
-                        if pred_bc == 0:
-                            verdict = "SUPPORTED"
-                        elif prob2class > prob3class:
-                            verdict = "REFUTED"
-                        else:
-                            verdict = ["NEI", "SUPPORTED", "REFUTED"][pred_tc]
-                        if show_details:
-                            details = f"<p><strong>3-Class Probability:</strong> {prob3class.item():.2f} - <strong>2-Class Probability:</strong> {prob2class.item():.2f}</p>"
+                        if pred_tc != 0:
+                            prob2class, pred_bc = classify_claim(
+                                claim, evidence, model_bc, tokenizer_bc,
+                                "cuda" if torch.cuda.is_available() else "cpu"
+                            )
+                            if pred_bc == 0:
+                                verdict = "SUPPORTED"
+                            elif prob2class > prob3class:
+                                verdict = "REFUTED"
+                            else:
+                                verdict = ["NEI", "SUPPORTED", "REFUTED"][pred_tc]
+                            if show_details:
+                                details = f"""
+                                    <p><strong>3-Class Probability:</strong> {prob3class.item():.2f}</p>
+                                    <p><strong>3-Class Predicted Label:</strong> {['NEI', 'SUPPORTED', 'REFUTED'][pred_tc]}</p>
+                                    <p><strong>2-Class Probability:</strong> {prob2class.item():.2f}</p>
+                                    <p><strong>2-Class Predicted Label:</strong> {['SUPPORTED', 'REFUTED'][pred_bc]}</p>
+                                """
+                        verdict_time = time.time() - verdict_start_time
 
-                    # Store verification history and the latest result
-                    st.session_state.history.append({
-                        "claim": claim,
-                        "evidence": evidence,
-                        "verdict": verdict
-                    })
-                    st.session_state.latest_result = {
-                        "claim": claim,
-                        "evidence": evidence,
-                        "verdict": verdict,
-                        "details": details
-                    }
+                        # Store verification history and the latest result
+                        st.session_state.history.append({
+                            "claim": claim,
+                            "evidence": evidence,
+                            "verdict": verdict,
+                            "evidence_time": evidence_time,
+                            "verdict_time": verdict_time,
+                            "details": details
+                        })
+                        st.session_state.latest_result = {
+                            "claim": claim,
+                            "evidence": evidence,
+                            "verdict": verdict,
+                            "evidence_time": evidence_time,
+                            "verdict_time": verdict_time,
+                            "details": details
+                        }
 
-                    if torch.cuda.is_available():
-                        torch.cuda.empty_cache()
+                        if torch.cuda.is_available():
+                            torch.cuda.empty_cache()
 
-                # Display the result after verification
-                res = st.session_state.latest_result
-                result_placeholder.markdown(f"""
-                    <div class='result-box'>
-                        <p><strong>Claim:</strong> {res['claim']}</p>
-                        <p><strong>Evidence:</strong> {res['evidence']}</p>
-                        <p class='verdict'><span class='verdict-icon'>{verdict_icons.get(res['verdict'], '')}</span>{res['verdict']}</p>
-                        {res['details']}
-                    </div>
-                """, unsafe_allow_html=True)
-                # Download Verification Result Feature
-                result_text = f"Claim: {res['claim']}\nEvidence: {res['evidence']}\nVerdict: {res['verdict']}\nDetails: {res['details']}"
-                st.download_button("Download Result", data=result_text, file_name="verification_result.txt", mime="text/plain")
+                    # Display the result after verification
+                    res = st.session_state.latest_result
+                    st.markdown(f"""
+                        <div class='result-box'>
+                            <p><strong>Claim:</strong> {res['claim']}</p>
+                            <p><strong>Evidence:</strong> {res['evidence']}</p>
+                            <p><strong>Evidence Inference Time:</strong> {res['evidence_time']:.2f} seconds</p>
+                            <p><strong>Verdict Inference Time:</strong> {res['verdict_time']:.2f} seconds</p>
+                            <p class='verdict'><span class='verdict-icon'>{verdict_icons.get(res['verdict'], '')}</span>{res['verdict']}</p>
+                            {res['details']}
+                        </div>
+                    """, unsafe_allow_html=True)
+                    # Download Verification Result Feature
+                    result_text = f"Claim: {res['claim']}\nEvidence: {res['evidence']}\nVerdict: {res['verdict']}\nDetails: {res['details']}"
+                    st.download_button("Download Result", data=result_text, file_name="verification_result.txt", mime="text/plain")
             else:
                 st.info("No verification result yet.")