| | 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
|
| | ''' |
