Update app.py

app.py

@@ -1,9 +1,43 @@
 import gradio as gr
-import cv2
+from dalle2 import Dalle2
+from PIL import Image
+import requests
 
-def to_black(image):
-    output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    return output
 
-interface = gr.Interface(fn=to_black, inputs="image", outputs="image")
-interface.launch()
+def draw(description):
+    print("description"+description)
+    dalle = Dalle2("sess-mDEO1h4fEgZQ4Aksl1qJwNJ1t0e2qEeyrMP4PKnZ")
+    generations = dalle.generate(description)
+    img_list = []
+    for generation in generations:
+        image_url = generation["generation"]["image_path"]
+        # response = requests.get(image_url)
+        # img = Image.open(BytesIO(response.content))
+        im = Image.open(requests.get(image_url, stream=True).raw).convert('RGB')
+        img_list.append(im)
+    return img_list
+
+with gr.Blocks() as demo:
+    with gr.Column(variant="panel"):
+        gr.Markdown("#  <center> 🚀你说我画")
+        with gr.Row(variant="compact"):
+            text = gr.Textbox(
+                label="Enter your prompt",
+                show_label=False,
+                max_lines=1,
+                placeholder="请描述你想要的画",
+            ).style(
+                container=False,
+            )
+            btn = gr.Button("开始生成").style(full_width=False)
+
+        gallery = gr.Gallery(
+            label="结果", show_label=False, elem_id="gallery"
+        ).style(grid=[2], height="auto")
+
+        gr.Markdown("### 叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪叶伟豪")
+
+    btn.click(draw, text, gallery)
+
+if __name__ == "__main__":
+    demo.launch()