Update app.py

app.py

@@ -1,23 +1,28 @@
-import streamlit as st
 import pandas as pd
 import duckdb
 import polars as pl
 import pyarrow.csv as pv
+import pyarrow.parquet as pq
 import time
 import os
+
+
 import tempfile
 import matplotlib.pyplot as plt
 
+# === DEBUG + TEST RAPIDE ===
 print("=== APP STARTING ===")
+st.success("App démarrée avec succès !")
 
+# === CONFIG PAGE ===
 st.set_page_config(
-    page_title="Speed Loader Benchmark",
-    page_icon="lightning",
+    page_title="⚡ Speed Loader Benchmark",
+    page_icon="⚡",
     layout="wide",
     initial_sidebar_state="expanded"
 )
 
-# === CSS : boutons identiques + beau rendu ===
+# === CSS POUR BOUTONS ÉGAUX + BEAUX ===
 st.markdown("""
 <style>
     .stButton > button {
@@ -25,145 +30,159 @@ st.markdown("""
         font-size: 1.1rem !important;
         font-weight: bold;
         border-radius: 12px;
-        white-space: pre-line;
-        text-align: center;
-        line-height: 1.4;
+        border: 2px solid #e0e0e0;
+        background: linear-gradient(145deg, #f5f5f5, #e0e0e0);
+        box-shadow: 4px 4px 8px #cbced1, -4px -4px 8px #ffffff;
+        transition: all 0.3s;
+    }
+    .stButton > button:hover {
+        border: 2px solid #4CAF50;
+        transform: translateY(-2px);
+        box-shadow: 0 10px 20px rgba(0,0,0,0.1);
+    }
+    .stButton > button:active {
+        transform: translateY(2px);
     }
 </style>
 """, unsafe_allow_html=True)
 
-# === FONCTIONS DE CHARGEMENT (toutes corrigées) ===
-def load_pandas(path):
+# === FONCTIONS DE CHARGEMENT ===
+def load_with_pandas(path):
     start = time.time()
     df = pd.read_csv(path)
     return df, time.time() - start
 
-def load_polars(path):
+def load_with_polars(path):
     start = time.time()
-    df = pl.read_csv(path, infer_schema_length=10000).to_pandas()
-    return df, time.time() - start
+    df = pl.read_csv(path)
+    return df.to_pandas(), time.time() - start
 
-def load_duckdb(path):
+def load_with_duckdb(path):
     start = time.time()
     df = duckdb.read_csv(path).df()
     return df, time.time() - start
 
-def load_pyarrow(path):
+def load_with_pyarrow(path):
     start = time.time()
-    parse_options = pv.ParseOptions(newlines_in_values=True)
-    table = pv.read_csv(path, parse_options=parse_options)
+    table = pv.read_csv(path)
     df = table.to_pandas()
     return df, time.time() - start
 
-# === SESSION STATE ===
-for key in ["file_path", "file_name", "temp_file"]:
-    if key not in st.session_state:
-        st.session_state[key] = None
-
 # === SIDEBAR ===
-st.sidebar.markdown("# Speed Benchmark")
-st.sidebar.markdown("### Fichiers de test (~30 Mo)")
+st.sidebar.markdown("# ⚡ Speed Benchmark")
+st.sidebar.markdown("### 🧪 Fichiers de test (~30 Mo)")
 
-c1, c2 = st.sidebar.columns(2)
-with c1:
-    if st.button("Faker\nText", use_container_width=True, type="secondary"):
+col1, col2 = st.sidebar.columns(2)
+
+with col1:
+    if st.button("🧑‍💻 Faker\nText", use_container_width=True, type="secondary"):
         if os.path.exists("faker_text.csv"):
             st.session_state.file_path = "faker_text.csv"
             st.session_state.file_name = "faker_text.csv"
-            st.rerun()
         else:
-            st.sidebar.error("faker_text.csv absent")
+            st.sidebar.error("faker_text.csv manquant")
 
-with c2:
-    if st.button("Numeric\nOnly", use_container_width=True, type="secondary"):
+with col2:
+    if st.button("🔢 Numeric\nOnly", use_container_width=True, type="secondary"):
         if os.path.exists("numeric_only.csv"):
             st.session_state.file_path = "numeric_only.csv"
             st.session_state.file_name = "numeric_only.csv"
-            st.rerun()
         else:
-            st.sidebar.error("numeric_only.csv absent")
+            st.sidebar.error("numeric_only.csv manquant")
 
 st.sidebar.markdown("---")
 
-uploaded = st.sidebar.file_uploader("Ou chargez votre fichier", type=["csv", "parquet", "txt"])
-if uploaded:
-    suffix = os.path.splitext(uploaded.name)[1]
-    with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as f:
-        f.write(uploaded.read())
-        st.session_state.file_path = f.name
-        st.session_state.file_name = uploaded.name
-        st.session_state.temp_file = f.name
-    st.sidebar.success(f"{uploaded.name}")
-    st.rerun()
-
-# === MAIN ===
-st.title("Comparaison de vitesse de chargement")
-st.markdown("**Pandas • Polars • DuckDB • PyArrow** – Qui gagne en 2025 ?")
-
-if not st.session_state.file_path:
-    st.info("Choisissez un fichier dans la sidebar")
+uploaded_file = st.sidebar.file_uploader(
+    "📁 Ou chargez votre fichier",
+    type=["csv", "parquet", "txt"],
+    help="CSV, Parquet"
+)
+
+if uploaded_file is not None:
+    bytes_data = uploaded_file.read()
+    with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]) as tmp:
+        tmp.write(bytes_data)
+        st.session_state.file_path = tmp.name
+        st.session_state.file_name = uploaded_file.name
+        st.session_state.temp_file = tmp.name  # pour nettoyage
+    st.sidebar.success(f"Chargé : {uploaded_file.name}")
+
+# === MAIN TITLE ===
+st.title("⚡ Comparaison de vitesse de chargement")
+st.markdown("**Qui est le plus rapide en 2025 ?**")
+
+if 'file_path' not in st.session_state:
+    st.info("👈 Choisissez un fichier de test ou uploadez le vôtre")
     st.stop()
 
-st.markdown(f"### Fichier actif : `{st.session_state.file_name}`")
+file_path = st.session_state.file_path
+file_name = st.session_state.file_name
 
-# === BOUTON DE LANCEMENT DANS LA SIDEBAR ===
-run = st.sidebar.button("Lancer le benchmark", type="primary", use_container_width=True)
+st.markdown(f"### 📊 Fichier sélectionné : `{file_name}`")
 
-if run:
-    st.markdown("### Résultats")
+if st.button("🚀 Lancer le benchmark complet", type="primary", use_container_width=True):
+    st.markdown("### ⏱️ Résultats en direct")
 
     results = []
 
-    # Pandas
-    with st.spinner("Pandas"):
-        df, t = load_pandas(st.session_state.file_path)
-        results.append({"Moteur": "Pandas", "Temps (s)": t, "Lignes": len(df)})
+    # === 1. Pandas ===
+    with st.spinner("Pandas (référence)..."):
+        df, t = load_with_pandas(file_path)
+        results.append(("🐼 Pandas", t))
+        st.success(f"Pandas → {t:.3f}s")
+
+    # === 2. Polars ===
+    with st.spinner("Polars (le roi)..."):
+        df, t = load_with_polars(file_path)
+        results.append(("⚡ Polars", t))
+        st.success(f"Polars → {t:.3f}s")
+
+    # === 3. DuckDB ===
+    with st.spinner("DuckDB (SQL power)..."):
+        df, t = load_with_duckdb(file_path)
+        results.append(("🦆 DuckDB", t))
+        st.success(f"DuckDB → {t:.3f}s")
+
+    # === 4. PyArrow ===
+    with st.spinner("PyArrow (C++ speed)..."):
+        df, t = load_with_pyarrow(file_path)
+        results.append(("🏹 PyArrow", t))
+        st.success(f"PyArrow → {t:.3f}s")
+
+    # === Nettoyage temp file si upload ===
+    if hasattr(st.session_state, 'temp_file'):
 
-    # Polars
-    with st.spinner("Polars"):
-        df, t = load_polars(st.session_state.file_path)
-        results.append({"Moteur": "Polars", "Temps (s)": t, "Lignes": len(df)})
 
-    # DuckDB
-    with st.spinner("DuckDB"):
-        df, t = load_duckdb(st.session_state.file_path)
-        results.append({"Moteur": "DuckDB", "Temps (s)": t, "Lignes": len(df)})
 
-    # PyArrow
-    with st.spinner("PyArrow"):
-        df, t = load_pyarrow(st.session_state.file_path)
-        results.append({"Moteur": "PyArrow", "Temps (s)": t, "Lignes": len(df)})
 
-    # Nettoyage fichier temporaire
-    if st.session_state.temp_file:
         try:
             os.unlink(st.session_state.temp_file)
-            st.session_state.temp_file = None
+
         except:
             pass
 
-    # === AFFICHAGE ===
-    df_res = pd.DataFrame(results).sort_values("Temps (s)")
-
-    # Correction du bug len() sur numpy.int64 → on force int()
-    n_lines = int(df_res["Lignes"].iloc[0])
-
-    col1, col2 = st.columns(2)
-    with col1:
-        st.metric("Vainqueur", df_res.iloc[0]["Moteur"])
-        st.metric("Temps", f"{df_res.iloc[0]['Temps (s)']:.3f} s")
-    with col2:
-        st.metric("Lignes chargées", f"{n_lines:,}")
-        st.metric("Moteurs testés", len(df_res))
-
-    # Graphique
-    fig, ax = plt.subplots(figsize=(8, 3))
-    bars = ax.barh(df_res["Moteur"], df_res["Temps (s)"], color=["#FF6B6B","#4ECDC4","#45B7D1","#96CEB4"])
-    for bar in bars:
-        ax.text(bar.get_width()+0.01, bar.get_y()+bar.get_height()/2,
-                f'{bar.get_width():.3f}s', va='center', fontweight='bold')
-    ax.set_xlabel("Temps (secondes)")
-    ax.set_title(f"Vainqueur : {df_res.iloc[0]['Moteur']} ({df_res.iloc[0]['Temps (s)']:.3f}s)")
+
+
+    # === GRAPHIQUE FINAL ===
+    results_df = pd.DataFrame(results, columns=["Moteur", "Temps (s)"]).sort_values("Temps (s)")
+
+    fig, ax = plt.subplots(figsize=(10, 6))
+    colors = ["#FF6B6B", "#4ECDC4", "#45B7D1", "#96CEB4"]
+    bars = ax.barh(results_df["Moteur"], results_df["Temps (s)"], color=colors)
+
+    for i, bar in enumerate(bars):
+        width = bar.get_width()
+        ax.text(width + max(results_df["Temps (s)"]) * 0.01, bar.get_y() + bar.get_height()/2,
+                f'{width:.3f}s', va='center', fontweight='bold', fontsize=12)
+
+    ax.set_xlabel("Temps de chargement (secondes)", fontsize=12)
+    ax.set_title(f"🏆 Vainqueur : {results_df.iloc[0]['Moteur']} ({results_df.iloc[0]['Temps (s)']:.3f}s)", 
+                 fontsize=16, fontweight="bold", color="#1A5F7A")
     ax.invert_yaxis()
+    ax.grid(axis='x', alpha=0.3)
+
     st.pyplot(fig)
-    plt.close(fig)
+    plt.close(fig)
+
+    st.balloons()
+    st.markdown("### 🔥 **Polars gagne 99% du temps en 2025 !**")