如何在Streamlit应用中自动关联Google Drive图片URL并展示图片

核心思路 #

Google Drive桌面端同步的本地文件，对应云端的唯一文件ID，我们需要批量获取这些ID并与数据库中的文件名关联，再生成可在Streamlit中直接展示的嵌入URL。

具体实现步骤 #

1. 批量获取本地文件对应的Drive文件ID

更可靠的方式是通过Google Drive API批量查询云端文件的关联信息：

• 先创建Google Cloud项目，启用Drive API，生成服务账号密钥（JSON格式）。
• 用google-api-python-client库遍历云端sdr/pos目录下的所有图片，获取每个文件的文件名和云端ID。

示例代码：

from googleapiclient.discovery import build
from google.oauth2.service_account import Credentials

# 加载服务账号密钥
creds = Credentials.from_service_account_file('service_account_key.json')
drive_service = build('drive', 'v3', credentials=creds)

# 查询云端sdr/pos文件夹下的所有图片
folder_id = "你的云端sdr/pos文件夹ID"  # 从Drive网页版URL中提取
results = drive_service.files().list(
    q=f"'{folder_id}' in parents and mimeType contains 'image/'",
    fields="files(id, name)"
).execute()
# 生成文件名到云端ID的映射字典
file_map = {item['name']: item['id'] for item in results.get('files', [])}

2. 将映射关系更新到数据库

把file_map中的文件名-文件ID对应关系，批量更新到数据库（以SQL为例）：

-- 假设数据表名为image_files，包含filename和drive_id字段
UPDATE image_files
SET drive_id = CASE filename
    WHEN 'image1.png' THEN '1AWhXeAGB8qBWe9kItc5zW9eLytIguOQF'
    WHEN 'image2.png' THEN '对应云端ID'
    -- 其他文件依次补充
END;

3. 在Streamlit中生成可展示的URL并渲染

Google Drive默认的view URL无法直接嵌入Streamlit，需要改用嵌入格式URL：https://drive.google.com/uc?id=FILE_ID。在Streamlit中从数据库取出对应ID，生成URL后展示：

import streamlit as st
import sqlite3

# 连接数据库
conn = sqlite3.connect('your_database.db')
cursor = conn.cursor()

# 示例：获取指定图片的云端ID
cursor.execute("SELECT drive_id FROM image_files WHERE filename = 'image1.png'")
drive_id = cursor.fetchone()[0]

# 生成可嵌入的URL并展示图片
image_url = f"https://drive.google.com/uc?id={drive_id}"
st.image(image_url, caption="image1.png")

额外注意事项 #

• 确保云端图片设置了公开可查看权限（或允许服务账号访问），否则Streamlit无法加载图片。
• 若本地文件有重命名，需重新同步云端后再执行批量映射，避免文件名不匹配。

内容的提问来源于stack exchange，提问作者honeymoon