update

src/streamlit_app.py

@@ -3,53 +3,77 @@ from datasets import load_dataset
 
 st.set_page_config(page_title="Paper Explorer", layout="wide")
 
-# 1. Initialize Session State for the page number
+# 1. Initialize Session States
+if 'current_view' not in st.session_state:
+    st.session_state.current_view = "trending"  # Default view
 if 'page_number' not in st.session_state:
     st.session_state.page_number = 0
 
-st.title("📚 Papers with code Explorer")
-
-# 2. Setup constants
-PAGE_SIZE = 10
-dataset_name = "pwc-archive/papers-with-abstracts"
-
-# 3. Sidebar Navigation
-st.sidebar.header("Navigation")
-
-def next_page():
-    st.session_state.page_number += 1
-
-def prev_page():
-    if st.session_state.page_number > 0:
-        st.session_state.page_number -= 1
-
-col1, col2, col3 = st.sidebar.columns([1, 2, 1])
-
-with col1:
-    st.button("⬅️", on_click=prev_page, disabled=(st.session_state.page_number == 0))
-with col2:
-    st.write(f"**Page {st.session_state.page_number + 1}**")
-with col3:
-    st.button("➡️", on_click=next_page)
-
-# 4. Fetch Data based on page
-offset = st.session_state.page_number * PAGE_SIZE
-
-with st.spinner(f"Loading papers {offset} to {offset + PAGE_SIZE}..."):
-    dataset = load_dataset(dataset_name, split="train", streaming=True)
-    papers = list(dataset.skip(offset).take(PAGE_SIZE))
-
-# 5. Display the Results
-for p in papers:
-    with st.container():
-        st.subheader(p['title'])
-        st.caption(f"ArXiv ID: {p.get('arxiv_id', 'N/A')}")
-        
-        # Display abstract (limiting length for readability)
-        abstract = p.get('abstract', 'No abstract available.')
-        st.write(abstract[:500] + ("..." if len(abstract) > 500 else ""))
-        
-        if p.get('arxiv_id'):
-            st.link_button("View on ArXiv", f"https://arxiv.org/abs/{p['arxiv_id']}")
-        
-        st.divider()
+# --- 2. Custom Top Navigation ---
+nav_container = st.container()
+with nav_container:
+    col1, col2, col3, col4, col5 = st.columns([1, 1.5, 2, 1, 1])
+    
+    with col1:
+        st.markdown("### 📚 PWC")
+    with col2:
+        # When clicked, this button sets the view to "trending"
+        if st.button("🔥 Trending", use_container_width=True):
+            st.session_state.current_view = "trending"
+    with col3:
+        if st.button("🏆 Browse SOTA", use_container_width=True):
+            st.session_state.current_view = "sota"
+    with col4:
+        if st.button("📊 Datasets", use_container_width=True):
+            st.session_state.current_view = "datasets"
+    with col5:
+        if st.button("🧩 Methods", use_container_width=True):
+            st.session_state.current_view = "methods"
+
+st.divider()
+
+# --- 3. View Logic ---
+
+if st.session_state.current_view == "trending":
+    st.subheader("Trending Papers")
+    
+    # Pagination UI
+    PAGE_SIZE = 10
+    c1, c2, c3 = st.columns([1, 2, 1])
+    
+    with c1:
+        if st.button("⬅️ Previous", disabled=(st.session_state.page_number == 0)):
+            st.session_state.page_number -= 1
+            st.rerun() # Refresh to update data
+    with c2:
+        st.write(f"Center: Page {st.session_state.page_number + 1}")
+    with c3:
+        if st.button("Next ➡️"):
+            st.session_state.page_number += 1
+            st.rerun()
+
+    # Data Fetching
+    offset = st.session_state.page_number * PAGE_SIZE
+    with st.spinner("Loading papers..."):
+        dataset = load_dataset("pwc-archive/papers-with-abstracts", split="train", streaming=True)
+        papers = list(dataset.skip(offset).take(PAGE_SIZE))
+
+    # Display Papers
+    for p in papers:
+        with st.container():
+            st.markdown(f"### {p['title']}")
+            st.write(p['abstract'][:500] + "...")
+            if p.get('arxiv_id'):
+                st.link_button("View on ArXiv", f"https://arxiv.org/abs/{p['arxiv_id']}")
+            st.divider()
+
+elif st.session_state.current_view == "sota":
+    st.title("🏆 State-of-the-Art")
+    st.info("Leaderboards coming soon! This page will show rankings for different ML tasks.")
+    if st.button("Back to Trending"):
+        st.session_state.current_view = "trending"
+        st.rerun()
+
+else:
+    st.title("Under Construction")
+    st.write(f"The **{st.session_state.current_view}** section is currently being indexed.")