0
0

ChatGPT写Flask-Demo——有体验地址

ChatGPT中API的试用

最近ChatGPT智能AI很火，可以写代码，DEBUG，写简历等等
上去查看了一下，是可以调用API的，本着学习的态度就是用这个款AI辅助我写了一个demo（本人原来对flask只懂一点点）

目录

先看效果：

0x01主页

主要功能是验证API，可以用默认的（考虑到多人使用，所以用自己的更好）
在这里插入图片描述

0x02图片API

主要功能就是自带的API，创建图片、变体图片、编辑图片
在这里插入图片描述

0x03上传图片

作为一个网站怎么能没有上传口 =。= 在这里插入图片描述

0x04调用的API自动保存

主要是第二部调用的api创建的图片等进行自动保存
在这里插入图片描述

0x05上传视频

上传视频，AI写的，真牛啊
在这里插入图片描述

0x06日志功能

AI还写了自动后台记录日志功能
在这里插入图片描述

0x07目录结构

在这里插入图片描述

0x08APP.py

  1. @app.route('/')defindex():
  3.     key ={"api":-1}if k !="":
  4.         key["key"]=1else:
  5.         key["key"]=0
  7.     logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{key}>")return render_template('index.html', key=key)@app.route('/submit', methods=['POST'])defsubmit():
  8.     key ={"api":-1}if k !="":
  9.         key["key"]=1else:
  10.         key["key"]=0
  11.     logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{key}>")return render_template('index.html', key=key)@app.route('/image')defimage1():
  12.     key ={"key":0}
  13.     logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{key}>")return render_template('image.html',key=key,key2=key,key3=key)@app.route('/text')deftext():
  14.     key =-1
  15.     filenames = os.listdir("static\\text\\")# 打印文件名
  16.     items_gx =[]for filename in filenames:
  17.         items_gx.append(f"text/{filename}")
  18.     logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{key}>")return render_template('text.html',key=key,items_gx=items_gx)@app.route('/test_api',methods=['POST'])deftest_api():
  19.     key ={"key":1}
  20.     timestamp = time.time()try:
  21.         org = request.form['org']
  22.         api = request.form['api_key']
  23.         f =open("temp","w", encoding="utf-8")
  24.         f.write(api)
  25.         f.close()
  26.         openai.organization = org
  27.         openai.api_key_path ="temp"
  28.         api_dict[org]= api
  29.         try:
  30.             k = openai.Model.list()
  31.             key["api"]=1
  32.             f =open("share-api/org","a+", encoding="utf-8")
  33.             f.write(f"[{timestamp}]{org}\n")
  34.             f.close()
  35.             f =open("share-api/api","a+", encoding="utf-8")
  36.             f.write(f"[{timestamp}]{api}\n")
  37.             f.close()except:
  38.             key["api"]=0
  39.             openai.organization ="org-tuPuSBX5NRzKaE06CVvU2dHD"
  40.             openai.api_key_path ="api_key.txt"
  41.         logger.debug(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<[{api_dict}>")return render_template('index.html', key=key)except Exception as e:
  42.         logger.debug(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<[{e}>")return e
  44. @app.route('/img_sub1',methods=['POST'])defimg_sub1():
  45.     key ={"key":1}
  46.     keyn ={"key":0}
  47.     items =[]try:
  48.         prompt = request.form['prompt']
  49.         size = request.form['size']
  50.         num =int(request.form['num'])ifint(num)>10:for i inrange(0,num,10):
  51.                 sleep(31)
  52.                 ret = openai.Image.create(prompt=prompt, n=10, size=size)for i in ret['data']:
  53.                     items.append(i['url'])else:
  54.             ret = openai.Image.create(prompt=prompt, n=num, size=size)for i in ret['data']:
  55.                 items.append(i['url'])for i in items:
  56.             md5_name = md5(i)
  57.             response = requests.get(i)# 打开图片
  58.             img = Image.open(BytesIO(response.content))
  59.             img.save(f"{save_cimg}{md5_name}.png", optimize=True, quality=50)# urllib.request.urlretrieve(i, f"{save_cimg}{md5_name}.png")
  60.         logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{key}>::<{keyn}>::<{items}>")return render_template('image.html', key=key,key2=keyn,key3=keyn, items=items)except openai.error.OpenAIError as e:
  61.         stat =f"[{e.http_status}]{e.error}"
  62.         logger.debug(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<[{e.http_status}]{e.error}>")return stat
  64. @app.route('/img_sub2',methods=['POST'])defimg_sub2():
  65.     key ={"key":1}
  66.     keyn ={"key":0}
  67.     items =[]
  68.     file_src1 =""
  69.     file_src2 =""try:
  70.         image = request.form['main']
  71.         mask = request.form['mask']if image==""or mask =="":
  72.             image = file_src1
  73.             mask = file_src2
  74.         prompt = request.form['prompt']
  75.         size = request.form['size']
  76.         num =int(request.form['num'])ifint(num)>10:for i inrange(0, num,10):
  77.                 sleep(31)
  78.                 ret = openai.Image.create_edit(image=image,mask=mask,prompt=prompt, n=int(num), size=size)for i in ret['data']:
  79.                     items.append(i['url'])else:
  80.             ret = openai.Image.create_edit(image=image,mask=mask,prompt=prompt, n=int(num), size=size)for i in ret['data']:
  81.                 items.append(i['url'])for i in items:
  82.             md5_name = md5(i)
  83.             response = requests.get(i)# 打开图片
  84.             img = Image.open(BytesIO(response.content))
  85.             img.save(f"{save_cimg}{md5_name}.png", optimize=True, quality=50)# urllib.request.urlretrieve(i, f"{save_cimg}{md5_name}.png")
  86.         logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{key}>::<{keyn}>::<{items}>")return render_template('image.html',key=keyn, key2=key,key3=keyn,items2=items)except openai.error.OpenAIError as e:
  87.         stat =f"[{e.http_status}]{e.error}"
  88.         logger.debug(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<[{e.http_status}]{e.error}>")return stat
  90. @app.route('/img_sub3',methods=['POST'])defimg_sub3():
  91.     key ={"key":1}
  92.     keyn ={"key":0}
  93.     items =[]
  94.     file_src3 =""try:
  95.         image = request.form['main']if image=="":
  96.             image = file_src3
  97.         size = request.form['size']
  98.         num =int(request.form['num'])ifint(num)>10:for i inrange(0, num,10):
  99.                 sleep(31)
  100.                 ret = openai.Image.create_variation(image=image, n=int(num), size=size)for i in ret['data']:
  101.                     items.append(i['url'])else:
  102.             ret = openai.Image.create_variation(image=image, n=int(num), size=size)for i in ret['data']:
  103.                 items.append(i['url'])for i in items:
  104.             md5_name = md5(i)
  105.             response = requests.get(i)# 打开图片
  106.             img = Image.open(BytesIO(response.content))
  107.             img.save(f"{save_cimg}{md5_name}.png", optimize=True, quality=50)# urllib.request.urlretrieve(i, f"{save_cimg}{md5_name}.png")
  108.         logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{key}>::<{keyn}>::<{items}>")return render_template('image.html',key=keyn, key2=keyn, key3=key, items3=items)except openai.error.OpenAIError as e:
  109.         stat =f"[{e.http_status}]{e.error}"
  110.         logger.debug(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<[{e.http_status}]{e.error}>")return stat
  112. @app.route('/upload_edit1', methods=['POST'])defupload1():# 获取客户端上传的图片
  113.     image = request.files['image']# 获取图片的名称
  114.     image_name = image.filename
  115.     dirok= mkdir_date("img\\")if dirok !=0:# 保存图片到本地
  116.         file_src1 =f"img\\{dirok}\\{image_name}"
  117.         image.save(file_src1)
  118.         logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{file_src1}>")return"上传成功！"return"上传失败！"@app.route('/upload_edit2', methods=['POST'])defupload2():# 获取客户端上传的图片
  119.     image = request.files['image']# 获取图片的名称
  120.     image_name = image.filename
  121.     dirok= mkdir_date("img\\")if dirok !=0:# 保存图片到本地
  122.         file_src2 =f"img\\{dirok}\\{image_name}"
  123.         image.save(file_src2)
  124.         logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{file_src2}>")return"上传成功！"return"上传失败！"@app.route('/upload_edit3', methods=['POST'])defupload3():# 获取客户端上传的图片
  125.     image = request.files['image']# 获取图片的名称
  126.     image_name = image.filename
  127.     dirok= mkdir_date("img\\")if dirok !=0:# 保存图片到本地
  128.         file_src3 =f"img\\{dirok}\\{image_name}"
  129.         image.save(file_src3)
  130.         logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{file_src3}>")return"上传成功！"return"上传失败！"@app.route('/gx_upload', methods=['POST'])defgx_upload():
  131.     key =1
  132.     filenames = os.listdir("static\\text\\")# 打印文件名
  133.     items_gx =[]for filename in filenames:
  134.         items_gx.append(f"text/{filename}")try:# 获取客户端上传的图片
  135.         image = request.files['image']# 获取图片的名称
  136.         image_name = image.filename
  137.         filename = md5(str(image_name))
  138.         file_src_gx =f"static\\text\\{filename}.{image_name.split('.')[-1]}"
  139.         image.save(file_src_gx)
  140.         logger.info(f"[{request.remote_addr}]::[{request.method}]::{request.url}::{request.args}::<{key}::{file_src_gx}>")return render_template('text.html', key=key, items_gx=items_gx)except Exception as e:print(e)return render_template('text.html',key=0,items_gx=items_gx)if __name__ =='__main__':
  141.     app.run(host="0.0.0.0", port=5001)

0x09Key.py

  1. import os
  2. from datetime import datetime
  3. import hashlib
  4. import openai
  5. openai.organization ="org-xxxx"
  6. openai.api_key_path ="api_key.txt"defmkdir_date(dir):
  7.     nows = datetime.now()
  8.     date_str = nows.strftime('%Y-%m-%d')try:# 创建文件夹
  9.         os.mkdir(dir+date_str)return date_str
  10.     except Exception as e:return date_str
  12. defmd5(str1):
  13.     m = hashlib.md5()
  14.     m.update(str1.encode('utf-8'))returnstr(m.hexdigest())

0x0A结语

对于一个只会一点点代码水平的人来说，这个AI也是够了，解决能力大于某度，过于强大

源码整合包：传送门

Demo地址：传送门（请勿攻击）

标签： flask chatgpt python
本文转载自: https://blog.csdn.net/qq_19623861/article/details/128314049
版权归原作者 我重来不说话 所有， 如有侵权，请联系我们删除。
登录后发布评论

“ChatGPT写Flask-Demo——有体验地址”的评论:

还没有评论
关于作者
...
overfit同步小助手
文章同步
相关阅读

网络安全法-网络运行安全

使用selenium/drissionpage时如何阻止chrome自动跳转http到https

docker desktop 里部署的Open WebUI 管理员密码忘记了的处理方法

在ubuntu20.04中搭建onsite比赛运行环境

利用开源的低代码表单设计器FcDesigner高效管理和渲染复杂表单结构

Kafka学习笔记

【前端】浏览器输入url到页面呈现发生了什么？