用Python搭建本地离线短剧脚本生成工具

打开终端直接敲下面的安装命令，自己本地玩不用刻意搭隔离环境，省事儿。

pip install openai python-docx

这套调用逻辑走的是OpenAI兼容接口，不用付费买云服务，下个开源量化模型就能完全离线跑。

本地大模型启动配置

• 选7B参数级别的4bit量化模型就行，跑起来的资源占用看设备，一般带独显的普通笔记本都能带得动
• 启动模型的时候一定要开启OpenAI兼容接口，常用的本地模型启动工具（Ollama、LM Studio）都自带这个开关，勾选即可
• 默认本地接口地址一般是http://localhost:11434/v1 ，如果自己改了端口就记下来，后面代码里要对应修改，接口密钥随便填字符串就行，本地运行不做校验

提醒：别硬上大参数模型，不然风扇狂转半天出不来字，反而耽误写脚本的效率。

核心生成代码直接复制修改

from openai import OpenAI
from docx import Document

# 初始化本地客户端，对应修改端口和密钥即可
client = OpenAI(
    base_url="http://localhost:11434/v1",
    api_key="sk-local"
)

def generate_short_drama(theme, episode_count=3):
    prompt = f"""
    你是专业短剧编剧，围绕{theme}主题写{episode_count}集竖屏短剧脚本，
    每集包含场景、人物对话、反转点三个部分，语言口语化，节奏快，每集结尾留钩子。
    """
    response = client.chat.completions.create(
        model="qwen2.5:7b", # 换成你本地下载的模型名即可
        messages=[{"role":"user","content":prompt}],
        temperature=0.7
    )
    return response.choices[0].message.content

# 生成后自动保存为Word文档
if __name__ == "__main__":
    drama_content = generate_short_drama(theme="外卖员意外救下集团老总", episode_count=5)
    doc = Document()
    doc.add_heading("生成短剧脚本", level=1)
    doc.add_paragraph(drama_content)
    doc.save("我的短剧脚本.docx")
    print("生成完成，去脚本同目录找Word文件即可")

• model字段必须和你本地启动的模型名完全一致，不然会报模型不存在的错误，不知道名字的可以去模型启动器的已下载列表里查
• theme参数直接替换成你想写的短剧题材就行，婆媳矛盾、重生逆袭、赘婿爽文这类常见题材都适配
• episode_count改对应数字就能控制生成集数，别一次填太大的数字，不然等生成的时间够你喝半杯奶茶
• temperature参数调到0.6-0.8之间最合适，太低了输出内容太死板，太高了剧情容易飞得没边

反正我自己平时攒短剧灵感的时候就用这个，比开网页找各种在线AI工具方便，也不用担心写的题材被平台爬走。跑的时候要是卡就把模型量化等级调高点，别死磕全精度版本。