如何在 Streamlit 中下载多个文件

Question

我需要为多个文件制作一个下载按钮。 Streamlit 的下载按钮不允许您下载多个文件。 我试着做几个按钮，但是当我点击第一个按钮时 rest 就消失了。 有什么方法可以在 Streamlit 中下载两个或更多文件？

我从 Github 尝试了这个解决方案，代码如下所示：

if st.button("Rozpocznij proces"):
    raport2 = Raport.raport_naj_10(gender,year,week,engine)
    raportM = raport2[0]
    raportO = raport2[1]
    st.dataframe(raportM)
    st.dataframe(raportO)
    zipObj = ZipFile("sample.zip", "w")
    # Add multiple files to the zip
    zipObj.write("raportM")
    zipObj.write("raportO")
    # close the Zip File
    zipObj.close()

    ZipfileDotZip = "sample.zip"

    with open(ZipfileDotZip, "rb") as f:
        bytes = f.read()
        b64 = base64.b64encode(bytes).decode()
        href = f"<a href=\"data:file/zip;base64,{b64}\" download='{ZipfileDotZip}.zip'>\
            Click last model weights\
        </a>"
    st.sidebar.markdown(href, unsafe_allow_html=True)

但我得到这个错误：

FileNotFoundError: [WinError 2] Nie można odnaleźć określonego pliku: 'raportM'

它说找不到名为“raportM”的文件。

Answer 1

您遇到这些错误是因为代码是在假设您已经存储了文件并且您希望为它们生成 zip 文件的情况下编写的。 zipObj.write("raportM")正在寻找名为“raportM”的文件，但没有，因为在您的情况下，您没有存储这些文件。 我可以看到您将变量名作为文件传递，但这是行不通的。

在执行上述操作之前，您需要做的是在本地机器中保存这些变量名 CSV 文件。

在这种情况下，让我们修改您的代码。 但在此之前，我们需要为按钮st.button("Rozpocznij proces")初始化 session state 因为流光按钮没有回调。

processbtn = st.button("Rozpocznij proces")
# Initialized session states
if "processbtn_state" not in st.session_state:
    st.session_state.processbtn_state = False

if processbtn or st.session_state.processbtn_state:
    st.session_state.processbtn_state = True

    raport2 = Raport.raport_naj_10(gender,year,week,engine)
    raportM = raport2[0]
    raportO = raport2[1]
    st.dataframe(raportM)
    st.dataframe(raportO)

    # Save files
    raportM.to_csv('raportM.csv') # You can specify a directory where you want
    raportO.to_csv('raportO.csv') # these files to be stored

    # Create a zip folder
    zipObj = ZipFile("sample.zip", "w")

    # Add multiple files to the zip
    zipObj.write("raportM.csv")
    zipObj.write("raportO.csv")
    # close the Zip File
    zipObj.close()

    ZipfileDotZip = "sample.zip"

    with open(ZipfileDotZip, "rb") as f:
        bytes = f.read()
        b64 = base64.b64encode(bytes).decode()
        href = f"<a href=\"data:file/zip;base64,{b64}\" download='{ZipfileDotZip}.zip'>\
            Click last model weights\
        </a>"
    st.sidebar.markdown(href, unsafe_allow_html=True)

此时，当您密切注意您的目录时，您会发现'raportM.csv'和'raportO.csv'文件。 您可以将条件传递给下载按钮，以便在下载时删除文件，以防您不想保留它们。

注意：您可能会遇到fileNotFound 错误，但这并不意味着它不起作用，您只需要知道保存文件的位置。