import os
os.system("pip install paddlenlp==2.5.2")
os.system("pip install ppdiffusers==0.11.1")

from ppdiffusers import DiffusionPipeline, DPMSolverMultistepScheduler
import paddle
import streamlit as st

st.header("用LoRA 和 DreamBooth画出你眼中的HomeTown")
st.image("lora_outputs/validation_images/700.png")

st.subheader("设置Prompt")

st.write("输入你眼中家乡的几个关键句吧!看看能不能绘制出你心中的家乡!")

pr1 = st.text_input('你心中的家乡：',"")

test = st.button("⚡⚡开始生成⚡⚡")
if test:
  pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)

  prompt = pr1
  image = pipe(prompt).images[0]

  image.save("demo.png")
  st.image("demo.png")

  st.success('推理完毕！！', icon="✅")
  st.write("移动端长按图片可保存，PC端右键图片可另存到本地。")