Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,29 +8,37 @@ import json
 # =============================================================================
 # CONFIGURATION - Using Secrets Management
 # =============================================================================
-NOCODB_URL = "https://mtoft20-potm.hf.space/api/v1/db/data/noco/p9pozkcw81t9aee/mvtt3arw5ni7uqp"  # Updated with table ID
 
-# Get sensitive data from Streamlit secrets or environment variables
 def get_api_credentials():
-    """Get API credentials from secrets or environment"""
+    """Get API credentials and URLs from secrets or environment"""
     try:
         # 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", ""))
         
-        return api_token, together_key
-    except:
-        # Fallback to environment variables
-        api_token = os.environ.get("NOCODB_API_TOKEN", "")
-        together_key = os.environ.get("TOGETHER_API_KEY", "")
+        # Get NocoDB URLs from secrets or environment
+        base_url = st.secrets.get("NOCODB_BASE_URL", os.environ.get("NOCODB_BASE_URL", ""))
+        content_table_id = st.secrets.get("NOCODB_CONTENT_TABLE_ID", os.environ.get("NOCODB_CONTENT_TABLE_ID", ""))
+        similarity_table_ids = st.secrets.get("NOCODB_SIMILARITY_TABLE_IDS", os.environ.get("NOCODB_SIMILARITY_TABLE_IDS", "")).split(",")
         
-        return api_token, together_key
+        if not base_url or not content_table_id:
+            st.error("NocoDB URLs not configured properly in secrets.")
+            return None, None, None, None, None
+        
+        # Construct full URLs
+        content_url = f"{base_url}/{content_table_id}"
+        similarity_urls = [f"{base_url}/{table_id}" for table_id in similarity_table_ids if table_id]
+        
+        return api_token, together_key, content_url, similarity_urls
+    except Exception as e:
+        st.error(f"Error loading credentials: {str(e)}")
+        return None, None, None, None
 
 # Initialize Together AI client
 @st.cache_resource
 def get_ai_client():
     """Initialize Together AI client"""
-    _, together_key = get_api_credentials()
+    _, together_key, _, _ = get_api_credentials()
     if not together_key:
         st.error("Together AI API key not found. Please configure it in the secrets.")
         return None
@@ -42,9 +50,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, _ = get_api_credentials()
+    api_token, _, content_url, _ = get_api_credentials()
     
-    if not api_token:
+    if not api_token or not content_url:
         st.error("NocoDB credentials not configured. Please set up your secrets.")
         return []
     
@@ -60,7 +68,7 @@ def get_streaming_content():
     try:
         while True:
             offset = (page - 1) * page_size
-            url = f"{NOCODB_URL}?limit={page_size}&offset={offset}"
+            url = f"{content_url}?limit={page_size}&offset={offset}"
             
             response = requests.get(url, headers=headers)
             
@@ -251,8 +259,13 @@ def extract_unique_names(content_list, field):
 def get_similar_content(content, n_recommendations=5):
     """Get pre-computed similar content from database"""
     try:
-        # Get database credentials
-        api_token, _ = get_api_credentials()
+        # Get database credentials and URLs
+        api_token, _, content_url, similarity_urls = get_api_credentials()
+        
+        if not api_token or not similarity_urls:
+            st.error("NocoDB credentials or URLs not configured properly.")
+            return []
+            
         headers = {
             "xc-token": api_token,
             "accept": "application/json"
@@ -261,20 +274,13 @@ def get_similar_content(content, n_recommendations=5):
         title = content.get('title', '')
         show_id = content.get('show_id', '')
         
-        # URLs for similarity tables using table IDs
-        similarity_table_urls = [
-            "https://mtoft20-potm.hf.space/api/v1/db/data/noco/p9pozkcw81t9aee/mp7bnn9tzhojh7k",  # Part 1 similarities
-            "https://mtoft20-potm.hf.space/api/v1/db/data/noco/p9pozkcw81t9aee/m8e5rglns4acmef",  # Part 2 similarities
-            "https://mtoft20-potm.hf.space/api/v1/db/data/noco/p9pozkcw81t9aee/m2driodimid10k6"   # Part 3 similarities
-        ]
-        
         # Try finding by both show_id and title
         query = f'where=(show_id,eq,{show_id})~and(title,eq,{title})'
         params = {
             "where": query
         }
         
-        for table_url in similarity_table_urls:
+        for table_url in similarity_urls:
             try:
                 response = requests.get(table_url, headers=headers, params=params)
                 
@@ -288,14 +294,13 @@ def get_similar_content(content, n_recommendations=5):
                                 
                                 # Get full content details for each similar item
                                 similar_content = []
-                                for item in similar_items[:n_recommendations]:  # Only process top N items
-                                    # Query main content table for full details using show_id
+                                for item in similar_items[:n_recommendations]:
                                     show_id = item.get('show_id', '')
                                     query = f'where=(show_id,eq,{show_id})'
                                     content_params = {
                                         "where": query
                                     }
-                                    content_response = requests.get(NOCODB_URL, headers=headers, params=content_params)
+                                    content_response = requests.get(content_url, headers=headers, params=content_params)
                                     
                                     if content_response.status_code == 200:
                                         content_data = content_response.json()
@@ -304,7 +309,7 @@ def get_similar_content(content, n_recommendations=5):
                                             content_dict['similarity'] = f"{item['similarity']:.2%}"
                                             similar_content.append(content_dict)
                                 
-                                return similar_content[:n_recommendations]  # Return top N items
+                                return similar_content[:n_recommendations]
                             except Exception as parse_error:
                                 continue
             except Exception as e:
@@ -330,7 +335,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!")