Code Editor for app.py

import streamlit as st
import pandas as pd
from sqlalchemy import create_engine, text
from sqlalchemy.orm import Session

# Création de la connexion à la base de données
engine = create_engine('sqlite:///database.db')

def load_table_data(table_name):
    """Charge les données d'une table spécifique"""
    try:
        # Utiliser pandas pour lire directement la table
        query = f"SELECT * FROM {table_name}"
        df = pd.read_sql_query(query, engine)
        return df
    except Exception as e:
        st.error(f"Erreur lors de la lecture de la table: {str(e)}")
        return None

def main():
    st.title("Affichage des données d'une table SQL")
    
    # Obtenir la liste des tables disponibles dans la base de données
    with Session(engine) as session:
        query = text("SELECT name FROM sqlite_master WHERE type='table'")
        tables = [row[0] for row in session.execute(query)]
    
    if not tables:
        st.warning("Aucune table n'a été trouvée dans la base de données.")
        return
        
    # Sélecteur de table
    selected_table = st.selectbox(
        "Choisissez une table à afficher:",
        tables
    )
    
    if selected_table:
        # Charger et afficher les données
        df = load_table_data(selected_table)
        if df is not None:
            st.write(f"### Données de la table '{selected_table}'")
            st.dataframe(df)
            
            # Option pour télécharger les données
            csv = df.to_csv(index=False)
            st.download_button(
                label="Télécharger les données en CSV",
                data=csv,
                file_name=f"{selected_table}.csv",
                mime="text/csv"
            )

if __name__ == '__main__':
    main()