Upload folder using huggingface_hub

app.py

@@ -32,12 +32,19 @@ def main():
     )
 
     st.title("SongLift LyrGen2")
-    st.sidebar.markdown(f"**Model:** {Settings.LLM_MODEL}")
     if st.sidebar.button("New Song"):
         st.session_state.chat_history = []
         st.session_state.current_lyrics = None
         st.rerun()
 
+    # Show DB stats if available
+    if "db_stats" in st.session_state:
+        stats = st.session_state.db_stats
+        st.sidebar.markdown("---")
+        st.sidebar.metric("Artists", f"{stats['artists']:,}")
+        st.sidebar.metric("Songs", f"{stats['songs']:,}")
+        st.sidebar.metric("Chunks", f"{stats['chunks']:,}")
+
     # Only run startup once per session
     if 'initialized' not in st.session_state:
         print("===== Application Startup at", datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "=====\n")
@@ -51,6 +58,21 @@ def main():
             st.session_state.chat_history = []
             st.session_state.current_lyrics = None
             st.session_state.initialized = True
+
+            # Compute DB stats for sidebar
+            collection = generator.vector_store._collection
+            all_meta = collection.get(include=["metadatas"])["metadatas"]
+            artists = set()
+            songs = set()
+            for m in all_meta:
+                artists.add(m.get("artist", ""))
+                songs.add((m.get("artist", ""), m.get("song_title", "")))
+            st.session_state.db_stats = {
+                "chunks": len(all_meta),
+                "songs": len(songs),
+                "artists": len(artists),
+            }
+
             print("Generator initialized successfully")
             st.rerun()
         except Exception as e: