refactor: using openai api key from env vars

app.py

@@ -1,46 +1,60 @@
+from dotenv import load_dotenv
+import os
 import streamlit as st
 from cases.optimize_content import optimize_content_with_langchain
 from cases.content_generation import content_generation_with_langchain
 
+load_dotenv()
+
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+def display_home():
+    st.title("Bienvenido a la página de inicio")
+    st.markdown('''
+        En esta aplicación, exploraremos los siguientes casos de uso de SEO potenciado por IA:
+        1. **Optimización de Contenido**: Las herramientas de AI pueden sugerir mejoras a tu contenido, como cambios en la redacción o la ubicación de palabras clave, para que sea más amigable para SEO. También pueden analizar el rendimiento SEO de tu contenido existente y sugerir mejoras.
+        2. **Generación de Contenido**: Las herramientas de AI pueden generar tu contenido, como artículos, newsletters o posts para RRSS, de modo que sea más amigable para SEO.
+        Para continuar, selecciona el caso de uso en la barra lateral.
+        ''')
+
+def handle_seo_action(text, action, action_text):
+    if text:
+        with st.spinner(f'{action_text}...'):
+            return action(text, OPENAI_API_KEY)
+    return None
+
+def display_content_optimization():
+    st.title("Optimización de Contenido")
+    st.markdown("Por favor, copia y pega tu artículo a continuación:")
+    article_content = st.text_area("Contenido del Artículo")
+
+    optimized_content = handle_seo_action(article_content, optimize_content_with_langchain, 'Optimizando el contenido para SEO')
+    if optimized_content:
+        st.success('Optimización de contenido completada.')
+        st.write(optimized_content)
+
+def display_content_generation():
+    st.title("Generación de Contenido")
+    st.markdown("Por favor, una descripción del tipo de contenido que deseas generar:")
+    content_type = st.text_area("Tipo de Contenido (Artículo de ventas, Posts sobre IA en RRSS, Newsletter de Suscripcion mensual sobre Tecnologia)")
+
+    generated_content = handle_seo_action(content_type, content_generation_with_langchain, 'Creando el contenido optimizado para SEO')
+    if generated_content:
+        st.success('Generación de contenido completada.')
+        st.write(generated_content)
+
+MENU_FUNCTIONS = {
+    'Home': display_home,
+    'Optimización de Contenido': display_content_optimization,
+    'Generación de Contenido': display_content_generation,
+}
+
 def main():
     st.sidebar.title("Menú")
-    selection = st.sidebar.radio("SEO Powered By AI", ['Home', 'Optimización de Contenido', 'Generación de Contenido'])
-
-    if selection == 'Home':
-        st.title("Bienvenido a la página de inicio")
-        st.markdown('''
-            En esta aplicación, exploraremos los siguientes casos de uso de SEO potenciado por IA:
-            1. **Optimización de Contenido**: Las herramientas de AI pueden sugerir mejoras a tu contenido, como cambios en la redacción o la ubicación de palabras clave, para que sea más amigable para SEO. También pueden analizar el rendimiento SEO de tu contenido existente y sugerir mejoras.
-            2. **Generación de Contenido**: Las herramientas de AI pueden generar tu contenido, como articulos, newsletter o posts para RRSS, de modo que sea más amigable para SEO.
-
-            Para continuar, selecciona el caso de uso en la barra lateral.
-            ''')
-
-    elif selection == 'Optimización de Contenido':
-        st.title("Optimización de Contenido")
-        st.markdown("Por favor, copia y pega tu artículo y tu clave de API de OpenAI a continuación:")
-        openai_key = st.text_input("Clave API OpenAI", type='password')
-        article_content = st.text_area("Contenido del Artículo")
-
-        if article_content and openai_key:
-            with st.spinner('Optimizando el contenido para SEO...'):
-                optimized_content = optimize_content_with_langchain(article_content, openai_key)
-
-            st.success('Optimización de contenido completada.')
-            st.write(optimized_content)
-    
-    elif selection == 'Generación de Contenido':
-        st.title("Generación de Contenido")
-        st.markdown("Por favor, completa los siguientes campos y tu clave de API de OpenAI a continuación:")
-        openai_key = st.text_input("Clave API OpenAI", type='password')
-        content_type = st.text_area("Tipo de Contenido (Articulo, Posts, Newsletter)")
-
-        if content_type and openai_key:
-            with st.spinner('Creando el contenido optimizado para SEO...'):
-                optimized_content = content_generation_with_langchain(content_type, openai_key)
-
-            st.success('Optimización de contenido completada.')
-            st.write(optimized_content)
+    selection = st.sidebar.radio("SEO Powered By AI", list(MENU_FUNCTIONS.keys()))
+
+    menu_function = MENU_FUNCTIONS[selection]
+    menu_function()
 
 if __name__ == '__main__':
     main()

cases/content_generation.py

@@ -1,3 +1,4 @@
+import streamlit as st
 from langchain import LLMChain
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts.chat import (
@@ -19,6 +20,7 @@ def content_generation_with_langchain(content, openai_key):
     4. Optimización SEO: Asegúrate de que tus palabras clave aparecen en los lugares importantes de tu contenido, como el título, los subtítulos, el primer y último párrafo, y la descripción de la meta. 
     5. Analisis: Debes analizar el contenido generado con el fin de identificar palabras claves que puedan optimizarse para SEO, puntos de mejora en la estructura y detalle del contenido.
     """
+    st.text_area(template)
     system_message_prompt = SystemMessagePromptTemplate.from_template(template)
     human_template = "{content}"
     human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)

cases/optimize_content.py

@@ -1,3 +1,4 @@
+import streamlit as st
 from langchain import LLMChain
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts.chat import (
@@ -7,7 +8,10 @@ from langchain.prompts.chat import (
 )
 
 def optimize_content_with_langchain(content, openai_key):
-    chat = ChatOpenAI(temperature=0, openai_api_key=openai_key)
+    chat = ChatOpenAI(
+        temperature=0, 
+        openai_api_key=openai_key
+    )
 
     template = """
     Debes actuar como un agente experto en SEO y Marketing Digital, y utilizando tus habilidades y conocimientos
@@ -17,6 +21,7 @@ def optimize_content_with_langchain(content, openai_key):
     2. Proporcionar alternativas para titulo con el fin de generar más tracción a que los usuarios realicen click.
     3. Proporcionar una lista de keywords que puedan incluirse en el contenido del articulo y que NO se encuentren en el contenido proporcionado por el usuario.
     """
+    st.text_area(template)
     system_message_prompt = SystemMessagePromptTemplate.from_template(template)
     human_template = "{content}"
     human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)

requirements.txt

@@ -1,5 +1,4 @@
 streamlit
 openai
 langchain
-requests
-BeautifulSoup4
+python-dotenv