AIdeaText commited on
Commit
93baec5
·
verified ·
1 Parent(s): 5971e1f

Update modules/text_analysis/semantic_analysis.py

Browse files
modules/text_analysis/semantic_analysis.py CHANGED
@@ -11,6 +11,7 @@ from collections import Counter, defaultdict
11
  from sklearn.feature_extraction.text import TfidfVectorizer
12
  from sklearn.metrics.pairwise import cosine_similarity
13
  import logging
 
14
 
15
  logger = logging.getLogger(__name__)
16
 
@@ -71,100 +72,6 @@ ENTITY_LABELS = {
71
  }
72
  }
73
 
74
- CUSTOM_STOPWORDS = {
75
- 'es': {
76
- # Artículos
77
- 'el', 'la', 'los', 'las', 'un', 'una', 'unos', 'unas',
78
- # Preposiciones comunes
79
- 'a', 'ante', 'bajo', 'con', 'contra', 'de', 'desde', 'en',
80
- 'entre', 'hacia', 'hasta', 'para', 'por', 'según', 'sin',
81
- 'sobre', 'tras', 'durante', 'mediante',
82
- # Conjunciones
83
- 'y', 'e', 'ni', 'o', 'u', 'pero', 'sino', 'porque',
84
- # Pronombres
85
- 'yo', 'tú', 'él', 'ella', 'nosotros', 'vosotros', 'ellos',
86
- 'ellas', 'este', 'esta', 'ese', 'esa', 'aquel', 'aquella',
87
- # Verbos auxiliares comunes
88
- 'ser', 'estar', 'haber', 'tener',
89
- # Palabras comunes en textos académicos
90
- 'además', 'también', 'asimismo', 'sin embargo', 'no obstante',
91
- 'por lo tanto', 'entonces', 'así', 'luego', 'pues',
92
- # Números escritos
93
- 'uno', 'dos', 'tres', 'primer', 'primera', 'segundo', 'segunda',
94
- # Otras palabras comunes
95
- 'cada', 'todo', 'toda', 'todos', 'todas', 'otro', 'otra',
96
- 'donde', 'cuando', 'como', 'que', 'cual', 'quien',
97
- 'cuyo', 'cuya', 'hay', 'solo', 'ver', 'si', 'no',
98
- # Símbolos y caracteres especiales
99
- '#', '@', '/', '*', '+', '-', '=', '$', '%'
100
- },
101
- 'en': {
102
- # Articles
103
- 'the', 'a', 'an',
104
- # Common prepositions
105
- 'in', 'on', 'at', 'by', 'for', 'with', 'about', 'against',
106
- 'between', 'into', 'through', 'during', 'before', 'after',
107
- 'above', 'below', 'to', 'from', 'up', 'down', 'of',
108
- # Conjunctions
109
- 'and', 'or', 'but', 'nor', 'so', 'for', 'yet',
110
- # Pronouns
111
- 'i', 'you', 'he', 'she', 'it', 'we', 'they', 'this',
112
- 'that', 'these', 'those', 'my', 'your', 'his', 'her',
113
- # Auxiliary verbs
114
- 'be', 'am', 'is', 'are', 'was', 'were', 'been', 'have',
115
- 'has', 'had', 'do', 'does', 'did',
116
- # Common academic words
117
- 'therefore', 'however', 'thus', 'hence', 'moreover',
118
- 'furthermore', 'nevertheless',
119
- # Numbers written
120
- 'one', 'two', 'three', 'first', 'second', 'third',
121
- # Other common words
122
- 'where', 'when', 'how', 'what', 'which', 'who',
123
- 'whom', 'whose', 'there', 'here', 'just', 'only',
124
- # Symbols and special characters
125
- '#', '@', '/', '*', '+', '-', '=', '$', '%'
126
- },
127
- 'fr': {
128
- # Articles
129
- 'le', 'la', 'les', 'un', 'une', 'des',
130
- # Prepositions
131
- 'à', 'de', 'dans', 'sur', 'en', 'par', 'pour', 'avec',
132
- 'sans', 'sous', 'entre', 'derrière', 'chez', 'avant',
133
- # Conjunctions
134
- 'et', 'ou', 'mais', 'donc', 'car', 'ni', 'or',
135
- # Pronouns
136
- 'je', 'tu', 'il', 'elle', 'nous', 'vous', 'ils',
137
- 'elles', 'ce', 'cette', 'ces', 'celui', 'celle',
138
- # Auxiliary verbs
139
- 'être', 'avoir', 'faire',
140
- # Academic words
141
- 'donc', 'cependant', 'néanmoins', 'ainsi', 'toutefois',
142
- 'pourtant', 'alors',
143
- # Numbers
144
- 'un', 'deux', 'trois', 'premier', 'première', 'second',
145
- # Other common words
146
- 'où', 'quand', 'comment', 'que', 'qui', 'quoi',
147
- 'quel', 'quelle', 'plus', 'moins',
148
- # Symbols
149
- '#', '@', '/', '*', '+', '-', '=', '$', '%'
150
- }
151
- }
152
-
153
- ##############################################################################################################
154
- def get_stopwords(lang_code):
155
- """
156
- Obtiene el conjunto de stopwords para un idioma específico.
157
- Combina las stopwords de spaCy con las personalizadas.
158
- """
159
- try:
160
- nlp = spacy.load(f'{lang_code}_core_news_sm')
161
- spacy_stopwords = nlp.Defaults.stop_words
162
- custom_stopwords = CUSTOM_STOPWORDS.get(lang_code, set())
163
- return spacy_stopwords.union(custom_stopwords)
164
- except:
165
- return CUSTOM_STOPWORDS.get(lang_code, set())
166
-
167
-
168
  def perform_semantic_analysis(text, nlp, lang_code):
169
  """
170
  Realiza el análisis semántico completo del texto.
@@ -178,6 +85,7 @@ def perform_semantic_analysis(text, nlp, lang_code):
178
 
179
  logger.info(f"Starting semantic analysis for language: {lang_code}")
180
  try:
 
181
  doc = nlp(text)
182
  key_concepts = identify_key_concepts(doc)
183
  concept_graph = create_concept_graph(doc, key_concepts)
 
11
  from sklearn.feature_extraction.text import TfidfVectorizer
12
  from sklearn.metrics.pairwise import cosine_similarity
13
  import logging
14
+ from .stopwords import process_text
15
 
16
  logger = logging.getLogger(__name__)
17
 
 
72
  }
73
  }
74
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
75
  def perform_semantic_analysis(text, nlp, lang_code):
76
  """
77
  Realiza el análisis semántico completo del texto.
 
85
 
86
  logger.info(f"Starting semantic analysis for language: {lang_code}")
87
  try:
88
+ tokens = process_text(text, lang_code, nlp)
89
  doc = nlp(text)
90
  key_concepts = identify_key_concepts(doc)
91
  concept_graph = create_concept_graph(doc, key_concepts)