Upload 4 files

modules/utils/export_utils.py

@@ -0,0 +1,70 @@
+import streamlit as st
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+from docx import Document
+import io
+
+def export_data(user_data, t, format='pdf'):
+    if format == 'pdf':
+        return export_to_pdf(user_data, t)
+    elif format == 'docx':
+        return export_to_docx(user_data, t)
+    else:
+        raise ValueError(f"Unsupported format: {format}")
+
+def export_to_pdf(user_data, t):
+    buffer = io.BytesIO()
+    c = canvas.Canvas(buffer, pagesize=letter)
+    width, height = letter
+
+    # Título
+    c.setFont("Helvetica-Bold", 16)
+    c.drawString(50, height - 50, t['analysis_report'])
+
+    # Resumen
+    c.setFont("Helvetica", 12)
+    c.drawString(50, height - 80, f"{t['morpho_analyses']}: {len(user_data['morphosyntax_analyses'])}")
+    c.drawString(50, height - 100, f"{t['semantic_analyses']}: {len(user_data['semantic_analyses'])}")
+    c.drawString(50, height - 120, f"{t['discourse_analyses']}: {len(user_data['discourse_analyses'])}")
+
+    # Aquí agregarías más detalles de los análisis...
+
+    c.save()
+    buffer.seek(0)
+    return buffer
+
+def export_to_docx(user_data, t):
+    doc = Document()
+    doc.add_heading(t['analysis_report'], 0)
+
+    doc.add_paragraph(f"{t['morpho_analyses']}: {len(user_data['morphosyntax_analyses'])}")
+    doc.add_paragraph(f"{t['semantic_analyses']}: {len(user_data['semantic_analyses'])}")
+    doc.add_paragraph(f"{t['discourse_analyses']}: {len(user_data['discourse_analyses'])}")
+
+    # Aquí agregarías más detalles de los análisis...
+
+    buffer = io.BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return buffer
+
+def display_export_options(t):
+    format = st.radio(t['select_export_format'], ['PDF', 'DOCX'])
+    if st.button(t['export']):
+        user_data = st.session_state.user_data
+        if format == 'PDF':
+            buffer = export_data(user_data, t, format='pdf')
+            st.download_button(
+                label=t['download_pdf'],
+                data=buffer,
+                file_name="analysis_report.pdf",
+                mime="application/pdf"
+            )
+        elif format == 'DOCX':
+            buffer = export_data(user_data, t, format='docx')
+            st.download_button(
+                label=t['download_docx'],
+                data=buffer,
+                file_name="analysis_report.docx",
+                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+            )

modules/utils/spacy_utils.py

@@ -0,0 +1,9 @@
+# modules/spacy_utils.py
+import spacy
+
+def load_spacy_models():
+    return {
+        'es': spacy.load("es_core_news_lg"),
+        'en': spacy.load("en_core_web_lg"),
+        'uk': spacy.load("uk_core_news_lg")
+    }

modules/utils/svg_to_png_converter.py

@@ -0,0 +1,51 @@
+import io
+from svglib.svglib import svg2rlg
+from reportlab.graphics import renderPM
+from pymongo import MongoClient
+import base64
+
+# Asume que tienes una función para obtener la conexión a MongoDB
+from ..database.mongo_db import get_mongodb
+
+def convert_svg_to_png(svg_string):
+    """Convierte una cadena SVG a una imagen PNG."""
+    drawing = svg2rlg(io.BytesIO(svg_string.encode('utf-8')))
+    png_bio = io.BytesIO()
+    renderPM.drawToFile(drawing, png_bio, fmt="PNG")
+    return png_bio.getvalue()
+
+def save_png_to_database(username, analysis_id, png_data):
+    """Guarda la imagen PNG en la base de datos."""
+    client = get_mongodb()
+    db = client['aideatext_db']  # Asegúrate de usar el nombre correcto de tu base de datos
+    collection = db['png_diagrams']
+
+    png_base64 = base64.b64encode(png_data).decode('utf-8')
+
+    document = {
+        'username': username,
+        'analysis_id': analysis_id,
+        'png_data': png_base64
+    }
+
+    result = collection.insert_one(document)
+    return result.inserted_id
+
+def process_and_save_svg_diagrams(username, analysis_id, svg_diagrams):
+    """Procesa una lista de diagramas SVG, los convierte a PNG y los guarda en la base de datos."""
+    png_ids = []
+    for svg in svg_diagrams:
+        png_data = convert_svg_to_png(svg)
+        png_id = save_png_to_database(username, analysis_id, png_data)
+        png_ids.append(png_id)
+    return png_ids
+
+# Función para recuperar PNGs de la base de datos
+def get_png_diagrams(username, analysis_id):
+    """Recupera los diagramas PNG de la base de datos para un análisis específico."""
+    client = get_mongodb()
+    db = client['aideatext_db']
+    collection = db['png_diagrams']
+
+    diagrams = collection.find({'username': username, 'analysis_id': analysis_id})
+    return [base64.b64decode(doc['png_data']) for doc in diagrams]

modules/utils/widget_utils.py

@@ -0,0 +1,7 @@
+# modules/utils/widget_utils.py
+import streamlit as st
+
+def generate_unique_key(module_name, element_type="input", username=None):
+    # Si el nombre de usuario no se pasa explícitamente, lo toma de session_state
+    username = username or st.session_state.username
+    return f"{module_name}_{element_type}_{username}"