import numpy as np
import gradio as gr

# 生成一个文生图的gradio界面


def mock_img(text: str):
    return [
        {
            "caption": "植物克隆过程示意图",
            "url": "https://dashscope-result-sh.oss-cn-shanghai.aliyuncs.com/1d/85/20240710/1b61f1c0/5a69fb02-051b-4708-9a19-83b8f1f5fce2-1.png?Expires=1720710640&OSSAccessKeyId=LTAI5tQZd8AEcZX6KZV4G8qL&Signature=lk4qpxmXBYPzWl%2FHdfpg4RnSgSU%3D",
            "description": "图中展示了一棵植物和从植物上剪下的一小段枝条，该枝条被插入土壤中。过一段时间，这小段 枝条生根发芽，长成了一棵新的、和原植物一模一样的新植物。",
        },
        {
            "caption": "克隆羊多利的照片",
            "url": " https://dashscope-result-hz.oss-cn-hangzhou.aliyuncs.com/1d/63/20240710/522176a8/0572556b-f855-4fd4-bd02-aea8a72d7fd1-1.png?Expires=1720710681&OSSAccessKeyId=LTAI5tQZd8AEcZX6KZV4G8qL&Signature=RT21mWAWpFTwRA5LU20GomgBxuk%3D",
            "description": " 图中展示了克隆羊多利站在草地上，她看起来非常健康，和普通的羊没有区别。背景是实验室的轮廓 ，暗示了她的诞生地。",
        },
    ]


def mock_story(query: str):
    return """小朋友，你听说过“克隆”这个词吗？让我来给你讲一个有趣的故事吧！

有一天，科学家们想出了一个非常神奇的方法，可以让一个生命体“复制”出另一个一模一样的生命体，这个方法就叫做“克隆”。你可以把“克隆”想象成是一台神奇的复印机，但是这台复印机不是复印纸张，而是复印生命哦！

1997年的时候，科学家们用这种方法成功地“复印”出了一只名叫多利的绵羊。你知道吗，多利羊并没有爸爸，它是由一只母羊的细胞复制而来的。这就像是我们从电脑上复制一个文件一样，只不过这次我们复制的是一个活生生的羊！

克隆羊多利的诞生，就像是打开了一扇新的大门，让人们看到了生命的奇妙和科学的无限可能。从此以后，人们开始更加关注和研究克隆技术，希望用它来帮助我们解决更多的问题。

所以，克隆羊就是通过克隆技术创造出来的一只羊，它没有父亲，只有母亲，是科学家们用神奇的“生命复印机”复制出来的哦！"""


from kg import story_to_imgs

story_existed = False


def generate_image(story):

    # res = story_to_imgs(story)
    res = mock_img(story)
    md = ""
    # 改为markdown
    for i in range(len(res)):
        md += f"## {res[i]['caption']}\n\n"
        md += f"{res[i]['description']} \n "
        md += f"<img src=\"{res[i]['url']}\" width = \"500\" div align=center />\n"
        if i != len(res) - 1:
            md += "\n\n"
        # md += "![{res[i]['caption']}]({res[i]['url']})\n\n"

    # 保存为md文件
    with open("生成的绘本.md", "w", encoding="utf-8") as f:
        f.write(md)
    return md


from answerQuestions import generateStory


def generate_story(query):
    story_existed = True
    return mock_story(query)
    return generateStory(query)


with gr.Blocks() as demo:
    gr.Markdown("# 绘本生成器")

    with gr.Column():
        with gr.Row():
            input_text = gr.Textbox(label="输入")
        with gr.Row():
            clear_btn = gr.Button("清空")
            story_btn = gr.Button("生成故事", variant="primary")
            img_btn = gr.Button(
                "生成绘本", variant="primary", interactive=story_existed
            )
        with gr.Row():
            story_output = gr.Textbox(label="故事", interactive=False)
            story_btn.click(generate_story, inputs=input_text, outputs=story_output)
            # with gr.Row():

            img_output = gr.Markdown(label="绘本")
            img_btn.click(generate_image, inputs=story_output, outputs=img_output)

        clear_btn.click(
            lambda x: [
                gr.update(value=""),
                gr.update(value=""),
                gr.update(value=""),
            ],
            None,
            [input_text, story_output, img_output],
        )

if __name__ == "__main__":
    demo.launch()