Fix database connection handling for Docker build

add_secrets.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python3
+"""
+Add database credentials as secrets to Hugging Face Space
+"""
+
+from huggingface_hub import HfApi
+import os
+
+# Initialize HF API
+api = HfApi()
+
+# Database credentials from Supabase
+secrets = {
+    "DB_HOST": "fwwnrulrrlgtfzbfusno.supabase.co",
+    "DB_NAME": "postgres", 
+    "DB_USER": "postgres",
+    "DB_PASSWORD": "SentimentDB2024!@#",
+    "DB_PORT": "5432"
+}
+
+# Add secrets to the Space
+try:
+    api.add_space_secret(
+        repo_id="wintersoldier91/Antisentiment",
+        key="DB_HOST",
+        value=secrets["DB_HOST"]
+    )
+    print("✅ Added DB_HOST secret")
+    
+    api.add_space_secret(
+        repo_id="wintersoldier91/Antisentiment", 
+        key="DB_NAME",
+        value=secrets["DB_NAME"]
+    )
+    print("✅ Added DB_NAME secret")
+    
+    api.add_space_secret(
+        repo_id="wintersoldier91/Antisentiment",
+        key="DB_USER", 
+        value=secrets["DB_USER"]
+    )
+    print("✅ Added DB_USER secret")
+    
+    api.add_space_secret(
+        repo_id="wintersoldier91/Antisentiment",
+        key="DB_PASSWORD",
+        value=secrets["DB_PASSWORD"]
+    )
+    print("✅ Added DB_PASSWORD secret")
+    
+    api.add_space_secret(
+        repo_id="wintersoldier91/Antisentiment",
+        key="DB_PORT",
+        value=secrets["DB_PORT"]
+    )
+    print("✅ Added DB_PORT secret")
+    
+    print("\n🎉 All database secrets added successfully!")
+    print("Your Hugging Face Space now has access to the Supabase database.")
+    
+except Exception as e:
+    print(f"❌ Error adding secrets: {e}")
+    print("Please check your token permissions and try again.")

app.py

@@ -74,8 +74,7 @@ st.markdown(premium_cdn, unsafe_allow_html=True)
 # Database setup
 CACHE_MINUTES = 15  # Cache data for 15 minutes
 
-# Initialize database (tables created in Supabase)
-init_database()
+# Database initialization will be done lazily when needed
 
 # FinViz news fetching functions
 finviz_url = 'https://finviz.com/quote.ashx?t='

database.py

@@ -18,7 +18,11 @@ def get_db_connection():
         )
     except psycopg2.Error as e:
         print(f"Database connection error: {e}")
-        raise
+        # Return None instead of raising to allow graceful fallback
+        return None
+    except Exception as e:
+        print(f"Unexpected database error: {e}")
+        return None
 
 def init_database():
     """Initialize database tables (already done in Supabase)"""
@@ -27,6 +31,9 @@ def init_database():
 def get_historical_data(ticker, days=30):
     """Retrieve historical data from database"""
     conn = get_db_connection()
+    if conn is None:
+        return pd.DataFrame()
+    
     cursor = conn.cursor()
     
     ist = pytz.timezone('Asia/Kolkata')
@@ -59,6 +66,9 @@ def get_historical_data(ticker, days=30):
 def can_fetch_new_data(ticker):
     """Check if we can fetch new data based on rate limiting"""
     conn = get_db_connection()
+    if conn is None:
+        return True, "Database not available - fetching fresh data"
+    
     cursor = conn.cursor()
     
     cursor.execute('''
@@ -94,6 +104,9 @@ def can_fetch_new_data(ticker):
 def update_cache_metadata(ticker, article_count):
     """Update cache metadata after fetching"""
     conn = get_db_connection()
+    if conn is None:
+        return  # Skip if database not available
+    
     cursor = conn.cursor()
     ist = pytz.timezone('Asia/Kolkata')
     now = datetime.now(ist).isoformat()
@@ -114,6 +127,9 @@ def update_cache_metadata(ticker, article_count):
 def store_sentiment_data(ticker, parsed_and_scored_news):
     """Store sentiment data in PostgreSQL"""
     conn = get_db_connection()
+    if conn is None:
+        return  # Skip if database not available
+    
     cursor = conn.cursor()
     ist = pytz.timezone('Asia/Kolkata')
     fetched_at = datetime.now(ist).isoformat()