AIdeaText commited on
Commit
3ee3961
·
verified ·
1 Parent(s): e5fc1bf

Update modules/text_analysis/discourse_analysis.py

Browse files
modules/text_analysis/discourse_analysis.py CHANGED
@@ -33,14 +33,6 @@ from .stopwords import (
33
 
34
 
35
  #####################
36
- # Define colors for grammatical categories
37
- POS_COLORS = {
38
- 'ADJ': '#FFA07A', 'ADP': '#98FB98', 'ADV': '#87CEFA', 'AUX': '#DDA0DD',
39
- 'CCONJ': '#F0E68C', 'DET': '#FFB6C1', 'INTJ': '#FF6347', 'NOUN': '#90EE90',
40
- 'NUM': '#FAFAD2', 'PART': '#D3D3D3', 'PRON': '#FFA500', 'PROPN': '#20B2AA',
41
- 'SCONJ': '#DEB887', 'SYM': '#7B68EE', 'VERB': '#FF69B4', 'X': '#A9A9A9',
42
- }
43
-
44
  POS_TRANSLATIONS = {
45
  'es': {
46
  'ADJ': 'Adjetivo', 'ADP': 'Preposición', 'ADV': 'Adverbio', 'AUX': 'Auxiliar',
@@ -56,12 +48,12 @@ POS_TRANSLATIONS = {
56
  'PROPN': 'Proper Noun', 'SCONJ': 'Subordinating Conjunction', 'SYM': 'Symbol',
57
  'VERB': 'Verb', 'X': 'Other',
58
  },
59
- 'fr': {
60
- 'ADJ': 'Adjectif', 'ADP': 'Préposition', 'ADV': 'Adverbe', 'AUX': 'Auxiliaire',
61
- 'CCONJ': 'Conjonction de Coordination', 'DET': 'Déterminant', 'INTJ': 'Interjection',
62
- 'NOUN': 'Nom', 'NUM': 'Nombre', 'PART': 'Particule', 'PRON': 'Pronom',
63
- 'PROPN': 'Nom Propre', 'SCONJ': 'Conjonction de Subordination', 'SYM': 'Symbole',
64
- 'VERB': 'Verbe', 'X': 'Autre',
65
  }
66
  }
67
 
@@ -80,15 +72,14 @@ ENTITY_LABELS = {
80
  "Dates": "lightyellow",
81
  "Concepts": "lightpink"
82
  },
83
- 'fr': {
84
- "Personnes": "lightblue",
85
- "Lieux": "lightcoral",
86
- "Inventions": "lightgreen",
87
- "Dates": "lightyellow",
88
- "Concepts": "lightpink"
89
  }
90
  }
91
-
92
  #################
93
 
94
  def fig_to_bytes(fig, dpi=100):
 
33
 
34
 
35
  #####################
 
 
 
 
 
 
 
 
36
  POS_TRANSLATIONS = {
37
  'es': {
38
  'ADJ': 'Adjetivo', 'ADP': 'Preposición', 'ADV': 'Adverbio', 'AUX': 'Auxiliar',
 
48
  'PROPN': 'Proper Noun', 'SCONJ': 'Subordinating Conjunction', 'SYM': 'Symbol',
49
  'VERB': 'Verb', 'X': 'Other',
50
  },
51
+ 'uk': {
52
+ 'ADJ': 'Прикметник', 'ADP': 'Прийменник', 'ADV': 'Прислівник', 'AUX': 'Допоміжне дієслово',
53
+ 'CCONJ': 'Сурядний сполучник', 'DET': 'Означник', 'INTJ': 'Вигук',
54
+ 'NOUN': 'Іменник', 'NUM': 'Число', 'PART': 'Частка', 'PRON': 'Займенник',
55
+ 'PROPN': 'Власна назва', 'SCONJ': 'Підрядний сполучник', 'SYM': 'Символ',
56
+ 'VERB': 'Дієслово', 'X': 'Інше',
57
  }
58
  }
59
 
 
72
  "Dates": "lightyellow",
73
  "Concepts": "lightpink"
74
  },
75
+ 'uk': {
76
+ "Люди": "lightblue",
77
+ "Місця": "lightcoral",
78
+ "Винаходи": "lightgreen",
79
+ "Дати": "lightyellow",
80
+ "Концепції": "lightpink"
81
  }
82
  }
 
83
  #################
84
 
85
  def fig_to_bytes(fig, dpi=100):