Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,28 @@
 import streamlit as st
 import requests
 from together import Together
-import os
 import json
 
 # =============================================================================
 # CONFIGURATION - Using Secrets Management
 # =============================================================================
-NOCODB_URL = "https://mtoft20-potm.hf.space".strip()  # Base URL, ensure no extra spaces
+NOCODB_URL = "https://mtoft20-potm.hf.space".strip()  # Base URL
 
-# Get sensitive data from Streamlit secrets or environment variables
 def get_api_credentials():
-    """Get API credentials from secrets or environment"""
+    """Get API credentials from secrets"""
     try:
-        # Try Streamlit secrets first (for Hugging Face Spaces)
-        api_token = st.secrets.get("NOCODB_API_TOKEN", os.environ.get("NOCODB_API_TOKEN", "")).strip()
-        together_key = st.secrets.get("TOGETHER_API_KEY", os.environ.get("TOGETHER_API_KEY", "")).strip()
-        
-        # Get endpoints for content and similarities
-        content_endpoint = st.secrets.get("NOCODB_CONTENT_ENDPOINT", os.environ.get("NOCODB_CONTENT_ENDPOINT", "")).strip()
-        similarity_endpoints = [endpoint.strip() for endpoint in 
-                             st.secrets.get("NOCODB_SIMILARITY_ENDPOINTS", 
-                                          os.environ.get("NOCODB_SIMILARITY_ENDPOINTS", "")).split(",")
-                             if endpoint.strip()]
-        
-        return api_token, together_key, content_endpoint, similarity_endpoints
-    except:
-        # Fallback to environment variables
-        api_token = os.environ.get("NOCODB_API_TOKEN", "").strip()
-        together_key = os.environ.get("TOGETHER_API_KEY", "").strip()
-        content_endpoint = os.environ.get("NOCODB_CONTENT_ENDPOINT", "").strip()
+        # Get credentials from Streamlit secrets
+        api_token = st.secrets.get("NOCODB_API_TOKEN", "").strip()
+        together_key = st.secrets.get("TOGETHER_API_KEY", "").strip()
+        content_endpoint = st.secrets.get("NOCODB_CONTENT_ENDPOINT", "").strip()
         similarity_endpoints = [endpoint.strip() for endpoint in 
-                             os.environ.get("NOCODB_SIMILARITY_ENDPOINTS", "").split(",")
+                             st.secrets.get("NOCODB_SIMILARITY_ENDPOINTS", "").split(",")
                              if endpoint.strip()]
         
         return api_token, together_key, content_endpoint, similarity_endpoints
+    except Exception as e:
+        st.error(f"Error loading credentials: {str(e)}")
+        return "", "", "", []
 
 # Initialize Together AI client
 @st.cache_resource