Update src/streamlit_app.py

src/streamlit_app.py

@@ -191,7 +191,7 @@ def create_content_context(content_list):
     movies = sum(1 for c in content_list if c.get('type') == 'Movie')
     shows = sum(1 for c in content_list if c.get('type') == 'TV Show')
     
-    context = f"""Currently showing {total} Netflix titles ({movies} movies and {shows} TV shows). """
+    context = f"""Currently showing {total} titles ({movies} movies and {shows} TV shows). """
     
     # Add some genre info
     all_genres = []
@@ -214,7 +214,7 @@ def create_content_context(content_list):
 def get_ai_response(client, question, context, model_name):
     """Get response from Together AI"""
     try:
-        prompt = f"""You are a helpful Netflix content expert. Based on the current content data, please answer the user's question accurately and helpfully.
+        prompt = f"""You are a helpful streaming content expert. Based on the current content data, please answer the user's question accurately and helpfully.
 
 Current Content Data Context:
 {context}
@@ -226,7 +226,7 @@ Please provide a helpful, accurate response based on the data provided. Keep you
         response = client.chat.completions.create(
             model=model_name,
             messages=[
-                {"role": "system", "content": "You are a helpful Netflix content expert with deep knowledge of movies and TV shows."},
+                {"role": "system", "content": "You are a helpful content expert with deep knowledge of movies and TV shows."},
                 {"role": "user", "content": prompt}
             ],
             max_tokens=300,
@@ -519,9 +519,49 @@ def main():
             if filters['cast']:
                 st.write(f"- Cast: {filters['cast']}")
             st.write("---")
+
+            # Pagination controls
+            items_per_page = 10
+            total_pages = (filtered_count + items_per_page - 1) // items_per_page
+
+            # Initialize page number in session state if not exists
+            if 'current_page' not in st.session_state:
+                st.session_state.current_page = 1
+
+            # Page navigation
+            col_prev, col_page, col_next = st.columns([1, 2, 1])
+            
+            with col_prev:
+                if st.button("← Previous", disabled=st.session_state.current_page == 1):
+                    st.session_state.current_page -= 1
+                    st.rerun()
+            
+            with col_page:
+                page_input = st.number_input(
+                    f"Page (of {total_pages})",
+                    min_value=1,
+                    max_value=total_pages,
+                    value=st.session_state.current_page,
+                    key="page_number"
+                )
+                if page_input != st.session_state.current_page:
+                    st.session_state.current_page = page_input
+                    st.rerun()
+            
+            with col_next:
+                if st.button("Next →", disabled=st.session_state.current_page == total_pages):
+                    st.session_state.current_page += 1
+                    st.rerun()
+
+            # Calculate slice indices for current page
+            start_idx = (st.session_state.current_page - 1) * items_per_page
+            end_idx = min(start_idx + items_per_page, filtered_count)
+            
+            # Display current page info
+            st.write(f"Showing {start_idx + 1}-{end_idx} of {filtered_count:,} titles")
             
-            # Show first 10 items
-            for i, content in enumerate(st.session_state.filtered_content[:10]):
+            # Show items for current page
+            for i, content in enumerate(st.session_state.filtered_content[start_idx:end_idx], start=start_idx):
                 with st.expander(f"{content.get('title', 'N/A')} ({content.get('release_year', 'N/A')})"):
                     # Content details in columns
                     detail_col1, detail_col2 = st.columns(2)
@@ -558,8 +598,22 @@ def main():
                             else:
                                 st.info("No similar content found.")
             
-            if filtered_count > 10:
-                st.info(f"Showing first 10 of {filtered_count:,} titles. Adjust filters to narrow results.")
+            # Bottom pagination controls
+            st.write("---")
+            col_prev2, col_page2, col_next2 = st.columns([1, 2, 1])
+            
+            with col_prev2:
+                if st.button("← Previous", disabled=st.session_state.current_page == 1, key="prev_bottom"):
+                    st.session_state.current_page -= 1
+                    st.rerun()
+            
+            with col_page2:
+                st.write(f"Page {st.session_state.current_page} of {total_pages}")
+            
+            with col_next2:
+                if st.button("Next →", disabled=st.session_state.current_page == total_pages, key="next_bottom"):
+                    st.session_state.current_page += 1
+                    st.rerun()
         else:
             st.info("No content matches your current filters. Try adjusting the criteria.")