본문 바로가기
Streamlit

스트림릿 다운로드버튼 download_button( )

by 하니__ 2024. 5. 2.

유저가 파일을 업로드하여 재가공하는 사이트를 만들었다고 하자
 
 
 

 
 
유저가 재 가공을 했으면 그것을 저장할수 있게끔 해주자
 
 


 
 

csv = df.to_csv(index=False)
st.download_button(label="CSV 파일 저장", data=csv, file_name="new_data.csv", mime="text/csv")

를 입력하여
 
 
csv를 저장(생성)하는 함수를 csv라는 변수로 저장
index=False는

항상 이상한 인덱스가 저장되기때문에 빼버리는것,


그리고 스트림릿에서 제공하는 download_button을

이용하여 유저가 저장할수 있게 하는데

첫번째 파라미터는 버튼의 표현되는 이름이고,

2번째 파라미터는 다운로드 하게끔 하는 데이터가 어떤것인지

= 위에서 저장한 csv 변수

 

 

file_name="new_data.csv"
mime="text/csv"

1. file_name같은 경우 적지 않게 되면

유저가 이름을 직접 저장하게끔 하는것이 가능하고

 

2. mime 같은 경우 파일의 형식이 csv다 라고 알려주는 것에 불과하지만

 

2의 경우 csv 파일인것을  정확히 알려주어 적절한 처리를 유도하는 기능이고

1의 경우 마찬가지로 이 파일이 csv파일임을 확실히 알려주기 위하여 지정해주는 편이다

 

물론 적지 않아도 작동은 가능하다

 

 

저장이 잘 되는것을 볼 수 있다