streamlitのfile_uploaderでアップロードしたファイルをメモリ上に保存したい

Question

前提

streamlitのfile_uploaderでアップロードしたファイルをメモリ上に保存したい

streamlitにてファイルをアップロードし、処理を行うプログラムを作っています。
file_uploaderを使ってローカルからファイルをアップロードした後、
ローカル上に保存して処理を行っていましたが、メモリ上に保存して処理をする方法に変えたく思っています。

実現したいこと

file_uploaderでアップロードしたファイルをメモリ上に保存する

発生している問題・エラーメッセージ

BytesIOを使って、メモリ上に保持したいと思っていたのですが
file_uploaderのアップロードしたデータの取得方法がわからず困っています。
(今回、PDFファイルを扱います)

該当のソースコード

import streamlit as st
import io

upload_file = st.file_uploader('ファイルを選択してください', type='pdf')

##この後のメモリ上への保存方法が分からないです、、

初学者であるため、初歩的な質問ですがご教授頂ければと思います。

Comments

[8524ba23]

なぜファイルではなく「メモリ上に保持」したいのか、具体的なユースケースとともに提示ください。
なお、streamlitの動作原理（何らか操作のたびにコードが再実行される）の都合上
メモリに保持する手法は実現しづらいと思います。

Answer 1

質問のコードの upload_file が束縛されているオブジェクトはすでにメモリ上に保存されたデータですよ。

しかも、公式のドキュメントを見ると、

If accept_multiple_files is False, returns either None or an UploadedFile object.
The UploadedFile class is a subclass of BytesIO

となっていますので、型もBytesIOのデータとなっています。

このページのサンプルにもあるように、byte列が必要であれば、getvalue()メソッドを使えばOKです。