Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -84,7 +84,10 @@ with st.form(key='my_form'):
|
|
84 |
text_input = st.text_area(label='Enter a sentence')
|
85 |
submit_button = st.form_submit_button(label='Analyze')
|
86 |
|
87 |
-
#
|
|
|
|
|
|
|
88 |
if text_input:
|
89 |
if selected_language == "German":
|
90 |
doc_de = model(text_input)
|
@@ -98,27 +101,11 @@ if text_input:
|
|
98 |
entity_data = get_entity_data(formatted_wikidata_url)
|
99 |
if entity_data is not None:
|
100 |
entities_data[entity_string] = entity_data
|
101 |
-
|
102 |
-
doc_de = model(text_input)
|
103 |
-
# Map entities to a format similar to English output
|
104 |
-
entities = [(ent.text, ent.label_, ent._.kb_qid, ent._.url_wikidata) for ent in doc_de.ents]
|
105 |
-
# Debug
|
106 |
-
#for ent in doc_de.ents:
|
107 |
-
# st.write(f"Entity: {ent.text}, Label: {ent.label_}, QID: {ent._.kb_qid}, URL: {ent._.url_wikidata}")
|
108 |
else:
|
109 |
entities = model.process_text(text_input)
|
110 |
|
111 |
-
|
112 |
-
entities_map = {}
|
113 |
-
entities_data = {}
|
114 |
-
for entity in entities:
|
115 |
-
if selected_language == "German":
|
116 |
-
entity_string, entity_type, wikidata_id, wikidata_url = entity
|
117 |
-
entities_map[entity_string] = {"id": wikidata_id, "link": wikidata_url}
|
118 |
-
entity_data = get_entity_data(wikidata_url)
|
119 |
-
if entity_data is not None:
|
120 |
-
entities_data[entity_string] = entity_data
|
121 |
-
else:
|
122 |
single_entity_list = str(entity).strip('][').replace("\'", "").split(', ')
|
123 |
if len(single_entity_list) >= 2 and "wikidata" in single_entity_list[1]:
|
124 |
entities_map[single_entity_list[0].strip()] = get_wikidata_id(single_entity_list[1])
|
|
|
84 |
text_input = st.text_area(label='Enter a sentence')
|
85 |
submit_button = st.form_submit_button(label='Analyze')
|
86 |
|
87 |
+
# Initialization
|
88 |
+
entities_map = {}
|
89 |
+
entities_data = {}
|
90 |
+
|
91 |
if text_input:
|
92 |
if selected_language == "German":
|
93 |
doc_de = model(text_input)
|
|
|
101 |
entity_data = get_entity_data(formatted_wikidata_url)
|
102 |
if entity_data is not None:
|
103 |
entities_data[entity_string] = entity_data
|
104 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
105 |
else:
|
106 |
entities = model.process_text(text_input)
|
107 |
|
108 |
+
for entity in entities:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
109 |
single_entity_list = str(entity).strip('][').replace("\'", "").split(', ')
|
110 |
if len(single_entity_list) >= 2 and "wikidata" in single_entity_list[1]:
|
111 |
entities_map[single_entity_list[0].strip()] = get_wikidata_id(single_entity_list[1])
|