[streamlit] 스트림릿 사용법 (입력 및 출력, 파일 업로드 및 출력)

[streamlit] python으로 간단하게 웹앱 만들기

라이브러리 설치부터 간단한 웹앱 띄우기는 위 링크에 !

---

웹앱에 문장 출력

title

st.title('hello’)

header

st.header('hello’)

subheader

st.subheader('hello’)

text

st.text('hello’)

 

위 순서대로 출력한 결과

---

사이드바

menu = ['첫번째', '두번째', '세번째']
choice = st.sidebar.selectbox('메뉴', menu)

---

입력

문자 입력

name = st.text_input('문자 입력')
st.text(f"문자 : {name}")

 

문장 입력

message = st.text_area('문장 입력')
st.text(f"문장 : {message}")

 

숫자 입력

number = st.number_input('숫자 입력')
st.text(number)

 

날짜 입력

my_date = st.date_input("날짜 입력")
st.text(my_date)

 

시간 선택

my_time = st.time_input('시간 선택')
st.text(my_time)

 

색 선택

color = st.color_picker('색 선택')
st.text(color)

 

비밀번호 입력

password = st.text_input('비밀번호 입력', type='password')
st.text(password)

---

이미지 & 영상 출력

from PIL import Image

# 폴더 내의 이미지 출력
img = Image.open('image.JPG')
st.image(img)

# 경로로 이미지 출력
st.image('streamlit/image.JPG')

# 영상 출력
video_file = open('video.mp4', 'rb')
st.video(video_file)

---

파일 업로드

이미지 파일 업로드

img_file = st.file_uploader('이미지 업로드', type=['png', 'jpg', 'jpeg'])

이미지명이 고유하도록 이미지명 변경

from datetime import datetime

if img_file is not None:
	current_time = datetime.now()
	filename = current_time.isoformat().replace(":", "_")
	img_file.name = filename + '.jpg'

이미지명이 겹칠 수도 있기 때문에 업로드한 시간을 활용하여 이미지명을 변경한다

 

실제로 저장

import os

if not os.path.exists('image'):
    os.makedirs('image')
	
with open(os.path.join('image', img_file.name), 'wb') as f:
    f.write(img_file.getbuffer())

st.success('파일 업로드 성공')

실제로 저장된 모습

 

CSV 파일 업로드

csv_file = st.file_uploader('CSV 업로드', type=['csv'])

해당 부분만 바꿔주면 나머지는 동일하게 진행된다

---

앞의 코드를 사용하여 이미지 업로드 및 저장 후 출력

import streamlit as st
import pandas as pd
import numpy as np
from PIL import Image
from datetime import datetime
import os

# 이미지 업로드
st.subheader('이미지 업로드')
img_file = st.file_uploader('',type=['png', 'jpg', 'jpeg'])

if img_file is not None:

	# 이미지명이 고유하도록 시간을 활용하여 변경
	current_time = datetime.now()
	filename = current_time.isoformat().replace(":", "_")
	img_file.name = filename + '.jpg'

	# 실제로 저장
	if not os.path.exists('image'):
		os.makedirs('image')
		
	with open(os.path.join('image', img_file.name), 'wb') as f:
		f.write(img_file.getbuffer())

	st.success('파일 업로드 성공! 변경된 고유한 파일명 : ' + img_file.name)


	# 경로로 이미지 출력
	st.subheader('업로드한 이미지')
	img = Image.open('image/'+img_file.name)
	st.image(img)

 

streamlit을 실행시키고 이미지를 업로드하여 파일이 업로드된 것과 변경된 이미지명을 확인할 수 있고, 업로드한 이미지를 출력한 결과도 마찬가지로 확인할 수 있다

---