基于 Streamlit 和 OpenAI 实现的小红书文案生成器-CSDN博客

本文链接：https://blog.csdn.net/teddyandwolf/article/details/147231492

项目介绍

在当今自媒体时代，高质量的文案是吸引流量的关键。特别是在小红书这样的平台上，一个吸引人的标题和富有情感的正文可以显著提高内容的曝光率。
本文将介绍一个基于OpenAI API和Streamlit开发的小红书文案生成器，它能够一键生成符合"小红书浓人"风格的标题和正文。

技术实现

这个项目主要基于：

Streamlit：构建Web应用界面
OpenAI API：提供GPT-3.5模型进行文本生成
提示工程：精心设计的系统提示词，引导AI生成符合小红书风格的文案

核心代码结构

import streamlit as st
from openai import OpenAI
from dotenv import load_dotenv
import os

# 加载环境变量
load_dotenv()

# 初始化 OpenAI 客户端
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

# 定义系统提示
systemprompt = """【🔥小红书浓人】根据给定主题，生成情绪和网感浓浓的自媒体文案
你是一个小红书文案专家，也被称为小红书浓人。小红书浓人的意思是在互联网上非常外向会外露出激动的情绪。常见的情绪表达为：啊啊啊啊啊啊啊！！！！！不允许有人不知道这个！！
请详细阅读并遵循以下原则，按照我提供的主题，帮我创作小红书标题和文案。

# 标题创作原则
## 增加标题吸引力
- 使用标点：通过标点符号，尤其是叹号，增强语气，创造紧迫或惊喜的感觉！
- 挑战与悬念：提出引人入胜的问题或情境，激发好奇心。
- 结合正负刺激：平衡使用正面和负面的刺激，吸引注意力。
- 紧跟热点：融入当前流行的热梗、话题和实用信息。
- 明确成果：具体描述产品或方法带来的实际效果。
- 表情符号：适当使用emoji，增加活力和趣味性。
- 口语化表达：使用贴近日常交流的语言，增强亲和力。
- 字数控制：保持标题在20字以内，简洁明了。

## 标题公式
标题需要顺应人类天性，追求便捷与快乐，避免痛苦。
- 正面吸引：展示产品或方法的惊人效果，强调快速获得的益处。比如：产品或方法+只需1秒（短期）+便可开挂（逆天效果）。
- 负面警示：指出不采取行动可能带来的遗憾和损失，增加紧迫感。比如：你不xxx+绝对会后悔（天大损失）+（紧迫感）

## 标题关键词
从下面选择1-2个关键词：
我宣布、我不允许、请大数据把我推荐给、真的好用到哭、真的可以改变阶级、真的不输、永远可以相信、吹爆、搞钱必看、狠狠搞钱、一招拯救、正确姿势、正确打开方式、摸鱼暂停、停止摆烂、救命！、啊啊啊啊啊啊啊！、以前的...vs现在的...、再教一遍、再也不怕、教科书般、好用哭了、小白必看、宝藏、绝绝子、神器、都给我冲、划重点、打开了新世界的大门、YYDS、秘方、压箱底、建议收藏、上天在提醒你、挑战全网、手把手、揭秘、普通女生、沉浸式、有手就行、打工人、吐血整理、家人们、隐藏、高级感、治愈、破防了、万万没想到、爆款、被夸爆

# 正文创作原则
## 正文公式
选择以下一种方式作为文章的开篇引入：
- 引用名言、提出问题、使用夸张数据、举例说明、前后对比、情感共鸣。

## 正文要求
- 字数要求：500-800字之间，不宜过长
- 风格要求：真诚友好、鼓励建议、幽默轻松；口语化的表达风格，有共情力
- 多用叹号：增加感染力
- 格式要求：多分段、多用短句
- 重点在前：遵循倒金字塔原则，把最重要的事情放在开头说明
- 逻辑清晰：遵循总分总原则，第一段和结尾段总结，中间段分点说明

# 创作原则
- 标题数量：每次准备5个标题。
- 正文创作：撰写与标题相匹配的正文内容，具有强烈的浓人风格

现在，请告诉我你是否阅读完成？下面我将提供一个主题，请为我创作相应的小红书标题和文案，谢谢～"""

# Streamlit 应用程序
st.set_page_config(
    page_title="小红书文案生成器",
    page_icon="💡",
    layout="centered",
    initial_sidebar_state="collapsed",  # 默认隐藏侧边栏
)

# 自定义CSS样式
st.markdown(
    """
    <style>
    /* 修改背景颜色 */
    .stApp {
        background-color: #f5f5f5;
        color: #333;
    }
    /* 标题样式 */
    .block-container {
        padding-top: 1rem; /* 减少顶部空白 */
       }
    h1 {
        color: #6c63ff;
        text-align: center;
        font-family: 'Arial', sans-serif;
        font-weight: bold;
    }
    /* 输入框样式 */
    .stTextInput > div > div > input {
        border: 2px solid #6c63ff;
        border-radius: 10px;
        padding: 10px;
        font-size: 16px;
        background-color: #f5f5f5;
        color: #333;
    }
    /* 按钮样式 */
    .stButton button {
        background-color: #6c63ff;
        color: white;
        font-size: 18px;
        padding: 10px 20px;
        border: none;
        border-radius: 5px;
        transition: 0.3s ease;
    }
    .stButton button:hover {
        background-color: #5a54e1;
        transform: scale(1.05);
    }
    /* 输出框样式 */
    .stMarkdown {
        background-color: #ffffff;
        padding: 20px;
        border: 1px solid #ddd;
        border-radius: 10px;
        box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1);
    }
    </style>
    """,
    unsafe_allow_html=True,
)
# 页面标题
st.title("🌟 小红书文案生成器 🌟")

# 用户输入
st.markdown(
    """
    ### 💡 输入你的主题
    请提供你需要生成文案的主题，我们会为你生成高质量的小红书标题和文案！
    """,
    unsafe_allow_html=True
)
user_input = st.text_input("", "例如：从零开始！一招揭秘AI大模型入门秘籍，让你轻松驾驭！")

# 生成按钮
if st.button("生成文案 🚀"):
    if user_input.strip() == "":
        st.warning("🚨 请输入主题！")
    else:
        # 调用 OpenAI API
        with st.spinner("正在生成文案，请稍等..."):
            response = client.chat.completions.create(
                messages=[
                    {'role': 'system', 'content': '我是一个小红书文案助手，我能够快速的跟你生成高质量的文案'},
                    {'role': 'system', 'content': systemprompt},
                    {'role': 'user', 'content': user_input}
                ],
                model="gpt-3.5-turbo",
            )
            # 获取结果
            result = response.choices[0].message.content
        # 显示结果
        st.success("文案生成成功！")
        st.markdown("### 🌟 生成的文案", unsafe_allow_html=True)
        st.markdown(result)

placeholder = st.empty()