Danielchris145/TruthCheck-AI
0
1�
2 ��.h� � � � d dl Zd dlmZ d dlmZ d dlmZ d dlm Z d dl3m Z e� Z e� Z
e� Z e � Z G d� d� Z e� Zd � Z ej& e ej( d4d d �
� ej* d�� dddgdgdgdgdggej, j/ � �� Zedk( r ed� ej7 ddd � � yy) � N)�ClaimExtractor)�KeywordExtractor)�EvidenceRetriever)�
NLIClassifier)�calculate_similarityc � � e Zd Zd� Zd� Zy)�TruthCheckSystemc �\ � t | _ t | _ t | _ t | _ y )N)�claim_extractor�keyword_extractor�evidence_retriever�nli_classifier)�selfs �E:\TruthCheck\run.py�__init__zTruthCheckSystem.__init__ s" � � .�� �!2�� �"4�� � ,��� c �. � | j j |� }|sy|d }| j j |� }| j j |� }|sy|D �cg c] }t
||d � dkD r|�� }}|sy| j j ||d d � }|d dk( rd } |d5 }6n|d d k( rd } |d7 }8nd
} d}9|d d d d dz } d|d d � �} | |10| � d| � �fS c c}w # t $ r}
dddt |
� � �fcY d }
~
S d }
~
ww xY w)N)zNo valid claims found� z(Please provide a clear factual statementr )�Low Confidenceg333333�?z"Not enough reliable evidence found�contentg �?)r g�������?z'No semantically relevant evidence found�label�11ENTAILMENT�True�12confidence�
CONTRADICTION�Falser �� z...zSource: �sourcez13 14�Errorr zAn error occurred: ) r �extract_claimsr �extract_keywordsr
� get_evidencer r �classify� Exception�str)r �text�claims�claim�keywords�evidence�item�relevant_evidence�resultr r �evidence_snippetr �es r � verify_claimz TruthCheckSystem.verify_claim s� � �, @��)�)�8�8��>�F��_� �1�I�E� �-�-�>�>�u�E�H� �.�.�;�;�H�E�H��R� "*�!� �'��t�I��?�#�E� �!� � !� %�W� �(�(�1�1�%�9J�1�9M�i�9X�Y�F� �g��,�.� �� #�L� 1�15��� �O�3� �� #�L� 1�16�(�� �17� 0�� 3�I� >�t�� D�u� L� � � 1�!� 4�X� >� ?�@�F��*�)9�(:�$�v�h�&G�G� G��1!��2 � @��C�#6�s�1�v�h�!?�?� ?�� @�sA � C0 �=C0 � C0 �"C+�=C0 �A(C0 �+C0 �0 D�9D� D�DN)�__name__�18__module__� __qualname__r r0 � r r r r s
� �-� -@r r c � � t j | � \ }}}|dz d�d�}|dk( rd}d}n|dk( rd}d }nd19}d }d |� d
|� d|� d|� d|� d� }t S )N�d z.1f�%r �greenu ✅r �redu ❌�orangeu ⚠️zW20<div style='padding: 20px; border-radius: 10px; margin: 10px;'>21 <h2 style='color: z;'>� z*</h2>22 <p><strong>Confidence:</strong> z�</p>23 <p><strong>Evidence:</strong></p>24 <p style='background-color: #f0f0f0; color: #000; padding: 10px; border-radius: 5px;'>z </p>25</div>26)�27truthcheckr0 �formatted_result)�28claim_textr r r* �confidence_pct� result_color�emoji�fformatted_results r �gradio_interfacerC I s� � �",�"9�"9�*�"E� �E�:�x�"�S�(��-�Q�/�N� ��� � ��� �'� � � ��� � ��� �#�n�C��w�a��w� 7%�%3�$4� 5[�[c�Zd� e� �� �r zEnter a claim to verify:z+E.g., 'NASA landed a rover on Mars in 2021'� )r � placeholder�lineszVerification Result)r u- 🔍 TruthCheck: AI-Powered Fact VerificationzbEnter any factual claim and get it verified using AI and real-time evidence from credible sources.z#NASA landed a rover on Mars in 2021zIndia is a continentzThe Earth is flatz2Donald Trump won the 2024 US Presidential electionz$COVID-19 vaccines contain microchips)�fn�inputs�outputs�title� description�examples�theme�__main__u0 🖥️ Gradio Interface: http://localhost:7860z0.0.0.0i� F)� server_name� server_port�share) �gradio�gr�models.claim_extractorr �models.keyword_extractorr �models.evidence_retrieverr �models.nli_classifierr �utils.similarityr r r r
r r r< rC � Interface�Textbox�HTML�themes�Soft� interfacer1 �print�launchr4 r r �<module>ra s� �� � 1� 5� 7� /� 1� !�"��$�&� �&�(� � � ��4@� 4@�n �
�29� �4
�B�L�L�� �2�:�:�(�A�� �30
�B�G�G�/� 0�319�t� .�/� � � � � =�>� /�0� �
�)�)�.�.�32�#
� �( �z�� �33<�=�
�����E��J� r