update

src/streamlit_app.py

@@ -1,42 +1,54 @@
 import streamlit as st
+import requests
 from datasets import load_dataset
 
-st.set_page_config(page_title="Paper Explorer", layout="wide")
+st.set_page_config(page_title="PWC Explorer", layout="wide")
 
 # 1. Initialize Session States
 if 'current_view' not in st.session_state:
-    st.session_state.current_view = "trending"  # Default view
+    st.session_state.current_view = "trending"
 if 'page_number' not in st.session_state:
     st.session_state.page_number = 0
 
-# --- 2. Custom Top Navigation ---
+# --- 2. Top Navigation Bar ---
 nav_container = st.container()
 with nav_container:
-    col1, col2, col3, col4, col5 = st.columns([1, 1.5, 2, 1, 1])
+    # We added a 6th column to balance the layout
+    col1, col2, col3, col4, col5 = st.columns([1, 1, 1, 1.5, 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"
+            st.session_state.page_number = 0 # Reset page when switching
+            st.rerun()
     with col3:
+        if st.button("🔍 Search", use_container_width=True):
+            st.session_state.current_view = "search"
+            st.session_state.page_number = 0
+            st.rerun()
+    with col4:
         if st.button("🏆 Browse SOTA", use_container_width=True):
             st.session_state.current_view = "sota"
-    with col4:
+            st.rerun()
+    with col5:
         if st.button("📊 Datasets", use_container_width=True):
             st.session_state.current_view = "datasets"
-    with col5:
+            st.rerun()
+    with col6:
         if st.button("🧩 Methods", use_container_width=True):
             st.session_state.current_view = "methods"
+            st.rerun()
 
 st.divider()
 
 # --- 3. View Logic ---
 
+# --- VIEW: TRENDING ---
 if st.session_state.current_view == "trending":
-    st.subheader("Trending Papers")
-    
+    st.subheader("🔥 Trending Research Papers")
+
     # Pagination UI
     PAGE_SIZE = 10
     c1, c2, c3 = st.columns([1, 2, 1])
@@ -66,7 +78,40 @@ if st.session_state.current_view == "trending":
             if p.get('arxiv_id'):
                 st.link_button("View on ArXiv", f"https://arxiv.org/abs/{p['arxiv_id']}")
             st.divider()
+            
+# --- VIEW: SEARCH (The New Section) ---
+elif st.session_state.current_view == "search":
+    st.subheader("🔍 Global Archive Search")
+    st.info("Search through the entire 1GB archive of 500,000+ papers.")
+    
+    query = st.text_input("Enter keywords (e.g., 'Diffusion Models', 'LLM')", key="global_search")
+    
+    if query:
+        PAGE_SIZE = 10
+        offset = st.session_state.page_number * PAGE_SIZE
+        
+        search_url = f"https://datasets-server.huggingface.co/search?dataset=pwc-archive/papers-with-abstracts&split=train&query={query}&offset={offset}&length={PAGE_SIZE}"
+        
+        with st.spinner("Searching..."):
+            response = requests.get(search_url)
+            if response.status_code == 200:
+                data = response.json()
+                papers = [item['row'] for item in data['rows']]
+                
+                if not papers:
+                    st.warning("No matches found.")
+                
+                for p in papers:
+                    with st.expander(p['title']):
+                        st.write(p['abstract'])
+                        if p.get('arxiv_id'):
+                            st.link_button("ArXiv", f"https://arxiv.org/abs/{p['arxiv_id']}")
+            else:
+                st.error("The search service is currently unavailable.")
+    else:
+        st.write("Type something above to begin your search.")
 
+# --- VIEW: SOTA ---
 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.")