Merge branch 'wds/gradio_tmp' of rongrunxiang/why10w into master

main.py

@@ -0,0 +1,69 @@
+import numpy as np
+import gradio as gr
+
+# 生成一个文生图的gradio界面
+
+
+def mock_img(text: str):
+    return [
+        {
+            "caption": "科学家们使用克隆技术",
+            "url": "https://image.limaogushi.com/file/picture-books/image/f2/d8/7ae32095-6e8c-4328-a218-6fe57e49bbc2.jpg",
+        },
+        {
+            "caption": "克隆羊多利的诞生",
+            "url": "https://image.limaogushi.com/file/picture-books/image/97/98/75cf0a7b-cb27-44b3-beae-7ab24488ff8c.jpg",
+        },
+    ]
+
+
+def mock_story(query: str):
+    return """小朋友，你听说过“克隆”这个词吗？让我来给你讲一个有趣的故事吧！
+
+有一天，科学家们想出了一个非常神奇的方法，可以让一个生命体“复制”出另一个一模一样的生命体，这个方法就叫做“克隆”。你可以把“克隆”想象成是一台神奇的复印机，但是这台复印机不是复印纸张，而是复印生命哦！
+
+1997年的时候，科学家们用这种方法成功地“复印”出了一只名叫多利的绵羊。你知道吗，多利羊并没有爸爸，它是由一只母羊的细胞复制而来的。这就像是我们从电脑上复制一个文件一样，只不过这次我们复制的是一个活生生的羊！
+
+克隆羊多利的诞生，就像是打开了一扇新的大门，让人们看到了生命的奇妙和科学的无限可能。从此以后，人们开始更加关注和研究克隆技术，希望用它来帮助我们解决更多的问题。
+
+所以，克隆羊就是通过克隆技术创造出来的一只羊，它没有父亲，只有母亲，是科学家们用神奇的“生命复印机”复制出来的哦！"""
+
+
+def generate_image(text):
+    """ """
+    res = mock_img(text)
+    md = ""
+    # 改为markdown
+    for i in range(len(res)):
+        md += f"{res[i]['caption']}\n![{res[i]['caption']}]({res[i]['url']})\n\n"
+    return md
+
+
+def generate_story(query):
+    """ """
+    return mock_story(query)
+
+
+def generate_story_and_image(query):
+    """ """
+    story = generate_story(query)
+    image = generate_image(query)
+    return story, image
+
+
+# 生成一个gradio界面
+demo = gr.Interface(
+    fn=generate_story_and_image,
+    inputs=gr.Textbox(lines=5, label="问题"),
+    outputs=[
+        gr.Textbox(lines=15, label="故事"),
+        gr.Markdown(label="绘本"),
+    ],
+    title="绘本生成器",
+    description="输入一个问题，生成一个故事和绘本图片",
+    theme="compact",
+)
+
+
+if __name__ == "__main__":
+    demo.launch()