MakiAi commited on
Commit
5a9e67c
1 Parent(s): 690144b

📦 feat: ユーティリティ関数の追加

Browse files

- 画像をバイト列に変換する関数を実装しました。
- Streamlitのダウンロードボタンを作成する関数を追加しました。

Files changed (1) hide show
  1. pic_to_header/modules/utils.py +23 -0
pic_to_header/modules/utils.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from PIL import Image
3
+ import io
4
+
5
+ def convert_image_to_bytes(image):
6
+ """画像をバイト列に変換"""
7
+ if image is None:
8
+ return None
9
+
10
+ img_byte_arr = io.BytesIO()
11
+ image.save(img_byte_arr, format='PNG')
12
+ return img_byte_arr.getvalue()
13
+
14
+ def create_download_button(image, filename):
15
+ """ダウンロードボタンを作成"""
16
+ if image is not None:
17
+ image_bytes = convert_image_to_bytes(image)
18
+ st.download_button(
19
+ label="画像をダウンロード",
20
+ data=image_bytes,
21
+ file_name=filename,
22
+ mime="image/png"
23
+ )