Update src/streamlit_app.py

src/streamlit_app.py

@@ -43,8 +43,8 @@ def get_ai_client():
 # HELPER FUNCTIONS
 # =============================================================================
 @st.cache_data(ttl=300)  # Cache for 5 minutes
-def get_netflix_content():
-    """Fetch Netflix content from NocoDB with pagination"""
+def get_streaming_content():
+    """Fetch streaming content from NocoDB with pagination"""
     api_token, _ = get_api_credentials()
     
     if not api_token:
@@ -64,7 +64,7 @@ def get_netflix_content():
         while True:
             offset = (page - 1) * page_size
             response = requests.get(
-                f"{NOCODB_URL}/cleaned_netflix_titles_csv?limit={page_size}&offset={offset}",
+                f"{NOCODB_URL}/cleaned_combined_streaming?limit={page_size}&offset={offset}",
                 headers=headers
             )
             
@@ -94,14 +94,17 @@ def get_netflix_content():
         return []
 
 def filter_content(content_list, filters):
-    """Apply filters to Netflix content list"""
+    """Apply filters to streaming content list"""
     filtered = []
     
     for content in content_list:
         matches_all_filters = True
         
-        # For debugging
-        title = content.get('title', 'N/A')
+        # Streaming service filter
+        if filters['streaming_services']:
+            if content.get('streaming_service') not in filters['streaming_services']:
+                matches_all_filters = False
+                continue
         
         # Type filter - only apply if not "All"
         if filters['content_type']:
@@ -303,14 +306,14 @@ def find_similar_content(content_list, selected_content, n_clusters=20, n_recomm
 def main():
     # Page config
     st.set_page_config(
-        page_title="Netflix Content Explorer",
+        page_title="Streaming Content Explorer",
         page_icon="🎬",
         layout="wide"
     )
     
     # Header
-    st.title("🎬 Netflix Content Explorer")
-    st.write("Explore Netflix movies and TV shows with AI-powered insights!")
+    st.title("🎬 Streaming Content Explorer")
+    st.write("Explore movies and TV shows across multiple streaming platforms!")
     
     # Check API credentials
     api_token, together_key = get_api_credentials()
@@ -335,11 +338,11 @@ def main():
         st.stop()
     
     # Load all content first
-    with st.spinner("Loading Netflix content..."):
-        all_content = get_netflix_content()
+    with st.spinner("Loading streaming content..."):
+        all_content = get_streaming_content()
     
     if not all_content:
-        st.error("Could not load Netflix content. Please check your NocoDB connection.")
+        st.error("Could not load streaming content. Please check your NocoDB connection.")
         st.stop()
     
     # Extract unique values for filters
@@ -350,6 +353,7 @@ def main():
         for genre in c.get('listed_in', '').split(',')
         if genre.strip()
     )))
+    all_streaming_services = sorted(list(set([c.get('streaming_service') for c in all_content if c.get('streaming_service')])))
     
     # Extract unique directors and cast members
     all_directors = extract_unique_names(all_content, 'director')
@@ -359,7 +363,21 @@ def main():
     st.sidebar.header("🔍 Filter Content")
 
     with st.sidebar.form("filter_form"):
-        st.subheader("Required Filters")
+        st.subheader("Streaming Services")
+        
+        # Streaming service selection (required)
+        selected_services = st.multiselect(
+            "Select Your Streaming Services",
+            options=all_streaming_services,
+            default=all_streaming_services[:1],  # Default to first service
+            help="Select the streaming services you have access to",
+            key="streaming_services"
+        )
+        
+        if not selected_services:
+            st.warning("Please select at least one streaming service")
+        
+        st.subheader("Content Filters")
         
         content_type = st.selectbox(
             "Content Type",
@@ -435,6 +453,7 @@ def main():
     
     # Create filter dictionary
     filters = {
+        'streaming_services': selected_services,
         'content_type': content_type if content_type != "All" else None,
         'ratings': selected_ratings,
         'genres': selected_genres,
@@ -486,6 +505,7 @@ def main():
                     detail_col1, detail_col2 = st.columns(2)
                     
                     with detail_col1:
+                        st.write(f"**📺 Available on:** {content.get('streaming_service', 'N/A')}")
                         st.write(f"**🎭 Type:** {content.get('type', 'N/A')}")
                         st.write(f"**⭐ Rating:** {content.get('rating', 'N/A')}")
                         st.write(f"**⏱️ Duration:** {content.get('duration', 'N/A')}")
@@ -524,7 +544,7 @@ def main():
     with col2:
         # AI Chat Section
         st.subheader("🤖 Ask AI Assistant")
-        st.write("Ask questions about the Netflix content!")
+        st.write("Ask questions about the streaming content!")
         
         # Model selection for Together AI
         model_choice = st.selectbox(
@@ -600,21 +620,27 @@ def main():
         total_items = len(all_content)
         filtered_items = len(st.session_state.filtered_content)
         
-        stat_col1, stat_col2, stat_col3, stat_col4 = st.columns(4)
+        # Create columns for stats
+        stat_cols = st.columns(len(selected_services) + 3)
         
-        with stat_col1:
-            st.metric("Total Titles", total_items)
+        # Basic stats
+        with stat_cols[0]:
+            st.metric("Total Available", total_items)
         
-        with stat_col2:
+        with stat_cols[1]:
             st.metric("Filtered Results", filtered_items)
         
-        with stat_col3:
+        with stat_cols[2]:
             movies = sum(1 for c in st.session_state.filtered_content if c.get('type') == 'Movie')
-            st.metric("Movies", movies)
-        
-        with stat_col4:
             shows = sum(1 for c in st.session_state.filtered_content if c.get('type') == 'TV Show')
-            st.metric("TV Shows", shows)
+            st.metric("Movies / Shows", f"{movies} / {shows}")
+        
+        # Streaming service breakdown
+        for i, service in enumerate(selected_services, 3):
+            if i < len(stat_cols):
+                service_count = sum(1 for c in st.session_state.filtered_content if c.get('streaming_service') == service)
+                with stat_cols[i]:
+                    st.metric(service, service_count)
 
 if __name__ == "__main__":
     main()