Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,29 +1,40 @@
 import streamlit as st
 import requests
 from together import Together
-import json
 import os
+import json
 
 # =============================================================================
 # CONFIGURATION - Using Secrets Management
 # =============================================================================
-NOCODB_URL = "https://mtoft20-potm.hf.space".strip()  # Base URL
+NOCODB_URL = "https://mtoft20-potm.hf.space".strip()  # Base URL, ensure no extra spaces
 
+# Get sensitive data from Streamlit secrets or environment variables
 def get_api_credentials():
-    """Get API credentials from secrets"""
+    """Get API credentials from secrets or environment"""
     try:
-        # 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()
+        # 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", "").split(",")
+                             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()
+        similarity_endpoints = [endpoint.strip() for endpoint in 
+                             os.environ.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