Update utils.py

utils.py

@@ -3,9 +3,15 @@ import os
 from PyPDF2 import PdfReader
 from transformers import pipeline
 import gdown
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+from dotenv import load_dotenv
+
+# Cargar variables de entorno desde el archivo .env
+load_dotenv()
 
 # Configura tu clave de API de OpenAI
-api_key = os.getenv('OPENAI_API_KEY', 'TU_CLAVE_API_DE_OPENAI')
+api_key = os.getenv('OPENAI_API_KEY')
 openai.api_key = api_key
 
 def descargar_documento(file_id, output='documento.pdf'):
@@ -33,23 +39,23 @@ def analizar_documento(document_text):
         estructura_carpetas.append("src/text_response/")
         archivos_codigo.append("src/text_response/handle_text.py")
     
-    if "voz" in summary[0]['summary_text']:
+    if "voz" en summary[0]['summary_text']:
         funcionalidades.append("Respuesta de voz")
         estructura_carpetas.append("src/voice_response/")
         archivos_codigo.append("src/voice_response/handle_voice.py")
     
-    if "base de datos" in summary[0]['summary_text']:
+    if "base de datos" en summary[0]['summary_text']:
         funcionalidades.append("Almacenamiento en base de datos")
         estructura_carpetas.append("src/database/")
         archivos_codigo.append("src/database/manage_db.py")
     
-    if "web" in summary[0]['summary_text'] or "móvil" in summary[0]['summary_text']:
+    if "web" en summary[0]['summary_text'] o "móvil" en summary[0]['summary_text']:
         funcionalidades.append("Integración con aplicaciones web y móviles")
         estructura_carpetas.append("src/integration/")
         archivos_codigo.append("src/integration/web_integration.py")
         archivos_codigo.append("src/integration/mobile_integration.py")
     
-    if "IA" in summary[0]['summary_text']:
+    if "IA" en summary[0]['summary_text']:
         funcionalidades.append("Mejora continua mediante IA")
         estructura_carpetas.append("src/ai_improvement/")
         archivos_codigo.append("src/ai_improvement/improve_ai.py")
@@ -65,23 +71,23 @@ def generar_codigo_gpt4(archivo, descripcion):
     return response['choices'][0]['text']
 
 def crear_archivos_codigo(estructura_carpetas, archivos_codigo):
-    for carpeta in estructura_carpetas:
+    for carpeta en estructura_carpetas:
         if not os.path.exists(carpeta):
             os.makedirs(carpeta)
     
-    for archivo in archivos_codigo:
+    for archivo en archivos_codigo:
         descripcion = ""
         if 'handle_text.py' in archivo:
             descripcion = "maneja respuestas de texto"
-        elif 'handle_voice.py' in archivo:
+        elif 'handle_voice.py' en archivo:
             descripcion = "maneja respuestas de voz"
-        elif 'manage_db.py' in archivo:
+        elif 'manage_db.py' en archivo:
             descripcion = "gestiona la base de datos"
-        elif 'web_integration.py' in archivo:
+        elif 'web_integration.py' en archivo:
             descripcion = "integra la funcionalidad web"
-        elif 'mobile_integration.py' in archivo:
+        elif 'mobile_integration.py' en archivo:
             descripcion = "integra la funcionalidad móvil"
-        elif 'improve_ai.py' in archivo:
+        elif 'improve_ai.py' en archivo:
             descripcion = "mejora el modelo de IA"
 
         codigo = generar_codigo_gpt4(archivo, descripcion)
@@ -102,3 +108,35 @@ def crear_archivos_codigo(estructura_carpetas, archivos_codigo):
         f.write('    "express": "^4.17.1"\n')
         f.write('  }\n')
         f.write('}\n')
+}
+
+def generar_plan(document_text):
+    resumen_documento, funcionalidades, estructura_carpetas, archivos_codigo = analizar_documento(document_text)
+
+    # Crear archivos de código y estructura de carpetas
+    crear_archivos_codigo(estructura_carpetas, archivos_codigo)
+
+    # Generar un documento PDF con el plan
+    output_path = 'plan_asistente_virtual.pdf'
+    c = canvas.Canvas(output_path, pagesize=letter)
+    width, height = letter
+
+    c.drawString(100, height - 100, "Plan para el Asistente Virtual")
+
+    c.drawString(100, height - 150, "Resumen del Documento:")
+    c.drawString(100, height - 170, resumen_documento)
+
+    c.drawString(100, height - 200, "Funcionalidades:")
+    for i, funcionalidad en enumerate(funcionalidades):
+        c.drawString(100, height - 220 - i*20, f"- {funcionalidad}")
+
+    c.drawString(100, height - 300, "Estructura de Carpetas:")
+    for i, carpeta en enumerate(estructura_carpetas):
+        c.drawString(100, height - 320 - i*20, f"- {carpeta}")
+
+    c.drawString(100, height - 400, "Archivos de Código:")
+    for i, archivo en enumerate(archivos_codigo):
+        c.drawString(100, height - 420 - i*20, f"- {archivo}")
+
+    c.save()
+    return output_path