Update db/connection.py

db/connection.py

@@ -21,33 +21,39 @@ def get_database_config():
         config_sources.append(('HuggingFace Secrets', hf_config))
         print("✅ Found HuggingFace database configuration")
     
-    # 2. Streamlit Secrets (For Streamlit Cloud)
+    # 2. Streamlit Secrets (For Streamlit Cloud) - FIXED: Better error handling
     try:
+        # Only import streamlit if we're actually in Streamlit environment
         import streamlit as st
-        if hasattr(st, 'secrets') and 'supabase' in st.secrets:
-            secrets = st.secrets["supabase"]
-            st_config = {
-                'user': secrets.get('user'),
-                'password': secrets.get('password'),
-                'host': secrets.get('host'),
-                'port': secrets.get('port', '6543'),  # Transaction Pooler port
-                'dbname': secrets.get('dbname')
-            }
-            if all(st_config.values()):
-                config_sources.append(('Streamlit Secrets', st_config))
-                print("✅ Found Streamlit database configuration")
-    except (ImportError, AttributeError, KeyError):
+        if hasattr(st, 'secrets') and st.secrets:
+            if 'supabase' in st.secrets:
+                secrets = st.secrets["supabase"]
+                st_config = {
+                    'user': secrets.get('user'),
+                    'password': secrets.get('password'),
+                    'host': secrets.get('host'),
+                    'port': secrets.get('port', '6543'),
+                    'dbname': secrets.get('dbname')
+                }
+                if all(st_config.values()):
+                    config_sources.append(('Streamlit Secrets', st_config))
+                    print("✅ Found Streamlit database configuration")
+    except Exception as e:
+        # Silently continue if not in Streamlit environment
         pass
     
     # 3. Local Development Environment Variables
-    from dotenv import load_dotenv
-    load_dotenv()  # Load .env file for local development
+    try:
+        from dotenv import load_dotenv
+        load_dotenv()  # Load .env file for local development
+    except ImportError:
+        pass  # dotenv not available
     
     local_config = {
         'user': os.getenv("DB_USER") or os.getenv("user"),
         'password': os.getenv("DB_PASSWORD") or os.getenv("password"),
         'host': os.getenv("DB_HOST") or os.getenv("host"),
-        'port': os.getenv("DB_PORT") or os.getenv("port", "6543"),  # Default to transaction pooler
+        'port': os.getenv("DB_PORT") or os.getenv("port", "6543"),
         'dbname': os.getenv("DB_NAME") or os.getenv("dbname")
     }
     
@@ -209,8 +215,12 @@ def get_database_status():
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
-# Initialize database on import
-init_db()
+# Initialize database on import with error handling
+try:
+    init_db()
+except Exception as e:
+    print(f"⚠️ Database initialization failed: {e}")
+    print("💡 App will run in database-less mode")
 
 # Export status
 database_status = get_database_status()