Added everything else

app.py

@@ -4,16 +4,16 @@ import streamlit as st
 import os
 import base64
 
-st.set_page_config(layout="wide", page_title='CoNR demo', page_icon="🪐")
+st.set_page_config(layout="wide", page_title="CoNR demo", page_icon="🪐")
 
-st.title('CoNR demo')
+st.title("CoNR demo")
 st.markdown(""" <style> 
             #MainMenu {visibility: hidden;} 
             footer {visibility: hidden;} 
             </style> """, unsafe_allow_html=True)
 
 def get_base64(bin_file):
-    with open(bin_file, 'rb') as f:
+    with open(bin_file, "rb") as f:
         data = f.read()
     return base64.b64encode(data).decode()
 
@@ -35,19 +35,25 @@ def get_base64(bin_file):
 # set_background('ipad_bg.png')
 
 upload_img = (st.file_uploader("输入character sheet", "png", accept_multiple_files=True))
+upload_img2 = (st.file_uploader("pose images", "png", accept_multiple_files=True))
 
-if st.button('RUN!'):
+# os.system("sh download.sh")
+
+if st.button("RUN!"):
     if upload_img is not None:
         os.makedirs("character_sheet", exist_ok=True)
         for i in range(len(upload_img)):
-            with open('character_sheet/{}.png'.format(i), 'wb') as f:
+            with open("character_sheet/{}.png".format(i), "wb") as f:
                 f.write(upload_img[i].read())
-
-        st.info('努力推理中...')
-        os.system('sh infer.sh')
-        st.info('Done!')
-        video_file=open('output.mp4', 'rb')
+        os.makedirs("poses", exist_ok=True)
+        for i, e in enumerate(upload_img2):
+            with open(f"poses/{i}.png", "wb") as f:
+                f.write(e.read())
+        st.info("努力推理中...")
+        os.system("sh infer.sh")
+        st.info("Done!")
+        video_file=open("output.mp4", "rb")
         video_bytes = video_file.read()
         st.video(video_bytes, start_time=0)
     else:
-        st.info('还没上传图片呢> <')
+        st.info("还没上传图片呢> <")