refactor: helicone

app.py

@@ -31,7 +31,7 @@ def display_content_optimization():
     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)
+        st.text_area('Contenido Generado', optimized_content)
 
 def display_content_generation():
     st.title("Generación de Contenido")
@@ -41,7 +41,7 @@ def display_content_generation():
     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)
+        st.text_area('Contenido Generado', generated_content)
 
 MENU_FUNCTIONS = {
     'Home': display_home,

cases/content_generation.py

@@ -1,3 +1,4 @@
+from monitoring import HEADERS
 import streamlit as st
 from langchain import LLMChain
 from langchain.chat_models import ChatOpenAI
@@ -7,8 +8,14 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
 )
 
+
 def content_generation_with_langchain(content, openai_key):
-    chat = ChatOpenAI(temperature=0, openai_api_key=openai_key)
+    chat = ChatOpenAI(
+        model="gpt-3.5-turbo-0613",
+        temperature=0, 
+        openai_api_key=openai_key,
+        headers=HEADERS,
+    )
 
     template = """
     Debes actuar como un agente experto en SEO y Marketing Digital, y utilizando tus habilidades y conocimientos
@@ -20,6 +27,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.markdown("**Instrucciones del bot:**")
     st.text_area(template)
     system_message_prompt = SystemMessagePromptTemplate.from_template(template)
     human_template = "{content}"

cases/monitoring.py

@@ -0,0 +1,12 @@
+import openai
+from os import getenv
+
+openai.api_base = "https://oai.hconeai.com/v1"
+HELICONE_API_KEY = getenv("HELICONE_API_KEY")
+
+HEADERS = {
+    "Helicone-Auth": f"Bearer {HELICONE_API_KEY}",
+    "Helicone-Cache-Enabled": "true",
+    "Helicone-Property-App": "HuggingFace",
+    "Helicone-Property-DataSource": "SEO Powered by AI",
+}

cases/optimize_content.py

@@ -7,10 +7,14 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
 )
 
+from monitoring import HEADERS
+
 def optimize_content_with_langchain(content, openai_key):
     chat = ChatOpenAI(
+        model="gpt-3.5-turbo-0613",
         temperature=0, 
-        openai_api_key=openai_key
+        openai_api_key=openai_key,
+        headers=HEADERS
     )
 
     template = """
@@ -21,7 +25,8 @@ 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)
+    st.markdown("**Instrucciones del bot:**")
+    st.text_area(template) 
     system_message_prompt = SystemMessagePromptTemplate.from_template(template)
     human_template = "{content}"
     human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)