app.py

import streamlit as st
from utils import descargar_documento, leer_documento, generar_plan
import os

st.title("Generador de Asistente Virtual")

# Cargar la clave API desde el archivo .env
api_key = st.text_input("Ingrese su clave API de OpenAI:", type="password")

file_id = st.text_input("Ingrese el ID del archivo de Google Drive:")
if st.button("Descargar Documento"):
    descargar_documento(file_id)

if os.path.exists('documento.pdf'):
    st.success("Documento descargado correctamente.")
    if st.button("Leer y Procesar Documento"):
        documento_texto = leer_documento('documento.pdf')
        ruta_plan = generar_plan(documento_texto)
        st.success(f"El plan del asistente virtual se ha guardado en {ruta_plan}")
        st.download_button(label="Descargar Plan", data=open(ruta_plan, 'rb'), file_name='plan_asistente_virtual.pdf')
else:
    st.warning("No se encontró el documento PDF.")