update

Dockerfile

@@ -23,9 +23,6 @@ RUN mkdir app
 WORKDIR $HOME/app
 COPY . $HOME/app
 
-RUN chown -R user:user /home/user/app
-RUN chmod -R 755 /home/user/app
-
 EXPOSE 8501
 CMD streamlit run app.py \
     --server.headless true \

app.py

@@ -188,7 +188,7 @@ if __name__ == "__main__":
                         if screenshot:
                             st.session_state.image_index = 0
                             print("get screenshot!")
-                            image_folder = os.getcwd() + "/images/"
+                            image_folder = "./"
                             with open(image_folder+str(st.session_state.image_index)+".png", "wb") as f:
                                 f.write(base64.b64decode(screenshot))
                             image_files = os.listdir(image_folder)
@@ -280,7 +280,7 @@ class VisitPage(BaseTool):
                     st.markdown('**📸Observation**')
                     if screenshot:
                         print("get screenshot!")
-                        image_folder = os.getcwd() + "/images/"
+                        image_folder = "./"
                         with open(image_folder+str(st.session_state.image_index+1)+".png", "wb") as f:
                             f.write(base64.b64decode(screenshot))
                         st.session_state.image_index += 1