|
from typing import Dict, Any
|
|
import base64
|
|
from io import BytesIO
|
|
from matplotlib.figure import Figure
|
|
|
|
class FlexibleAnalysisHandler:
|
|
def __init__(self, analysis_data):
|
|
self.data = analysis_data
|
|
|
|
def get_key_concepts(self):
|
|
return self.data.get('key_concepts', [])
|
|
|
|
def get_concept_graph(self):
|
|
return self.data.get('concept_graph')
|
|
|
|
def get_entity_graph(self):
|
|
return self.data.get('entity_graph')
|
|
|
|
|
|
def get_graph(self, graph_type):
|
|
return self.data.get(graph_type)
|
|
|
|
|
|
|
|
|
|
'''
|
|
class FlexibleAnalysisHandler:
|
|
def __init__(self, analysis_data: Dict[str, Any]):
|
|
self.data = analysis_data
|
|
|
|
def get_key_concepts(self):
|
|
if 'key_concepts' in self.data:
|
|
return self.data['key_concepts']
|
|
elif 'word_count' in self.data:
|
|
# Convertir word_count a un formato similar a key_concepts
|
|
return [(word, count) for word, count in self.data['word_count'].items()]
|
|
return []
|
|
|
|
def get_graph(self):
|
|
if 'graph' in self.data:
|
|
# Decodificar la imagen base64
|
|
image_data = base64.b64decode(self.data['graph'])
|
|
return BytesIO(image_data)
|
|
elif 'arc_diagrams' in self.data:
|
|
# Devolver el primer diagrama de arco como SVG
|
|
return self.data['arc_diagrams'][0]
|
|
return None
|
|
|
|
def get_pos_analysis(self):
|
|
return self.data.get('pos_analysis', [])
|
|
|
|
def get_morphological_analysis(self):
|
|
return self.data.get('morphological_analysis', [])
|
|
|
|
def get_sentence_structure(self):
|
|
return self.data.get('sentence_structure', [])
|
|
|
|
# Agregar más métodos según sea necesario para otros tipos de análisis
|
|
''' |
