Streamlit : 비주얼 스튜디오 코드의 터미널을 이용해서 스트림릿 실행하는 방법

########파일 업로드 방법########

########이미지 파일, csv 파일 업로드########

from numpy import isfortran

from sklearn.ensemble import IsolationForest

import streamlit as st

from PIL import Image

import pandas as pd

import os

from datetime import datetime

# 디렉토리 정보와 파일을 알려주면, 해당 디렉토리에

# 파일을 저장하는 함수

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())

    return st.success("Saved file : {} in {}".format(file.name, directory))

def main():

        # 사이드바 만들기

        st.title('파일 업로드 프로젝트')

        menu = ['Image','CSV','About']

        choice = st.sidebar.selectbox('메뉴',menu)

        if choice == menu[0]:

            st.subheader('이미지 파일 업로드')

            upload_file = st.file_uploader('이미지 파일 선택',type=['jpg','png','jpeg'])

            if upload_file is not None:  #있으면

                print(upload_file.name)

                print(upload_file.size)

                print(upload_file.type)

            #파일명을 유니크하게 만들어서 저장

            # 현재시간을 활용해서 만든다.

            currnet_time = datetime.now()

            print(currnet_time.isoformat().replace(':','_'))

            new_filename = currnet_time.isoformat().replace(':','_') + '.jpg'

            upload_file.name = new_filename

            save_uploaded_file('temp', upload_file)

        elif choice == menu[1]:

            st.subheader('CSV 파일 업로드')

        else:

            st.subheader('파일 업로드 프로젝트 입니다.')

if __name__ == '__main__':

    main()

 

---