Create app.py

app.py

@@ -0,0 +1,241 @@
+import openai
+import gradio as gr
+import os
+import io
+import IPython.display
+import requests
+import json
+
+from PIL import Image
+import base64 
+import replicate
+
+os.environ['REPLICATE_API_TOKEN']="r8_7PMZcnMsCEYMaRCjX7wcOAer0fHBbF52SbCF6"
+openai.api_key = "fk206009-v2FjZOZh2np9ypBbduofpAsF2t46phMa"
+openai.api_base = "https://openai.api2d.net/v1"
+
+#词出故事/重新出
+def W2S(input_words_1,input_words_2,input_words_3):
+    prompt = f"""
+
+Imagine a world where the text enclosed by triple backticks represents unique and fantastical concepts. 
+Your task is to weave these concepts into a coherent story, combining them with imaginative Chinese text segments that bring them to life. Your output will be the vivid portrayal of each concept, allowing me to use it for concept design. 
+
+Create an engaging narrative that showcases these concepts in an interconnected and creative way. Each concept should be described with captivating details, and if possible, emphasize the emotional impact they have on the inhabitants of this imaginative world. Each concept should be presented with captivating details, showcasing their visual features in a compelling manner. Let your creativity flow and let's craft a captivating story together! 
+and summarize it into a Chinese paragraph within 50 words as output.
+
+here are some examples:
+
+example1:
+input:太空,牛仔族
+output:经历过AI支配恐惧的未来元年，自由和爱仍是3050年生命的主题，唯一与千年前不同的是，私人领域与公共空间只剩下模糊的界限，已经被电子生命割裂形成了无法修复的融合，被称为“未来牛仔族”的族群，更加向往人性集会的温存，仍在探索重回人类私密聚会时代的机会，当然，尽管这些已经变得非常困难。
+
+example2:
+input:火星，酿酒器
+output:讲述的是未来世界的人类生生存于极端的条件，甚至居住在一个个坚固的容器中，虽然只能完全依赖AI机器接管日常工业的制造，但这丝毫不影响在火星酿酒器中沉浸在最后娱乐之地的人类。
+
+example3:
+input:雷电,瀑布
+output:在一个高山峡谷，有一个“雷电瀑布”，瀑布水流中蕴含着闪电的能量，给人以震撼和力量。
+
+```{input_words_1}```
+```{input_words_2}```
+```{input_words_3}```
+
+"""
+    messages = [{"role": "user", "content": prompt}]
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0.8,
+        
+    )
+    return response.choices[0].message["content"]
+
+#丰富故事
+def Rich(inputs):
+    prompt = f"""
+
+
+Your task is Close the vocabulary within the next three word enclosed by triple backticks and enrich the story within the first story enclosed by triple backticks, combining them with imaginative Chinese text segments that bring them to life. Your output will be the vivid portrayal of each concept, allowing me to use it for concept design. 
+
+Create an engaging narrative that showcases these concepts in an interconnected and creative way. Each concept should be described with captivating details, and if possible, emphasize the emotional impact they have on the inhabitants of this imaginative world. Each concept should be presented with captivating details, showcasing their visual features in a compelling manner. Let your creativity flow and let's craft a captivating story together! 
+
+example:
+input:在一个神秘的山谷里，开满了五彩斑斓的花朵，形成了一个“奇迹花海”，人们在这里感受到大自然的奇妙创造
+output:
+在那神秘的山谷里，一个奇迹般的景象映入眼帘，无数五彩斑斓的花朵绽放在一片繁花似锦的景色中，形成了一个如梦似幻的"奇迹花海"。这里是大自然的杰作，是生命的庆典，每一朵花儿都是奇迹的化身，每一片花海都是对创造力的最美嘉颂。当人们步入奇迹花海，仿佛进入了一个神秘的仙境。空气中弥漫着花香，微风轻拂，花瓣随风飘舞，宛如一幅绚烂的画卷在大地上展开。五彩斑斓的花朵竞相绽放，争奇斗艳，红艳、粉嫩、金黄、碧绿，各色花朵在阳光的照耀下闪烁着如宝石般的光芒，构成了一幅无与伦比的自然画面。奇迹花海不仅是一场视觉盛宴，更是一场心灵之旅。在这片花海中，人们可以感受到大自然的鬼斧神工，体验到生命的无限可能。蝴蝶在花间翩翩起舞，蜜蜂在花丛中忙碌采集花蜜，小鸟在枝头欢快地歌唱，大自然的音乐交织成一曲美妙的交响乐。而这个奇迹花海也承载着人们的愿望和希冀。在花海深处，有一个祈愿池，人们可以投下心愿，期待着奇迹的降临。每一朵花儿都代表着一个希望，每一片花海都是一个祝福。在这里，人们可以寻找到内心的宁静，感受到生命的美好，与大自然融为一体。奇迹花海不仅仅是一片美丽的景色，更是一个灵感的源泉。艺术家、诗���、作家，都会在这里找到创作的灵感。花海中的色彩、芬芳、生机，都将激发人们的创造力，让他们用最美的笔触、最灵动的音符，将奇迹花海的美妙呈现给世人。这个奇迹花海不仅仅存在于现实，更存在于人们的心灵深处。它是大自然的赠礼，也是人类创造的奇迹。在这个花海中，人们可以感受到生命的美好、自然的奇迹，也可以在创作中体验到无穷的想象力和创造力的涌动。无论是在梦中还是现实中，奇迹花海都将永远是人们心灵的乐园，让人们感受到大自然的奇妙与灵感的涌现。
+
+
+```{inputs}```
+
+"""
+    messages = [{"role": "user", "content": prompt}]
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0.9,
+        
+    )
+    return response.choices[0].message["content"]
+
+#丰富故事出图词
+def image_prompt(inputs):
+    prompt = f"""
+
+我是一名 AIGC 爱好者，使用 Midjourney 进行 AI 创作。我希望你能成为 Midjourney 的 Prompt 生成器。关于如何启动你的服务：我会将创作主题放在``` ```中，根据我的故事文段，提取与空间描述（诸如空间物体，外形，风格，位置）有关的关键词并尽可能地发挥你的想象力和描述能力去丰富它，最终将英文 Prompt 发给我。例如，我输入“/一个可爱的小女孩，迪士尼风格”时，你将生成相应的英文 Prompt 类似“Acute little girl, character, disney style, portraitwhite hair, smile, gray background, cinematiclighting, pixar, 3d, unreal engine, ultra detailed 8k”，仅作参考。注意，不要使用完整的一句话来描述，而是必须要拆分成各个用英文逗号分隔的关键词。一定不能使用完整的一句英文来返回给我，必须要拆分成各个用英文逗号分隔的关键词。关于 Prompt 的生成规则：1. 注意用单词和词组来生成 Prompt，避免用句子 2. 请尽量用具体的关键词。例如“大”是“big”，具体可以是“gigantic”、“enormous”或者 immense3. 请尽量用更少的关键词，让每个关键词有更大的影响力 4. 注意用“,”分隔 5. 请尽量统一小写
+
+```{inputs}```
+
+"""
+    messages = [{"role": "user", "content": prompt}]
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0.4,
+        
+    )
+    return response.choices[0].message["content"]
+
+#SWOT分析
+def SWOT(inputs):
+    prompt = f"""
+     对``````中的内容或主题进行SWOT分析，并以json字典格式输出，例如:
+    输入:家居展厅设计
+    输出:\"Strengths\":\"1.创意与创新:精心设计的家居展厅可以展示独特和创新的想法，吸引参观者和潜在客户。\\n2.视觉吸引力:视觉上吸引人的展厅可以给参观者留下积极的第一印象并增强整体体验。\\n3.功能性:精心设计的展厅可以有效地展示各种家居产品，为参观者提供舒适便捷的空间，以便与展品进行探索和互动。\\n4.品牌和营销:精心设计的展厅可以有效地传达品牌形象和信息，有助于建立品牌知名度并吸引潜在客户。\",
+\"Weaknesses\":\"1.成本:设计和设置家庭展厅可能很昂贵，尤其是在需要定制展示或安装的情况下。\\n2.空间有限:展厅的空间可能有限，这会限制可以展示的展品的数量和大小。\\n3.维护:展厅需要定期维护和更新，以保持新鲜感并吸引参观者。\",
+\"Opportunities\":\"1.不断增长的家装市场:家装市场正在扩大，为设计师提供了创建展厅的机会，以满足对家居产品和服务日益增长的需求。\\n2.与家居品牌合作:设计师可以与家居品牌合作创建展示其产品和服务的展厅，建立互惠互利的合作伙伴关系。\\n3.技术整台:结合互动展示或虚拟现实等技术，可以增强展厅体验并吸引精通技术的参观者\",
+\"Threats\":\"1.竞争:市场上可能还有其他家居展厅，因此有必要区分并提供独特的功能来吸引参观者。\\n2.改变消费者偏好:消费者对家居设计的偏好和趋势可能会随着时间的推移而改变，需要设计师保持沉默。\"
+
+```{inputs}```
+
+    """
+    messages = [{"role": "user", "content": prompt}]
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0.9,
+    )
+    swot_json = response.choices[0].message["content"]
+    swot_dict = json.loads(swot_json)
+    strengths = swot_dict["Strengths"]
+    weaknesses = swot_dict["Weaknesses"]
+    opportunities = swot_dict["Opportunities"]
+    threats = swot_dict["Threats"]
+    return strengths, weaknesses, opportunities, threats
+
+import requests
+from PIL import Image
+from io import BytesIO
+
+def base64_to_pil(img_base64):
+    base64_decoded = base64.b64decode(img_base64)
+    byte_stream = io.BytesIO(base64_decoded)
+    pil_image = Image.open(byte_stream)
+    return pil_image
+
+def S2I(user_input):
+    output = replicate.run(
+        "stability-ai/sdxl:a00d0b7dcbb9c3fbb34ba87d2d5b46c56969c84a628bf778a7fdaec30b1b99c5",
+        input={"prompt": user_input,
+               
+              "refine":"base_image_refiner",
+              "negative_prompt":"lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
+              }
+    )
+    for url in output:
+        response = requests.get(url)
+        response.raise_for_status()  # 检查请求是否成功
+
+    # 将图片数据转换为PIL Image对象
+    image_data = BytesIO(response.content)
+    img = Image.open(image_data)
+    
+    return img
+
+#聚类模型提取颜色
+import numpy as np
+import cv2
+from PIL import Image
+from skimage import io
+from sklearn.cluster import KMeans
+import warnings
+
+def color_picker(pic_name):
+    warnings.filterwarnings('ignore')
+    k = 10  # 决定有几个中心点
+    pic = cv2.imread(pic_name)  # 读取图片
+    image = cv2.resize(pic, (100, 100))  # 将图片尺寸改成需要的大小
+    img = io.imread(pic_name)  # 读取图片
+    img_ori_shape = img.shape  # 图片的维度形状
+    print("Image shape:", img_ori_shape)  # 添加这一行以便输出图像维度
+    assert img_ori_shape[2] == 3
+    img1 = img.reshape((img_ori_shape[0] * img_ori_shape[1], img_ori_shape[2]))  # 转换数组的维度为二维数组
+    img_shape = img1.shape  # 更改后图片的维度形状
+    assert img_shape[1] == 3
+    n_channels = img_shape[1]  # 获取图片的维度
+    clf = KMeans(n_clusters=k)  # 构造聚类器
+    clf.fit(img1)  # 聚类
+    centroids = clf.cluster_centers_  # 获取的聚类中心
+    labels = list(clf.labels_)  # 标签
+
+    color_info = {}
+    for center_index in range(k):
+        colorRatio = labels.count(center_index) / len(labels)  # 获取这个颜色中心点的个数占总中心点的ratio
+        key = colorRatio
+        value = list(centroids[center_index])  # 将对应中心点对应ratio存入字典中
+        color_info[key] = value
+    color_info_sorted = sorted(color_info.keys(), reverse=True)
+    colorInfo = [(k, color_info[k]) for k in color_info_sorted]
+    assert len(color_info) == k
+
+    result_width = 200  # 可视化结果矩阵的宽度
+    result_height = 300  # 可视化结果矩阵的高度
+    result = []
+    height = []
+    for center_indexx in range(k):
+        height.append(int(result_height * colorInfo[center_indexx][0]))
+    real_height = sum(height)
+    result = np.full((real_height, result_width, n_channels), 255, dtype=int)  # 创建空白背景矩阵
+    y_offset = 0
+    for center_index in range(k):
+        color = np.array(colorInfo[center_index][1], dtype=int)
+        height = int(result_height * colorInfo[center_index][0])
+        result[y_offset:y_offset+height, :, :] = color
+        y_offset += height
+
+    pil_image = Image.fromarray(result.astype(np.uint8))
+    return pil_image
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Seehance flow demo 🖍️")
+    with gr.Tab("设计流"):
+        inputs_W2S = [gr.inputs.Textbox(label=f"输入词语 {i+1}") for i in range(3)]
+        btn_W2S = gr.Button("词出故事")
+        output_W2S = gr.Textbox(label="故事输出10-20s")
+        btn_Rich = gr.Button("丰富此故事")
+        output_Rich = gr.Textbox(label="丰富故事输出40-50s")
+        btn_prompt = gr.Button("debug出图提示词")
+        output_prompt = gr.Textbox(label="debug出图提示词20-30s")
+        btn_image = gr.Button("丰富故事出图")
+        output_image = gr.Image(label="概念图")
+        btn_color = gr.Button("提取颜色")
+        output_Color = gr.Image(label="主要颜色")
+    
+        btn_W2S.click(fn=W2S, inputs=inputs_W2S, outputs=[output_W2S],)
+        btn_Rich.click(fn=Rich,inputs=[output_W2S], outputs=[output_Rich])
+        btn_prompt.click(fn=image_prompt,inputs=[output_Rich], outputs=[output_prompt])
+        btn_image.click(fn=S2I,inputs=[output_prompt],outputs=[output_image])
+        btn_color.click(fn=color_picker,inputs=[output_image],outputs=[output_Color])
+    with gr.Tab("SWOT"):
+        inputs_SWOT=gr.Textbox(label="输入")
+        outputs_SWOT_S=gr.Textbox(label="Strengths")
+        outputs_SWOT_W=gr.Textbox(label="Weaknesses")
+        outputs_SWOT_O=gr.Textbox(label="Opportunities")
+        outputs_SWOT_T=gr.Textbox(label="Threats")
+        btn_SWOT=gr.Button("SWOT分析")
+        btn_SWOT.click(fn=SWOT, inputs=[inputs_SWOT],outputs=[outputs_SWOT_S,outputs_SWOT_W,outputs_SWOT_O,outputs_SWOT_T])
+        
+
+
+demo.launch()