Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,37 +8,31 @@ import json
 # =============================================================================
 # CONFIGURATION - Using Secrets Management
 # =============================================================================
+NOCODB_URL = "https://mtoft20-potm.hf.space"  # Base URL
 
+# Get sensitive data from Streamlit secrets or environment variables
 def get_api_credentials():
-    """Get API credentials and URLs from secrets or environment"""
+    """Get API credentials 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", ""))
+        endpoint_path = st.secrets.get("NOCODB_ENDPOINT_PATH", os.environ.get("NOCODB_ENDPOINT_PATH", ""))
         
-        # 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, endpoint_path
+    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", "")
         
-        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
+        return api_token, together_key, endpoint_path
 
 # 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
@@ -50,9 +44,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, _, content_url, _ = get_api_credentials()
+    api_token, _, endpoint_path = get_api_credentials()
     
-    if not api_token or not content_url:
+    if not api_token or not endpoint_path:
         st.error("NocoDB credentials not configured. Please set up your secrets.")
         return []
     
@@ -68,7 +62,7 @@ def get_streaming_content():
     try:
         while True:
             offset = (page - 1) * page_size
-            url = f"{content_url}?limit={page_size}&offset={offset}"
+            url = f"{NOCODB_URL}{endpoint_path}?limit={page_size}&offset={offset}"
             
             response = requests.get(url, headers=headers)
             
@@ -76,9 +70,6 @@ def get_streaming_content():
                 data = response.json()
                 current_page_data = data.get('list', [])
                 
-                # Filter out None values and ensure all items are dictionaries
-                current_page_data = [item for item in current_page_data if item and isinstance(item, dict)]
-                
                 if not current_page_data:  # No more data to fetch
                     break
                     
@@ -259,11 +250,11 @@ 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 and URLs
-        api_token, _, content_url, similarity_urls = get_api_credentials()
+        # Get database credentials
+        api_token, _, endpoint_path = get_api_credentials()
         
-        if not api_token or not similarity_urls:
-            st.error("NocoDB credentials or URLs not configured properly.")
+        if not api_token or not endpoint_path:
+            st.error("NocoDB credentials not configured properly.")
             return []
             
         headers = {
@@ -280,15 +271,22 @@ def get_similar_content(content, n_recommendations=5):
             "where": query
         }
         
-        for table_url in similarity_urls:
+        # 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:
             try:
-                response = requests.get(table_url, headers=headers, params=params)
+                url = f"{NOCODB_URL}{table_path}"
+                response = requests.get(url, headers=headers, params=params)
                 
                 if response.status_code == 200:
                     data = response.json()
                     if data.get('list'):
                         for entry in data['list']:
-                            # Parse similar items
                             try:
                                 similar_items = json.loads(entry['similar_items'])
                                 
@@ -300,6 +298,7 @@ def get_similar_content(content, n_recommendations=5):
                                     content_params = {
                                         "where": query
                                     }
+                                    content_url = f"{NOCODB_URL}{endpoint_path}"
                                     content_response = requests.get(content_url, headers=headers, params=content_params)
                                     
                                     if content_response.status_code == 200:
@@ -335,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!")