Web Dashboard

streamlit 에 파일을 업로드 & 저장하는 함수 만들기

yugyeong 2022. 12. 13. 12:29

 

아래의 함수를 이용하여서 streamlit 웹페이지에서 파일을 업로드 하고 저장할 수 있도록 만든다.

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
# 파일을 업로드 하는 방법
 
import streamlit as st
import pandas as pd 
import os 
from datetime import date, datetime
from PIL import Image
 
# 함수정의 
# 디렉토리(폴더)명과 파일을 알려주면,
# 해당 디렉토리에 파일을 저장해 주는 함수.
def save_uploaded_file(directory, file) :
    # 1. 디렉토리가 있는지 확인하여, 없으면 먼저, 디렉토리부터 만든다.
    if not os.path.exists(directory) :
        os.makedirs(directory)
 
    # 2. 디렉토리가 있으니, 파일을 저장한다.
    with open(os.path.join(directory, file.name), 'wb') as f:
        f.write(file.getbuffer())
    
    # 3. 파일 저장이 성공했으니, 화면에 성공했다고 보여주면서 리턴
    return st.success('{} 에 {} 파일이 저장되었습니다.'.format(directory, file.name))
 
 
def main():
    st.title('파일 업로드 프로젝트')
 
    menu = ['Image', 'CSV', 'About']
 
    choice = st.sidebar.selectbox('메뉴', menu )
 
    if choice == 'Image' :
        st.subheader('이미지 파일 업로드')
 
        file = st.file_uploader('이미지를 업로드 하세요', type=['jpg','jpeg','png'])
 
        if file is not None :
            # st.text(file.name)
            # st.text(file.size)
            # st.text(file.type)
 
            # 파일명을 일관성있게, 회사의 파일명 규칙대로 바꾼다.
            # 현재시간을 조합하여 파일명을 만들면, 
            # 유니크하게 파일명을 지을수 있다.
 
            current_time = datetime.now()
            print(current_time.isoformat().replace(':','_') )
            current_time = current_time.isoformat().replace(':','_')
            print( current_time + '.jpg' )
 
            file.name = current_time + '.jpg'
 
            # 바꾼파일명으로, 파일을 서버에 저장한다.
            save_uploaded_file('tmp', file)
 
 
            # 파일을 웹 화면에 나오게.
            img = Image.open(file)
            st.image(img)
 
    elif choice == 'CSV' :
        st.subheader('CSV 파일 업로드')
 
        file = st.file_uploader('CSV파일 업로드', type=['csv'])
 
        if file is not None :
            # 파일명을 유니크하게 만든다.
            current_time = datetime.now()
            current_time = current_time.isoformat().replace(':','_')
            file.name = current_time + '.csv'
 
            # 파일을 서버에 저장한다.
            save_uploaded_file('csv', file)
 
            # csv 파일은, 판다스로 읽어서 화면에 보여준다.
            df = pd.read_csv(file)
            st.dataframe( df )
 
    elif choice == 'About' :
        st.subheader('파일 업로드 프로젝트 입니다.')
 
 
if __name__=='__main__' :
    main()
 cs