update

app.py

@@ -1,7 +1,225 @@
+from __future__ import annotations
 import gradio as gr
+import logging
+logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.INFO,
+format='%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s')
+import subprocess
+def runcmd(command):
+    ret = subprocess.run(command,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE,encoding="utf-8",timeout=60)
+    if ret.returncode == 0:
+        print("success:",ret)
+    else:
+        print("error:",ret)
+runcmd("pip3 install --upgrade clueai")
 
-def greet(name):
-    return "Hello " + name + "!!"
+import clueai
+cl = clueai.Client("", check_api_key=False)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+'''
+#luck_t2i_btn_1, #luck_s2i_btn_1, #luck_i2i_btn_1, #luck_ici_btn_1{
+            color: #fff;
+            --tw-gradient-from: #BED336;
+            --tw-gradient-stops: var(--tw-gradient-from), var(--tw-gradient-to);
+            --tw-gradient-to: #BED336;
+            border-color: #BED336;
+        }
+
+        #luck_easy_btn_1, #luck_iti_btn_1, #luck_tsi_btn_1, #luck_isi_btn_1{
+            color: #fff;
+            --tw-gradient-from: #BED336;
+            --tw-gradient-stops: var(--tw-gradient-from), var(--tw-gradient-to);
+            --tw-gradient-to: #BED336;
+            border-color: #BED336;
+        }
+'''
+css='''
+        .container { max-width: 800px; margin: auto; }
+        #gen_btn_1{
+            color: #fff;
+            --tw-gradient-from: #f44336;
+            --tw-gradient-stops: var(--tw-gradient-from), var(--tw-gradient-to);
+            --tw-gradient-to: #ff9800;
+            border-color: #ff9800; 
+        }
+        #t2i_btn_1, #s2i_btn_1, #i2i_btn_1, #ici_btn_1, #easy_btn_1, #iti_btn_1, #tsi_btn_1, #isi_btn_1{
+            color: #fff;
+            --tw-gradient-from: #f44336;
+            --tw-gradient-stops: var(--tw-gradient-from), var(--tw-gradient-to);
+            --tw-gradient-to: #ff9800;
+            border-color: #ff9800;
+        }
+        
+
+        #import_t2i_btn_1, #import_s2i_btn_1, #import_i2i_btn_1, #import_ici_btn_1{
+            color: #fff;
+            --tw-gradient-from: #BED336;
+            --tw-gradient-stops: var(--tw-gradient-from), var(--tw-gradient-to);
+            --tw-gradient-to: #BED336;
+            border-color: #BED336;
+        }
+
+        #import_easy_btn_1, #import_iti_btn_1, #import_tsi_btn_1, #import_isi_btn_1{
+            color: #fff;
+            --tw-gradient-from: #BED336;
+            --tw-gradient-stops: var(--tw-gradient-from), var(--tw-gradient-to);
+            --tw-gradient-to: #BED336;
+            border-color: #BED336;
+        }
+
+        #record_btn{
+            
+        }
+        #record_btn > div > button > span {
+            width: 2.375rem;
+            height: 2.375rem;
+        }
+        #record_btn > div > button > span > span {
+            width: 2.375rem;
+            height: 2.375rem;
+        }
+        audio {
+            margin-bottom: 10px;
+        }
+        div#record_btn > .mt-6{
+            margin-top: 0!important;
+        }
+        div#record_btn > .mt-6 button {
+            font-size: 1em;
+            width: 100%;
+            padding: 20px;
+            height: 60px;
+        }
+
+        div#txt2img_tab {
+            color: #BED336;
+        }
+
+'''
+task_styles = []
+examples_list = []
+task_style_to_task_prefix = {}
+import csv
+def read_examples(input_file):
+    header = True
+    with open(input_file) as finput:
+        csv_input = csv.reader(finput)
+        for line in csv_input:
+            if header:
+                header = False
+                continue
+            task_style, task_prefix, example = line
+            task_styles.append(task_style)
+            task_style_to_task_prefix[task_style] = task_prefix
+            examples_list.append([task_style, example])
+
+read_examples("./examples.csv")
+#print(task_styles)
+def preprocess(text, task):
+    if task == "问答":
+        text = text.replace("?", ":").replace("？", ":")
+        text = text + "："
+
+    return task_style_to_task_prefix[task] + "\n" + text + "\n答案："
+      
+def inference_gen(text, task):
+    text = preprocess(text, task)
+    #print(text)
+    try:
+        prediction = cl.generate(
+            model_name='clueai-base',
+            prompt=text)
+    except Exception as e:
+        logger.error(f"error, e")
+        return
+
+    return prediction.generations[0].text
+     
+t2i_default_img_path_list = []
+import base64, requests
+from io import BytesIO
+from PIL import Image
+def inference_image(text, n_text, guidance_scale, style, shape, clarity, steps, shape_scale):
+    try:
+        res = requests.get(f"https://www.clueai.cn/clueai/hf_text2image?text={text}")
+    except Exception as e:
+        logger.error(f"error, {e}")
+        return 
+    json_dict = res.json()
+    file_path_list = []
+    for i, image in enumerate(json_dict["images"]):
+        image = image.encode('utf-8')
+        binary_data = base64.b64decode(image)
+        img_data = BytesIO(binary_data)
+        img = Image.open(img_data)
+        file_path_list.append(img)
+
+    return file_path_list
+image_styles = ['无', '细节大师', '对称美', '虚拟引擎', '空间感', '机械风格', '形状艺术', '治愈', '电影构图', '电影构图（治愈）', '荒芜感', '漫画', '逃离艺术', '斯皮尔伯格', '幻想', '杰作', '壁画', '朦胧', '黑白（3d)', '梵高', '毕加索', '莫奈', '丰子恺', '现代', '欧美']
+with gr.Blocks(css=css, title="ClueAI") as demo:
+    gr.Markdown('<h1><center><font color=red style="font-size:50px;">ClueAI全能师</font></center></h1>')
+    with gr.TabItem("文本生成", id='_tab'):   
+        with gr.Row(variant="compact").style( equal_height=True):
+            text = gr.Textbox("俄罗斯天然气管道泄漏爆炸",
+                label="编辑内容", show_label=False, max_lines=20, 
+                placeholder="在这里输入...",
+            )
+        task = gr.Dropdown(label="任务", show_label=True, choices=task_styles, value="标题生成文章")
+        btn = gr.Button("生成",elem_id="gen_btn_1").style(full_width=False)
+        with gr.Accordion("高级操作", open=False):
+            n_text = gr.Textbox("",
+                label="不想要生成的元素", show_label=True, max_lines=2, 
+                placeholder="在这里输入你不需要包含的内容...",
+            )     
+            guidance_scale = gr.Slider(1, 20, value=7.5, step=0.5, label="和你的描述匹配程度，越大越匹配")
+            shape = gr.Radio(["1x1", "16x9", "手机壁纸"], label="尺寸", value="1x1")
+            shape_scale = gr.Radio([1, 2, 3], label="对图放大倍数", value=1)
+            steps = gr.Slider(10, 150, value=50, step=1, label="越大质量越好，生成时间越长")
+            clarity = gr.Radio(["标清", "高清"], label="清晰度", value="标清")
+        with gr.Row(variant="compact").style( equal_height=True):
+            output_text = gr.Textbox(
+                    label="输出", show_label=True, max_lines=50, 
+                    placeholder="在这里展示结果",
+                )
+        gr.Examples(examples_list, [task, text], label="示例")
+        input_params = [text, task]
+        #text.submit(inference_gen, inputs=input_params, outputs=output_text)
+        btn.click(inference_gen, inputs=input_params, outputs=output_text)
+
+    with gr.TabItem("图像生成", id='txt2img_tab'):   
+        with gr.Row(variant="compact").style( equal_height=True):
+            text = gr.Textbox("美丽的风景",
+                label="编辑内容", show_label=False, max_lines=2, 
+                placeholder="在这里输入你的描述...",
+            )
+            btn = gr.Button("生成图像",elem_id="t2i_btn_1").style(full_width=False)
+            
+        with gr.Row().style( equal_height=True):
+            generate_prompt_btn = gr.Button("手气不错", elem_id="luck_t2i_btn_1")
+
+        style = gr.Dropdown(label="风格", show_label=True, choices=image_styles, value="无")
+        with gr.Accordion("高级操作", open=False):
+            n_text = gr.Textbox("",
+                label="不想要生成的元素", show_label=True, max_lines=2, 
+                placeholder="在这里输入你不需要包含的内容...",
+            )     
+            guidance_scale = gr.Slider(1, 20, value=7.5, step=0.5, label="和你的描述匹配程度，越大越匹配")
+            shape = gr.Radio(["1x1", "16x9", "手机壁纸"], label="尺寸", value="1x1")
+            shape_scale = gr.Radio([1, 2, 3], label="对图放大倍数", value=1)
+            steps = gr.Slider(10, 150, value=50, step=1, label="越大质量越好，生成时间越长")
+            clarity = gr.Radio(["标清", "高清"], label="清晰度", value="标清")
+
+        gr.Examples(["秋日的晚霞", "星空", "室内装修", "婚礼鲜花"], text, label="示例")
+        
+        t2i_gallery = gr.Gallery(
+            t2i_default_img_path_list,
+            label="生成图像",
+             show_label=False).style(
+            grid=[2], height="auto"
+        )
+
+        input_params = [text, n_text, guidance_scale, style, shape, clarity, steps, shape_scale]
+        generate_prompt_btn.click(inference_image, inputs=input_params, outputs=[t2i_gallery])
+        text.submit(inference_image, inputs=input_params, outputs=t2i_gallery)
+        btn.click(inference_image, inputs=input_params, outputs=t2i_gallery)
+demo.launch()

