基于ChatGPT实现电影推荐小程序
本帖最后由 nettman 于 2023-4-21 17:46 编辑问题导读:
1、怎样利用上ChatGPT的推荐功能?
2、怎样使用Python调用ChatGPT API?
3、怎样实现电影信息的爬取?
** 效果预览 **
1、根据电影明星推荐
https://www.aboutyun.com/data/attachment/forum/202304/07/1.gif
2、根据兴趣标签推荐
https://www.aboutyun.com/data/attachment/forum/202304/07/2.gif
3、根据电影名推荐
https://www.aboutyun.com/data/attachment/forum/202304/07/3.gif
一、需求背景
在我们日常想看电影的时候,经常会遇到一些问题:
1. 闲来无事想看个电影,打开电影列表,感觉都是看过的,一下子不知道该如何去发现新大陆?
2. 喜欢某个演员,想看与他风格类似的电影,可惜电影网站的影片推荐总是那么不尽人意!
3. 在不同的电影网站,填入自己的感兴趣的标签,结果推荐出来的电影题材并不是自己想要的效果!
既然说起推荐系统,这就刚好踩中了我那研究三年推荐系统的读研苦逼时光了。稍微介绍一下时下主流的推荐系统的架构和算法:
在如此庞大的数据体量和计算引擎的支持下,现如今的推荐系统仍然没有以完美的姿态来解决用户的冷启动问题,所以说时下,推荐系统在学术界的研究已经达到了一种登峰造极的状态,你我都知道可能多引入一些高性能的模型去加强,多跑几轮模型去调参优化,从而实现更美丽的推荐效果!这样我们可能可以得到一篇优秀的论文!但是在工业应用领域,对于推荐算法的优化,新投入的算力跟人工成本,通常并不会由于更准推荐效果从而产生更丰厚的营收,可以说投入跟产出完全不成正比!这对于时下资本退却的互联网来说,这是最要命的!
于是乎,我们可以转换一下思路,有没有什么模型和算法可以实现推荐效果最优化,不惧怕因为用户数据量少而导致的冷启动问题 ------ 那么这个时候ChatGPT获取可以申请一战,他有超海量的全人类用户数据、连续产生内容及记忆理解上下文功能!
好那么,基于此,让我们来用ChatGPT做一个电影推荐小程序! 做一个可以满住你的任意无理要求的电影小程序。
二、项目原理及架构
2.1 实现原理
1. 要利用上ChatGPT的推荐功能,首先构造好目标明确的问题是成功的关键。
2. 在获取到GPT的推荐数据之后,我们需要将推荐结果中的电影内容获取并展示在小程序端,这里我们需要采用Python爬虫对豆瓣电影网进行爬取!
这里我们构造了三类推荐类目: 兴趣标签、电影主题、电影明星
(1)根据用户的兴趣标签
通过用户输入的兴趣标签进行电影的匹配
(2)根据关联类似主题的题材
(3)根据特定的电影明星
2.2 技术架构
2.3 技术栈
三、项目功能的实现
3.1 小程序端设计与实现
首页
标签选择
推荐电影列表页
电影详情页
缺图
3.2 数据后端设计与实现
注重介绍后端服务接入 ChatGPT API,需要按照以下步骤进行操作:
注册一个账号并登录到OpenAI的官网:https://openai.com/
在Dashboard页面上,创建一个API密钥。在“API Keys”选项卡下,点击“Generate New Key”按钮。将生成的密钥保存好,以备后续使用。
选择所需的API服务,例如“Completion” API,以使用OpenAI的文本生成功能。
使用Python调用ChatGPT API实现代码如下:
方法一:使用request库
import requests
import json
# 构建API请求
url = "https://api.openai.com/v1/engines/davinci-codex/completions"
headers = {"Content-Type": "application/json",
"Authorization": "Bearer YOUR_API_KEY"}
data = {
"prompt": "Hello, my name is",
"max_tokens": 5
}
# 发送API请求
response = requests.post(url, headers=headers, data=json.dumps(data))
# 解析API响应
response_data = json.loads(response.text)
generated_text = response_data["choices"]["text"]
print(generated_text)
方式二:使用openAI库
from flask import Flask, request
import openai
app = Flask(__name__)
openai.api_key = "YOUR_API_KEY_HERE"
@app.route("/")
def home():
return "Hello, World!"
@app.route("/chat", methods=["POST"])
def chat():
data = request.json
response = openai.Completion.create(
engine="davinci",
prompt=data["message"],
max_tokens=60
)
return response.choices.text
if __name__ == "__main__":
app.run()
3.3 数据智能获取功能设计与实现
定义一个函数来实现电影信息的爬取,该函数的输入参数为电影名,输出为该电影的名称、导演、主演、类型、上映时间、评分等信息。具体实现步骤如下:
构造请求 URL,其中电影名需要进行 URL 编码。
发送 HTTP 请求,获取豆瓣电影页面的 HTML 内容。
使用 lxml 库解析 HTML 文档,提取电影信息。
import requests
from lxml import etree
def crawl_movie_info(movie_name):
# 1. 构造请求 URL
url = f'https://www.douban.com/search?q={movie_name}'
# 2. 发送 HTTP 请求,获取 HTML 内容
response = requests.get(url)
html = response.content.decode()
# 3. 解析 HTML 文档,提取电影信息
tree = etree.HTML(html)
# 获取搜索结果列表中的第一个电影条目
movie_link = tree.xpath('//div[@class="result"]/div[@class="content"]/h3/a/@href')
response = requests.get(movie_link)
html = response.content.decode()
# 解析电影页面 HTML 文档,提取电影信息
tree = etree.HTML(html)
# 电影名称
movie_title = tree.xpath('//span[@property="v:itemreviewed"]/text()')
# 导演
directors = tree.xpath('//a[@rel="v:directedBy"]/text()')
director = ' / '.join(directors)
# 主演
actors = tree.xpath('//span[@class="actor"]/span[@class="attrs"]/a/text()')
actor = ' / '.join(actors)
# 类型
genres = tree.xpath('//span[@property="v:genre"]/text()')
genre = ' / '.join(genres)
# 上映日期
release_date = tree.xpath('//span[@property="v:initialReleaseDate"]/text()')
# 评分
rating = tree.xpath('//strong[@class="ll rating_num"]/text()')
# 返回电影信息
return {
'电影名称': movie_title,
'导演': director,
'主演': actor,
'类型': genre,
'上映时间': release_date,
'评分': rating,
}
我们可以通过调用该函数,传入电影名参数来获取电影信息。例如:
Copy code
movie_name = '大赢家'
movie_info = crawl_movie_info(movie_name)
print(movie_info)
输出结果如下:
{'电影名称': '大赢家', '导演': '李伟 / 黄伟明', '主演': '赵本山 / 贾玲 / 小沈阳 / 张子枫 / 李春
最新经典文章,欢迎关注公众号http://www.aboutyun.com/data/attachment/forum/201903/18/215536lzpn7n3u7m7u90vm.jpg
---------------------
作者:陶人超有料
来源:csdn
原文:微信小程序 | 基于ChatGPT实现电影推荐小程序
页:
[1]