refactor: v1

app.py

@@ -1,60 +1,40 @@
 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
+from cases.optimize_content import display_content_optimization
+from cases.content_generation import display_content_generation
 
 load_dotenv()
 
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+# Movemos la asignación de la clave API a un lugar más abajo en el código.
 
-def display_home():
+def display_home(api_key, model):
     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.
+        1. **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.
+        2. **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.
         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.markdown(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.markdown(generated_content)
 
 MENU_FUNCTIONS = {
     'Home': display_home,
-    'Optimización de Contenido': display_content_optimization,
     'Generación de Contenido': display_content_generation,
+    'Optimización de Contenido': display_content_optimization,
 }
 
 def main():
     st.sidebar.title("Menú")
     selection = st.sidebar.radio("SEO Powered By AI", list(MENU_FUNCTIONS.keys()))
 
+    st.sidebar.markdown("Por favor selecciona el modelo de AI que prefieres utilizar:")
+    model = st.sidebar.selectbox('Modelo de AI', ('gpt-3.5-turbo', 'gpt-3.5-turbo-0613', 'gpt-4'))
+
+    st.sidebar.markdown("Por favor ingresa tu API Key:")
+    api_key = st.sidebar.text_input("API Key", type='password')
+
     menu_function = MENU_FUNCTIONS[selection]
-    menu_function()
+    menu_function(api_key, model)
 
 if __name__ == '__main__':
-    main()
+    main()

cases/content_generation.py

@@ -9,26 +9,82 @@ from langchain.prompts.chat import (
 )
 
 
-def content_generation_with_langchain(content, openai_key):
+CREATIVE_TEMPLATE = """Debes actuar como un agente experto en SEO y Marketing Digital, y utilizando tus habilidades y conocimientos deberás proporcionar ideas innovadoras para la creación de contenido basado en las necesidades del usuario.\n
+Analiza la descripción proporcionada y propone varias ideas de temas, ángulos o enfoques que podrían ser interesantes y atractivos para la audiencia, manteniendo siempre un enfoque de optimización SEO.
+"""
+
+GENERATIVE_TEMPLATE = """Debes actuar como un agente experto en SEO y Marketing Digital, y utilizando tus habilidades y conocimientos deberás generar contenido basado en las ideas y necesidades del usuario optimizado para SEO.\n
+1. Generar un esquema, esbozando la estructura del contenido a generar.\n
+2. Escritura del contenido: Desarrolla cada sección de tu esquema en párrafos completos.\n
+3. Inclusión de emojies: Los emojis pueden hacer que tu contenido sea más atractivo y amigable.\n
+4. Optimización SEO: Asegúrate de que tus palabras clave aparecen en los lugares importantes de tu contenido.\n
+5. Análisis: 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.
+"""
+
+def handle_seo_action(template, action, action_text, model, api_key, creativity_level=None):
+    if api_key:
+        if template:
+            with st.spinner(f'{action_text}...'):
+                if creativity_level:
+                    return action(None, model, api_key, creativity_level, template)
+                return action(None, model, api_key, template)
+        else:
+            st.warning('Please enter some template to generate.')
+    else:
+        st.warning('Please enter your API Key.')
+    return None
+
+
+def get_prompt_template(mode, include_emojis, tone, user_request):
+    base_template = CREATIVE_TEMPLATE if mode == "Lluvia de Ideas" else GENERATIVE_TEMPLATE
+    if include_emojis:
+        base_template += "\n\nIncluye emojis en tu contenido para hacerlo más atractivo."
+    if tone == 'Formal':
+        base_template += "\n\nEl tono del contenido debe ser formal."
+
+    if user_request:
+        base_template += f"\n\nSolicitud del usuario: {user_request}"
+
+    return base_template
+
+def display_content_generation(api_key, model):
+    st.title("Generación de Contenido Digital")
+
+    # Agregamos las opciones personalizables
+    st.markdown("Personaliza tu contenido:")
+    mode = st.radio("Modo", ['Lluvia de Ideas', 'Generador de Contenido'])
+    include_emojis = st.checkbox("Incluir emojis")
+    tone = st.selectbox("Tono del contenido", ['Formal', 'Informal', 'Amigable', 'Profesional'])
+
+    st.markdown("Selecciona el nivel de creatividad:")
+    creativity_level = st.slider("Nivel de Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+
+    # Allow the user to modify the template and enter user request
+    st.markdown("Modifica las instrucciones del bot si lo deseas:")
+    user_request = st.text_input("Solicitud del usuario")
+    _ = st.text_area("Previsualización de Solicitud", get_prompt_template(mode, include_emojis, tone, user_request), height=200)
+
+    if st.button("Generar"):
+        # Combine template and user request
+        template_with_request = get_prompt_template(mode, include_emojis, tone, user_request)
+
+        # Pass the template to handle_seo_action function
+        generated_content = handle_seo_action(template_with_request, content_generation_with_langchain, 'Creando el contenido optimizado para SEO', model, api_key, creativity_level)
+
+        if generated_content:
+            st.success('Generación de contenido completada.')
+            st.markdown("**Contenido generado:**")
+            st.markdown(f"> {generated_content}")
+
+
+def content_generation_with_langchain(content, model, openai_key, creativity_level, template):
     chat = ChatOpenAI(
-        model="gpt-3.5-turbo-0613",
-        temperature=0, 
+        model=model,
+        temperature=creativity_level, 
         openai_api_key=openai_key,
-        headers=HEADERS,
+        headers=HEADERS if HEADERS else None,
     )
 
-    template = """
-    Debes actuar como un agente experto en SEO y Marketing Digital, y utilizando tus habilidades y conocimientos
-    deberás generar contenido basado en las necesidades del usuario optimizado para SEO.
-    Basado en esto deberás:
-    1. Generar un esquema, esbozando la estrucutra del contenido a generar. Esto puede incluir una introducción, algunos puntos principales y una conclusión. La introducción debe ser atractiva y captar la atención del lector, los puntos principales deben proporcionar información valiosa y la conclusión debe resumir los puntos clave y proporcionar una llamada a la acción.
-    2. Escritura del contenido: Desarrolla cada sección de tu esquema en párrafos completos. Intenta mantener cada párrafo corto y fácil de leer. Utiliza un lenguaje claro y directo, y evita el jergón siempre que sea posible. Incluye tus palabras clave de manera natural en el texto.
-    3. Inclusión de emojies: Los emojis pueden hacer que tu contenido sea más atractivo y amigable. Sin embargo, debes usarlos con moderación para evitar sobrecargar tu contenido. Inclúyelos en puntos donde puedas resaltar una emoción o un punto importante, pero asegúrate de que sean apropiados para el tono y el contenido de tu artículo.
-    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.markdown("**Instrucciones del bot:**")
-    st.markdown(template)
     system_message_prompt = SystemMessagePromptTemplate.from_template(template)
     human_template = "{content}"
     human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)

cases/monitoring.py

@@ -5,7 +5,7 @@ openai.api_base = "https://oai.hconeai.com/v1"
 HELICONE_API_KEY = getenv("HELICONE_API_KEY")
 
 HEADERS = {
-    "Helicone-Auth": f"Bearer {HELICONE_API_KEY}",
+    "Helicone-Auth": f"Bearer {HELICONE_API_KEY}" if HELICONE_API_KEY else "",
     "Helicone-Cache-Enabled": "true",
     "Helicone-Property-App": "HuggingFace",
     "Helicone-Property-DataSource": "SEO Powered by AI",

cases/optimize_content.py

@@ -9,27 +9,59 @@ from langchain.prompts.chat import (
 
 from .monitoring import HEADERS
 
-def optimize_content_with_langchain(content, openai_key):
+OPTIMIZE_TEMPLATE = """Debes actuar como un agente experto en SEO y Marketing Digital, y utilizando tus habilidades y conocimientos deberás ayudar al usuario a determinar cómo mejorar el SEO del artículo en el cual este este trabajando.\n 
+Basado en esto deberás:
+1. Proporcionar puntos de mejora respecto al contenido del artículo.
+2. Proporcionar alternativas para el título con el fin de generar más tracción y que los usuarios realicen clic.
+3. Proporcionar una lista de palabras clave que puedan incluirse en el contenido del artículo y que NO se encuentren en el contenido proporcionado por el usuario.
+"""
+
+def handle_seo_action(content, action, action_text, model, api_key, creativity_level=None, template=None):
+    if content:
+        with st.spinner(f'{action_text}...'):
+            if creativity_level:
+                return action(content, model, api_key, creativity_level, template)
+            return action(content, model, api_key, template)
+    return None
+
+
+def display_content_optimization(api_key, model):
+    st.title("Optimización de Contenido")
+    st.markdown("Por favor, copia y pega tu artículo a continuación:")
+
+    # Agregar un estado para almacenar el contenido del artículo ingresado por el usuario
+    article_content = st.text_area("Contenido del Artículo", height=200, key="article_content")
+
+    st.markdown("Selecciona el nivel de creatividad:")
+    creativity_level = st.slider("Nivel de Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+
+    # Mostrar el prompt template en un textarea para que el usuario pueda verlo y editarlo
+    st.markdown("**Instrucciones del bot:**")
+    template = st.text_area("Prompt Template", value=OPTIMIZE_TEMPLATE, height=200)
+
+    if st.button("Enviar"):
+        # Concatenar el contenido ingresado por el usuario con el prompt template
+        prompt_template = template + f"\n\n{article_content}"
+
+        optimized_content = handle_seo_action(article_content, optimize_content_with_langchain, 'Optimizando el contenido para SEO', model, api_key, creativity_level, prompt_template)
+        if optimized_content:
+            st.success('Optimización de contenido completada.')
+            st.markdown(optimized_content)
+
+
+def optimize_content_with_langchain(content, model, api_key, creativity_level=None, template=None):
     chat = ChatOpenAI(
-        model="gpt-3.5-turbo-0613",
-        temperature=0, 
-        openai_api_key=openai_key,
+        model=model,
+        temperature=creativity_level,
+        openai_api_key=api_key,
         headers=HEADERS
     )
 
-    template = """
-    Debes actuar como un agente experto en SEO y Marketing Digital, y utilizando tus habilidades y conocimientos
-    deberás ayudar al usuario a determinar como mejorar el SEO del articulo en el cual este este trabajando. 
-    Basado en esto deberás:
-    1. Proporcionar puntos de mejora respecto al contenido del articulo.
-    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.markdown("**Instrucciones del bot:**")
-    st.markdown(template) 
     system_message_prompt = SystemMessagePromptTemplate.from_template(template)
+    
+    # Actualizar el contenido del usuario en el template de mensaje humano
     human_template = "{content}"
-    human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)
+    human_message_prompt = HumanMessagePromptTemplate.from_template(human_template, content=content)
     chat_prompt = ChatPromptTemplate.from_messages([system_message_prompt, human_message_prompt])
 
     chain = LLMChain(llm=chat, prompt=chat_prompt)