examples.csv

@@ -0,0 +1,48 @@
+任务类型,任务前缀,文本内容
+问答,问答：,问题：阿里巴巴的总部在哪里
+纠错,文本纠错：,你必须服从命令，不要的考虑。你的思想被别人手下来。
+改写,生成与下列文字相同意思的句子：,长期通胀前景在今天可能不确定性更大。
+中心词提取,中心词提取：,现在有京东国际太方便了，可以安全的买到正经的电子设备，游戏机完全没问题，发货非常快，第二天就到
+标题生成文章,根据标题生成文章：,标题：俄罗斯天然气管道泄漏爆炸
+情感倾向,文字中包含了怎样的情感：,"超可爱的帅哥，爱了。。。
+选项：厌恶，喜欢，开心，悲伤，惊讶，生气，害怕"
+关键词抽取, 抽取关键词：,当地时间21日，美国联邦储备委员会宣布加息75个基点，将联邦基金利率目标区间上调到3.00%至3.25%之间，符合市场预期。这是美联储今年以来第五次加息，也是连续第三次加息，创自1981年以来的最大密集加息幅度。
+指代消解,代词指向哪个名词短语：,"段落：
+我回来，咱们两个人劳动，一定要把少平和兰香的书供成。只要他两个有本事，能考到哪里，咱们就把他们（代词）供到哪里。
+问题：代词“ 他们”指代的是？"
+问题生成,问题生成：,"本周早些时候，始于俄罗斯的两条巨型天然气管道分别发现了三个泄漏点。管道中充满燃料，破裂产生了宽度达800米的气泡，上浮到丹麦博恩霍尔姆岛附近的波罗的海表面。
+就在泄漏发生前，附近检测到了爆炸，而欧洲各国政府尚未确定被称为北溪一号和北溪二号管道泄漏的原因。欧洲和美国的领导人表示，这是一起蓄意破坏行为。
+相关猜测指向俄罗斯，该国的国有能源公司俄罗斯天然气工业股份公司是这些管道的主要所有者。俄罗斯发言人德米特里·佩斯科夫驳斥了有关俄罗斯参与的指控，称其“愚蠢”，并将矛头指向美国。"
+医疗语义相似度,下面句子是否表示了相同的语义：,"文本1：糖尿病腿麻木怎么办？
+文本2：糖尿病怎样控制生活方式
+选项：相似，不相似"
+电商客户需求分析,电商客户诉求分类：,"收到但不太合身，可以退换吗
+选项：买家咨询商品是否支持花呗付款，买家表示收藏关注店铺，买家咨询退换货规则，买家需要商品推荐"
+医疗信息抽取,找出指定的信息：,"我今天胃不舒服，手有点麻木，浑身不舒服，前几天去医院医生说我有胃炎
+问题：症状，治疗，检查，身体部位，疾病"
+简历信息抽取,阅读文本抽取关键信息：,"张玄武1990年出生中国国籍无境外居留权博士学历现任杭州线锁科技技术总监。
+问题：机构，人名，职位，籍贯，专业，国籍，学历，种族"
+通用信息抽取,信息抽取：,"据新华社电广东省清远市清城区政府昨日对外发布信息称,日前被实名举报涉嫌勒索企业、说“分分钟可以搞垮一间厂”的清城区环保局局长陈柏,已被免去清城区区委委员
+问题：机构名，人名，职位"
+翻译-英中,翻译成中文：,This is a dialogue robot that can talk to people.
+翻译-中英,翻译成英文：,前国务卿去了台湾，中国人民很愤怒。
+摘要,摘要生成：,"本周早些时候，始于俄罗斯的两条巨型天然气管道分别发现了三个泄漏点。管道中充满燃料，破裂产生了宽度达800米的气泡，上浮到丹麦博恩霍尔姆岛附近的波罗的海表面。
+就在泄漏发生前，附近检测到了爆炸，而欧洲各国政府尚未确定被称为北溪一号和北溪二号管道泄漏的原因。欧洲和美国的政治领导人表示，这是一起蓄意破坏行为。
+相关猜测指向俄罗斯，该国的国有能源公司俄罗斯天然气工业股份公司是这些管道的主要所有者。俄罗斯发言人德米特里·佩斯科夫驳斥了有关俄罗斯参与的指控，称其“愚蠢”，并将矛头指向美国。"
+阅读理解-自由式,阅读以下对话并回答问题。,"男：今天怎么这么晚才来上班啊？女：昨天工作到很晚，而且我还感冒了。男：那你回去休息吧，我帮你请假。女：谢谢你。
+问题：女的怎么样？
+选项：正在工作，感冒了，在打电话，要出差。"
+阅读理解,阅读理解：,"段落：本周早些时候，始于俄罗斯的两条巨型天然气管道分别发现了三个泄漏点。管道中充满燃料，破裂产生了宽度达800米的气泡，上浮到丹麦博恩霍尔姆岛附近的波罗的海表面。
+就在泄漏发生前，附近检测到了爆炸，而欧洲各国政府尚未确定被称为北溪一号和北溪二号管道泄漏的原因。欧洲和美国的政治领导人表示，这是一起蓄意破坏行为。
+相关猜测指向俄罗斯，该国的国有能源公司俄罗斯天然气工业股份公司是这些管道的主要所有者���俄罗斯发言人德米特里·佩斯科夫驳斥了有关俄罗斯参与的指控，称其“愚蠢”，并将矛头指向美国。
+问题：北溪管道是什么原因泄露的？"
+推理,推理关系判断：,"前提：小明今天在北京
+假设：小明在深圳旅游
+选项：矛盾，蕴含，中立"
+情感分析,情感分析：,"这个看上去还可以，但其实我不喜欢
+选项：积极，消极"
+新闻分类,新闻分类：,"本周早些时候，始于俄罗斯的两条巨型天然气管道分别发现了三个泄漏点。管道中充满燃料，破裂产生了宽度达800米的气泡，上浮到丹麦博恩霍尔姆岛附近的波罗的海表面。
+就在泄漏发生前，附近检测到了爆炸，而欧洲各国政府尚未确定被称为北溪一号和北溪二号管道泄漏的原因。欧洲和美国的政治领导人表示，这是一起蓄意破坏行为。
+相关猜测指向俄罗斯，该国的国有能源公司俄罗斯天然气工业股份公司是这些管道的主要所有者。俄罗斯发言人德米特里·佩斯科夫驳斥了有关俄罗斯参与的指控，称其“愚蠢”，并将矛头指向美国。"
+意图分类,意图分类：,"帮我定一个周日上海浦东的房间
+选项：闹钟，文学，酒店，艺术，体育，健康，天气，其他"