Update app.py

app.py

@@ -1,83 +1,64 @@
-import streamlit as st
-import openai
-import os
+# app.py
+import gradio as gr
+import anthropic
 
-# Set up OpenAI API key
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Configura el cliente de la API de Claude (Anthropic)
+client = anthropic.Anthropic(api_key="my_api_key")  # Reemplaza "my_api_key" con tu clave de API
 
-def generate_video_script(title, formality_level, emotional_tone, engagement_level):
-    # Construct the original detailed prompt as provided
-    prompt = (
-        f"[Use this video framework below]\n"
-        f"First we need to have a strong hook.\n"
-        f"A short intro that stops people from scrolling and holds their attention\n"
-        f"Example. \"{title}\"\n"
-        f"Now we add in “The Setup\"\n"
-        f"* This is used to Setup the scene and provide context to the Viewer\n"
-        f"* ex. \"I've been creating videos recently and I guess something finally clicked\"\n"
-        f"* This lets the viewer know that she’s been uploading more videos recently, and that could have led to an increase in views and followers.\n"
-        f"ACT 2: \"The Conflict\"\n"
-        f"* Introduce a point of conflict or problem that requires a solution\n"
-        f"* ex. \"Some of you have been asking about scripting and storytelling and there's a lot to talk about, but I gotchu\"\n"
-        f"* Setting up the foundation to lead to the topic of the video and posing the concern that 'there's a lot to talk about'\n"
-        f"* Which opens the question of how we're going to tackle the topic in a limited amount of time\n"
-        f"ACT 3: \"The Resolution\"\n"
-        f"* Addressing the conflict we introduced and resolving it\n"
-        f"* ex. \"Here's how I tell my stories. All you need are these 3: ACT I, ACT II, and ACT III.\"\n"
-        f"* She proceeds to explain how we structure and tell our stories, which answers and resolves the initial conflict\n"
-        f"A call to action or loop.\n"
-        f"Call-To-Action\n"
-        f"* After giving valuable information to your viewer, you have to be able to lead them somewhere or encourage them to engage\n"
-        f"* ex. \"And if you wanna know how that looks like, then watch this video back and follow for more.\"\n"
-        f"* This encourages the audience to rewatch the video to see the video is an actual example of what they just learned and it encourages them to follow and share the content if they found it valuable!\n"
-        f"Follow viral video recipe above in creating a Viral Video script for me. Do not repeat the instructions above, just use the framework for the viral video script. Every output must include: HOOK, THE SETUP, THE CONFLICT, THE RESOLUTION, THE CTA. \n"
-        f"Topic: {title}\n"
-        f"Tone: {emotional_tone}\n"
-        f"Formality Level: {formality_level}\n"
-        f"Engagement Level: {engagement_level}"
+def generate_headlines(number_of_headlines, target_audience, product):
+    # Llama a la API de Claude para generar titulares
+    message = client.messages.create(
+        model="claude-3-5-sonnet-20240620",
+        max_tokens=1000,
+        temperature=0,
+        system="You are a world-class copywriter, with experience in creating hooks, headlines, and subject lines that immediately capture attention. Your skill lies in deeply understanding the emotions, desires, and challenges of a specific audience, allowing you to design personalized marketing strategies that resonate and motivate action. You know how to use proven structures to attract your target audience, generating interest and achieving a powerful connection that drives desired results in advertising and content campaigns.\n\nAnswer in Spanish.",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": f"Generate {number_of_headlines} attention-grabbing headlines designed for {target_audience} to generate interest in {product}. Each headline should be crafted to encourage a specific action, such as making a purchase, signing up, or downloading. Use a variety of formats (questions, bold statements, intriguing facts) to test different approaches."
+                    }
+                ]
+            }
+        ]
     )
+    return message.content
 
-    try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=[
-                {"role": "system", "content": "Generate a script based on the following instructions."},
-                {"role": "user", "content": prompt}
-            ],
-            temperature=0.7,
-            max_tokens=1024,
-        )
-        # Accessing the generated script from the response
-        script = response['choices'][0]['message']['content'].strip()
-        return script
-    except Exception as e:
-        st.error(f"Failed to generate script: {str(e)}")
-        return "Error in generating script."
+# Configura la interfaz de usuario con Gradio
+def gradio_generate_headlines(number_of_headlines, target_audience, product):
+    return generate_headlines(number_of_headlines, target_audience, product)
 
-# Streamlit UI setup
-st.set_page_config(layout="wide")
-st.markdown("<h1 style='text-align: center; color: black;'>Viral Video Script Generator</h1>", unsafe_allow_html=True)
+# Define los colores de la interfaz según el logo de Anthropic (ejemplo)
+logo_colors = {
+    "background": "#f8f8f8",
+    "primary": "#2c7be5",
+    "text_color": "#212529"
+}
 
-# Create columns for input and output
-col1, col2 = st.columns([1, 2])
-
-with col1:
-    st.markdown("<h2 style='text-align: center; color: black;'>Video Information</h2>", unsafe_allow_html=True)
-    video_title = st.text_input("Title of Video", placeholder="Enter the title of your video/video idea")
-    formality_level = st.selectbox("Formality Level", ["Casual", "Neutral", "Formal"])
-    emotional_tone = st.selectbox("Emotional Tone", ["Positive", "Neutral", "Humorous"])
-    engagement_level = st.selectbox("Engagement Level", ["Interactive", "Informative", "Persuasive"])
-
-with col2:
-    st.markdown("<h2 style='text-align: center; color: black;'>Viral Script</h2>", unsafe_allow_html=True)
-    st.markdown('**Note:** Fill out the fields on the left, then click Generate')
+with gr.Blocks(css=".gradio-container { background-color: " + logo_colors["background"] + "; }") as demo:
+    gr.Markdown(
+        f"""
+        <h1 style="color: {logo_colors['primary']}; text-align: center;">Generador de Titulares</h1>
+        <p style="color: {logo_colors['text_color']}; text-align: center;">Usa el poder de Claude AI para crear titulares atractivos</p>
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column():
+            number_of_headlines = gr.Number(label="Número de Titulares", value=5)
+            target_audience = gr.Textbox(label="Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios")
+            product = gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés")
+            submit_btn = gr.Button("Generar Titulares", elem_id="submit-btn")
+    
+    output = gr.Textbox(label="Titulares Generados", lines=10)
 
-generate_button = col1.button('Generate Video Script')
+    submit_btn.click(
+        fn=gradio_generate_headlines,
+        inputs=[number_of_headlines, target_audience, product],
+        outputs=output
+    )
 
-if generate_button:
-    if not video_title:
-        st.error("Please enter the title of the video.")
-    else:
-        video_script = generate_video_script(video_title, formality_level, emotional_tone, engagement_level)
-        with col2:
-            st.write(video_script)
+# Lanza la interfaz
+demo.launch()