Spaces:
Runtime error
Runtime error
import streamlit as st | |
import os | |
import openai | |
from PIL import Image | |
# openai.api_key = os.environ.get('openai.api_key') | |
# 页面设置 | |
st.set_page_config( | |
page_title="小译学长|一张原图生成多张不同图", | |
page_icon=":robot:" | |
) | |
st.header("🔥AI一张原图生成多张不同图") | |
#检查账号登陆 | |
def get_text1(): | |
if 'openai_key' not in st.session_state: | |
input_text1 = st.text_input("📫请输入你的账号: ", key="input") | |
if st.button("确认登陆!", key="input3"): | |
st.session_state['openai_key'] = input_text1 | |
return input_text1 | |
else: | |
return st.session_state['openai_key'] | |
openai_key = get_text1() | |
if openai_key: | |
openai.api_key = openai_key | |
st.write("") | |
else: | |
# st.write("⚒️账号获取方式:扫描下方二维码或搜索关注微信公众号【正经人王同学】回复【小译学长】获取你的账号,然后将公共账号输入到此处,再两次点击【确认登陆!】就好") | |
st.image('https://pic4.zhimg.com/v2-401dd67cf027f85f53e4be3bd28dab5f_b.jpg') | |
st.markdown("""⚒️账号获取方式:[账号购买](https://xiaoyisenior.top/586e5309-d0a6-4e4c-bdc0-ae0ed450f55e) [正经人王同学|公众号](https://mp.weixin.qq.com/s?__biz=Mzg3ODcwNzk3Nw==&mid=2247485615&idx=1&sn=c691d496386b5972e36fea8eaee33b97&chksm=cf0edcc9f87955df95fa23a716d78d496e7456183da7cfa6b348db6988865da9060896abcdcd&token=1164458978&lang=zh_CN#rd) [抖音](https://www.douyin.com/user/MS4wLjABAAAAIdY0VlMSK0Shyd4FxHBgkXAtH4Zq8wsuKzIuSICWpy0) [小红书 ](https://www.xiaohongshu.com/user/profile/5f12a46a000000000101ff27)""") | |
uploaded_image = st.file_uploader("🌈小译学长温馨提示:你可以上传一张喜欢的图像,并根据原始图像的灵感创造出不同的变化图像", type=["jpg", "png","jpeg"]) | |
if uploaded_image is not None: | |
# Convert image to png format | |
img = Image.open(uploaded_image) | |
#rize the photo | |
img_width, img_height = img.size | |
if img_width != img_height: | |
img_size = min(img_width, img_height) | |
img = img.crop((0, 0, img_size, img_size)) | |
img.save("origin.png") | |
st.image("origin.png", caption='Uploaded image', use_column_width=True) | |
def image_variation(): | |
try: | |
images = openai.Image.create_variation( | |
image=open("origin.png", "rb"), | |
n=4, | |
size="1024x1024" | |
) | |
st.empty() | |
for image in images["data"]: | |
st.image(image["url"],width=300) | |
except Exception as e: | |
st.write("❌❌❌你的账号填写有误,请刷新页面重新填写正确的账号!") | |
if st.button("开始绘画"): | |
image_variation() | |
st.write(""" | |
### 上传图片前必看!: | |
👀上传图像必须满足以下要求:\n | |
1、支持上传的图像格式有jpg、jpeg、png\n | |
2、大小小于4MB\n | |
可用tinypng压缩原图像大小,地址:https://tinypng.com/ | |
""") | |