blacksquadece/Knowledge_graph_MachineLearning
0
1U
2 CJ�a7 � @ s d dl mZ d dlmZ d dlZd dlm mZ d dl Z3d dl m Z
d dlmZ d dlmZ d dlmZ d dlZd dlZd dlmZ dZd g g g g d4�Zd d � Zd
d� Zdd� Z dd� Z dd� Z dd� Z e� d� ej!dddd� d d � Z"ej#d d k�re� e"� dS ) � )�disable)� EggMetadataN)�Network)� SessionState)�Key)� clip_textz test.htmlF)�has_run�wiki_suggestions� wiki_text�nodes�topicsc C s* t �� D ] \} }| tjkr|tj| < qd S �N)�state_variables�items�st�
session_state)�k�v� r �?/mnt/c/Users/karel/Desktop/work/chapter work/REBEL/REBEL/app.py�wiki_init_state_variables s 5r c C s� dt jkrd S tt jd �dkr.t �d� d S t jdd��j t jd } g t jd< t�| t�}td|� |D ]*}|� � }|t jd krjt jd �6|� qjd t jd7< W 5 Q R X t � d � d S ) Nr8 r z1please enter a topic and select a wiki page firstzGenerating graph...��textr � gen_graphr Tr zDone!) r r �len�error�spinner�rebel�generate_knowledge_graph�GRAPH_FILENAME�print�lower�append�success)�textsr �nr r r �wiki_generate_graph s 9101112 13r& c C st g t jd< t jdd��T t jd dkrft jj} | d k rf| �d�}|D ] }t jd tj|dd�7 < qDW 5 Q R X d S ) Nr zfetching wiki topics...r � input_method� wikipedia�,� )�results)r r r r �splitr( �search)r �subjects�subjr r r �wiki_show_suggestion/ s 1415r0 c
C s� t jdd��� z<tj| dd�}t jd �t|j�� t jd �| �� � W nv tj k16r� } zVt jdd��@ t jd �17| � t jd |j d d � }t t
|��t jd< W 5 Q R X W 5 d }~X Y nX W 5 Q R X d S )18Nzfetching wiki page...r F��title� auto_suggestr19 r �/Woops, ambigious term, recalculating options...r r* )r r r( �pager r" r �summaryr! �DisambiguationError�remove�options�list�set)�20page_titler5 �e�tempr r r �wiki_show_text9 s r? c C s� z<t tj| dd�j�}tjd �|� tjd �| �� � W n� tjk21r� } zVtj dd��@ tjd �22| � tjd |j d d� }t t
|��tjd < W 5 Q R X W 5 d }~X Y n& tjk23r� tjd �24| � Y nX d S )25NFr1 r26 r r4 r r r* �node)r r( r5 r6 r r r" r! r7 r r8 r9 r: r; �WikipediaException)�term�27extra_textr= r> r r r �
wiki_add_textE s ,rD c C s t D ] } tj| = qd S r
)r r r )r r r r �wiki_reset_sessionR s rE z$REBELious knowledge graph generationz input method)r( z free textr' )�keyc C s� t j�d� t jjdtdd� t �ddg�} | d � t jdtd d28� W 5 Q R X | d �* t �d � t �d � t jd td
d� W 5 Q R X t t j29d �dk�rDt t j30d �}|dk r�|nd}t �dg| �}t d|| �D ]h}t t
|t j31d || |d | � ��D ]:\}\}}|�" t j|t|ft|�| d� W 5 Q R X �qq�t t j32d �dk�r�t t j33d �D ]B\}}t j|d d� d |dkd�} | � t �|� W 5 Q R X �qft t j34d �dk�r�t jdtdd� t j35d �r�t j�d� ttddd�}36|37�� } tj| dd d � t t j38d �}|d k �r(|nd }t �dg| dg �}tt j39d � t d|| �D ]j}t t
|t j40d || |d | � ��D ]:\}\}}|�" t j|t|ft|�| d� W 5 Q R X �q��q`d S ) Nz}41# how to42- Enter wikipedia search terms, separated by comma's43- Choose one or more of the suggested pages44- Click generate!45�Reset� reset_key)�on_clickrF � � r zwikipedia search termr )� on_changerF � �Search�show_suggestion_keyr )rI �argsrF r46 � z...)�label�expanded�Generater r z�47 # How to expand the graph48 - Click a button on the right to expand that node49 - Only nodes that have wiki pages will be expanded50 - Hit the Generate button again to expand your graph!51 �rzutf-8)�encodingi� iX )�width�heightr � )r �sidebar�markdown�buttonrE �columns�52text_inputr0 r r r �range� enumerate�zipr? �str�expanderr& �openr �read�53components�htmlr rD ) �cols� num_buttons�num_colsr] �q�i�c�s�t� new_expander�HtmlFile� source_coder r r �show_wiki_hub_page\ sR �545556574, � 4rs r( )$�loggingr �
pkg_resourcesr � streamlitr �streamlit.components.v1rf �v1�networkx�nx�matplotlib.pyplot�pyplot�plt�
pyvis.networkr � streamlit.state.session_stater �streamlit.type_utilr r r( �utilsr r r r r& r0 r? rD rE r2 � selectboxrs r r r r r �<module> sD �58
59 �=