【Streamlit】SQLデータベースと連携してデータを表示

pip install streamlit pandas

# create_books_db.py
import sqlite3

# データベース接続（なければ自動作成）
conn = sqlite3.connect("books.db")
c = conn.cursor()

# booksテーブルの作成
c.execute("""
CREATE TABLE IF NOT EXISTS books (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    title TEXT NOT NULL,
    author TEXT NOT NULL,
    published_year INTEGER
)
""")

# サンプルデータを追加（任意）
sample_books = [
    ("吾輩は猫である", "夏目漱石", 1905),
    ("人間失格", "太宰治", 1948),
    ("雪国", "川端康成", 1947),
]

c.executemany("INSERT INTO books (title, author, published_year) VALUES (?, ?, ?)", sample_books)

conn.commit()
conn.close()
print("データベースとテーブルを作成しました。")

import sqlite3

def get_connection():
    """データベース接続を返す関数"""
    return sqlite3.connect("books.db")

import streamlit as st
import pandas as pd
import sqlite3

def get_connection():
    """データベース接続を返す関数"""
    return sqlite3.connect("books.db")


st.title("📚 書籍管理アプリ")

# タブで機能を分ける
tab1, tab2 = st.tabs(["📖 書籍一覧", "➕ 書籍登録"])

# 書籍一覧表示
with tab1:
    st.header("📖 登録された書籍一覧")
    conn = get_connection()
    df = pd.read_sql_query("SELECT * FROM books", conn)
    conn.close()

    st.dataframe(df, use_container_width=True)

# 書籍登録フォーム
with tab2:
    st.header("➕ 新しい書籍を登録")

    with st.form("book_form", clear_on_submit=True):
        title = st.text_input("書籍名", max_chars=100)
        author = st.text_input("著者名", max_chars=50)
        year = st.number_input("出版年", min_value=1000, max_value=9999, step=1)

        submitted = st.form_submit_button("登録する")

        if submitted:
            if title and author:
                conn = get_connection()
                cursor = conn.cursor()
                cursor.execute(
                    "INSERT INTO books (title, author, published_year) VALUES (?, ?, ?)",
                    (title, author, int(year))
                )
                conn.commit()
                conn.close()
                st.success("✅ 書籍を登録しました！")
            else:
                st.warning("⚠️ 書籍名と著者名は必須です。")

streamlit run book_app.py