working with anthropic

app.py

@@ -3,11 +3,13 @@ import gradio as gr
 from openai import OpenAI
 import os
 from dotenv import load_dotenv
+import anthropic
 
 # Replace 'your_api_key_here' with your actual OpenAI API key
 # For better security, use environment variables or other secure methods to store and retrieve your API key.
 load_dotenv()
-client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+openai_client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+anthropic_client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
 
 
 prompt = """
@@ -109,10 +111,11 @@ puedes instalar unas pequeñas cortinas, que sean ligeras para que dejen pasar l
 por barillas estilo bambú en la barandilla, que además aportan un toque natural.\
 """
 
-def generate_article(title):
-    client = OpenAI()
-    completion = client.chat.completions.create(
+def generate_article_openai(title):
+    openai_client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+    completion = openai_client.chat.completions.create(
         model="gpt-4-turbo-preview",
+        temperature=1.0,
         messages=[
             {"role": "system", "content": prompt},
             {"role": "user", "content": f"Ahora que tienes tus instrucciones y referencias redacta un artículo basado en el siguiente título.\
@@ -121,8 +124,29 @@ def generate_article(title):
         ],
         
     )
-    article = completion.choices[0].message.content
-    return article
+    openai_article = completion.choices[0].message.content
+    return openai_article
+
+def generate_article_anthropic(title):    
+    anthropic_client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
+    message = anthropic_client.messages.create(
+        model="claude-3-sonnet-20240229",
+        max_tokens=2000,
+        temperature=1.0,
+        system=prompt,
+        messages=[
+            {"role": "user", "content": f"Ahora que tienes tus instrucciones y referencias redacta un artículo basado en el siguiente título.\
+            Título:'{title}'.\
+            Artículo:"}
+        ]
+    )
+    anthropic_article = message.content[0].text
+    return anthropic_article
+
+def generate_articles(title):
+    openai_article =  generate_article_openai(title)
+    anthropic_article =  generate_article_anthropic(title)
+    return openai_article, anthropic_article
 
 # Define your title, description, and examples
 title = "Primera demo del asistente de redacción."
@@ -145,12 +169,16 @@ css = """
 """
 
 iface = gr.Interface(
-    fn=generate_article, 
-    inputs="text", 
-    outputs="text",
-    title=title, 
-    description=description, 
-    examples=examples,
-    css=css # Add the custom CSS
+    fn=generate_articles,
+    inputs=gr.Textbox(lines=2, placeholder="Enter a title here..."),
+    outputs=[
+        gr.Textbox(label="OpenAI Output"),
+        gr.Textbox(label="Anthropic Output")
+    ],
+    title=title,
+    description=description,
+    examples=examples
 )
+
+# Run the interface
 iface.launch()

requirements.txt

@@ -1,2 +1,3 @@
 openai
-python-dotenv
+python-dotenv
+anthropic