用Python对接本地大模型 快速生成定制微短剧脚本

打开终端先装依赖，别整那些花里胡哨的云端API，本地跑数据不泄露还不用充Token。

依赖安装直接抄命令

pip install openai python-docx

装完不用重启，终端没报红字就是装成了。

本地大模型准备步骤

• 先去Ollama官网下载对应系统的安装包，双击走默认流程安装完就行，不用手动配环境变量
• 新开终端执行模型拉取命令，等进度条走完服务会自动启动，默认服务端口11434，没必要改

ollama run qwen2:7b-instruct

显存不够4G的话把模型名换成qwen2:1.8b-instruct，跑不动别硬扛大参数模型，出活优先。

可直接运行的生成脚本

注意配置本地接口的时候，api_key不需要填真实的密钥，随便写个字符串就能通，这是Ollama兼容OpenAI接口的特性。

from openai import OpenAI
from docx import Document
import os

# 本地接口不需要真实密钥，随便填字符串就行
client = OpenAI(
    base_url="http://localhost:11434/v1",
    api_key="ollama"
)

def gen_short_drama(theme, role_set, episode_num=3):
    prompt = f"""
    你是微短剧编剧，按照以下要求写单集1分钟左右、爽点密集的短剧脚本：
    核心主题：{theme}
    人物设定：{role_set}
    集数：{episode_num}集
    每集要包含明确冲突、小反转、结尾留钩子，台词接地气别太书面化。
    """
    resp = client.chat.completions.create(
        model="qwen2:7b-instruct",
        messages=[{"role":"user","content":prompt}],
        temperature=0.8
    )
    content = resp.choices[0].message.content
    # 自动导出成Word文档存到当前文件夹
    doc = Document()
    doc.add_heading(f"{theme}短剧脚本", 0)
    doc.add_paragraph(content)
    save_path = f"./{theme}_短剧脚本.docx"
    doc.save(save_path)
    print(f"脚本已保存到：{os.path.abspath(save_path)}")
    return content

if __name__ == "__main__":
    # 改成你自己需要的题材和设定就行
    gen_short_drama(
        theme="外卖员其实是隐退集团总裁",
        role_set="男主：30岁左右，隐姓埋名体验生活的上市集团总裁；女主：刚创业的奶茶店老板，善良直爽；反派：到处收保护费的街溜子头头",
        episode_num=3
    )

几个容易踩的坑提前说：

• 运行脚本前一定要确认Ollama服务在正常运行，拉模型的那个终端别关，关了服务就断了
• 脚本生成速度看设备配置，一般等一小会儿就能出结果，别着急强制终止进程
• 想写甜宠、悬疑、复仇类题材，直接改函数入参里的主题和人设就行，不用动核心逻辑
• 生成的内容不用全盘照搬，本来就是帮你搭框架省初稿时间的，挑可用的爽点改改就能用

反正我平时攒短剧选题的时候就靠这个工具出初稿，比坐屏幕前硬憋一下午效率高多了。你要是有攒了很久的私藏提示词，直接塞到prompt变量里就能用，没必要为了这点需求整啥复杂的可视化框架。