Update src/streamlit_app.py

src/streamlit_app.py

@@ -17,16 +17,20 @@ def get_api_credentials():
         # Try Streamlit secrets first (for Hugging Face Spaces)
         api_token = st.secrets.get("NOCODB_API_TOKEN", os.environ.get("NOCODB_API_TOKEN", ""))
         together_key = st.secrets.get("TOGETHER_API_KEY", os.environ.get("TOGETHER_API_KEY", ""))
-        endpoint_path = st.secrets.get("NOCODB_ENDPOINT_PATH", os.environ.get("NOCODB_ENDPOINT_PATH", ""))
         
-        return api_token, together_key, endpoint_path
+        # Get endpoints for content and similarities
+        content_endpoint = st.secrets.get("NOCODB_CONTENT_ENDPOINT", os.environ.get("NOCODB_CONTENT_ENDPOINT", ""))
+        similarity_endpoints = st.secrets.get("NOCODB_SIMILARITY_ENDPOINTS", os.environ.get("NOCODB_SIMILARITY_ENDPOINTS", "")).split(",")
+        
+        return api_token, together_key, content_endpoint, similarity_endpoints
     except:
         # Fallback to environment variables
         api_token = os.environ.get("NOCODB_API_TOKEN", "")
         together_key = os.environ.get("TOGETHER_API_KEY", "")
-        endpoint_path = os.environ.get("NOCODB_ENDPOINT_PATH", "")
+        content_endpoint = os.environ.get("NOCODB_CONTENT_ENDPOINT", "")
+        similarity_endpoints = os.environ.get("NOCODB_SIMILARITY_ENDPOINTS", "").split(",")
         
-        return api_token, together_key, endpoint_path
+        return api_token, together_key, content_endpoint, similarity_endpoints
 
 # Initialize Together AI client
 @st.cache_resource
@@ -44,9 +48,9 @@ def get_ai_client():
 @st.cache_data(ttl=300)  # Cache for 5 minutes
 def get_streaming_content():
     """Fetch streaming content from NocoDB with pagination"""
-    api_token, _, endpoint_path = get_api_credentials()
+    api_token, _, content_endpoint, _ = get_api_credentials()
     
-    if not api_token or not endpoint_path:
+    if not api_token or not content_endpoint:
         st.error("NocoDB credentials not configured. Please set up your secrets.")
         return []
     
@@ -62,7 +66,7 @@ def get_streaming_content():
     try:
         while True:
             offset = (page - 1) * page_size
-            url = f"{NOCODB_URL}{endpoint_path}?limit={page_size}&offset={offset}"
+            url = f"{NOCODB_URL}{content_endpoint}?limit={page_size}&offset={offset}"
             
             response = requests.get(url, headers=headers)
             
@@ -251,9 +255,9 @@ def get_similar_content(content, n_recommendations=5):
     """Get pre-computed similar content from database"""
     try:
         # Get database credentials
-        api_token, _, endpoint_path = get_api_credentials()
+        api_token, _, content_endpoint, similarity_endpoints = get_api_credentials()
         
-        if not api_token or not endpoint_path:
+        if not api_token or not similarity_endpoints:
             st.error("NocoDB credentials not configured properly.")
             return []
             
@@ -271,16 +275,12 @@ def get_similar_content(content, n_recommendations=5):
             "where": query
         }
         
-        # Use the same endpoint path but with different table IDs
-        similarity_table_paths = [
-            endpoint_path.replace("mvtt3arw5ni7uqp", "mp7bnn9tzhojh7k"),
-            endpoint_path.replace("mvtt3arw5ni7uqp", "m8e5rglns4acmef"),
-            endpoint_path.replace("mvtt3arw5ni7uqp", "m2driodimid10k6")
-        ]
-        
-        for table_path in similarity_table_paths:
+        for endpoint in similarity_endpoints:
+            if not endpoint.strip():  # Skip empty endpoints
+                continue
+                
             try:
-                url = f"{NOCODB_URL}{table_path}"
+                url = f"{NOCODB_URL}{endpoint}"
                 response = requests.get(url, headers=headers, params=params)
                 
                 if response.status_code == 200:
@@ -298,7 +298,7 @@ def get_similar_content(content, n_recommendations=5):
                                     content_params = {
                                         "where": query
                                     }
-                                    content_url = f"{NOCODB_URL}{endpoint_path}"
+                                    content_url = f"{NOCODB_URL}{content_endpoint}"
                                     content_response = requests.get(content_url, headers=headers, params=content_params)
                                     
                                     if content_response.status_code == 200:
@@ -334,7 +334,7 @@ def main():
     st.write("*At your service! Allow me to curate the perfect streaming entertainment for you.*")
     
     # Check API credentials
-    api_token, together_key, _ = get_api_credentials()
+    api_token, together_key, _, _ = get_api_credentials()
     
     if not together_key:
         st.error("⚠️ Together AI API key not configured!")