Duplicate from CrazyEric/cool-aigc

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Cool Aigc
+emoji: 🏆
+colorFrom: indigo
+colorTo: purple
+sdk: gradio
+sdk_version: 3.40.1
+app_file: app.py
+pinned: false
+license: openrail
+duplicated_from: CrazyEric/cool-aigc
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,47 @@
+import os
+import oss2
+import requests
+import gradio as gr
+import clipdrop
+
+def translateEN(zh):
+    result = requests.post(
+        "https://api-free.deepl.com/v2/translate", 
+        params={ 
+            "auth_key": "e8b4d428-ada5-3f8d-f965-bad01e8a06c1:fx", 
+            "target_lang": "EN-US", 
+            "text": zh})
+    return result.json()["translations"][0]["text"]
+
+def process_text(prompt):
+    print("输入的文本: \n", prompt)
+    prompt_trans = translateEN(prompt)
+    print("prompt: \n", prompt_trans)
+
+    image_path = clipdrop.text_to_image(prompt_trans)
+
+    # 使用阿里云OSS SDK上传图像到OSS存储桶
+    # 配置您的 OSS Bucket 信息
+    bucket_name = 'codejoyai-static'
+    endpoint = 'http://oss-cn-beijing.aliyuncs.com'
+
+    # 创建 OSS 客户端实例
+    auth = oss2.Auth('LTAI5tKa6jkMstCMDhrs2uMe', '402dwLJ3rsXOGcrYbMwNc2tyQnqeMe')
+    bucket = oss2.Bucket(auth, endpoint, bucket_name)
+
+    # 指定目录路径和文件名
+    directory = 'aigc-202308/'
+    filename = os.path.basename(image_path)
+
+    # 上传文件到指定目录
+    bucket.put_object_from_file(directory + filename, image_path)
+
+    img_url = "https://codejoyai-static.oss-cn-beijing.aliyuncs.com" + "/" + directory + filename
+    
+    print(img_url)
+    return img_url
+
+iface = gr.Interface(fn=process_text, inputs="text", outputs="image")
+
+# iface.queue(concurrency_count=1)
+iface.launch(enable_queue=True)

clipdrop.py

@@ -0,0 +1,49 @@
+import os
+import base64
+import requests
+
+def text_to_image(prompt):
+    url = "https://api.stability.ai/v1/generation/stable-diffusion-xl-1024-v1-0/text-to-image"
+
+    body = {
+      "steps": 30,
+      "width": 1024,
+      "height": 1024,
+      "seed": 0,
+      "cfg_scale": 7,
+      "samples": 1,
+      "style_preset": "enhance",
+      "text_prompts": [
+        {
+          "text": prompt,
+          "weight": 1
+        }
+      ],
+    }
+
+    headers = {
+      "Accept": "application/json",
+      "Content-Type": "application/json",
+      "Authorization": "Bearer sk-8n2tgRL6CzV91nQ2nvecVMhedg4GbEexDtUoDnCOWt09Wq2W",
+    }
+
+    response = requests.post(
+        url,
+        headers=headers,
+        json=body,
+        )
+
+    if response.status_code != 200:
+        raise Exception("Non-200 response: " + str(response.text))
+
+    data = response.json()
+
+    for i, image in enumerate(data["artifacts"]):
+        if not os.path.exists("output"):
+            os.makedirs("output")
+            print("成功创建文件夹 'output'")
+        image_path = f'./output/txt2img_{image["seed"]}.png'
+        with open(image_path, "wb") as f:
+            f.write(base64.b64decode(image["base64"]))
+
+    return image_path

requirements.txt

@@ -0,0 +1,2 @@
+oss2
+gradio