Update config.py

config.py

@@ -13,7 +13,7 @@ API_BASE_URL = "http://localhost:8501"  # Direct URL for local processing
 
 # News Scraping Settings
 ARTICLES_PER_SOURCE = int(os.getenv("ARTICLES_PER_SOURCE", "10"))
-USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+USER_AGENT = os.getenv("USER_AGENT", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
 
 # RSS Feed Settings
 RSS_FEEDS = {
@@ -45,14 +45,29 @@ SENTIMENT_CATEGORIES = {
     "news_tone": ["positive", "negative", "neutral"]
 }
 
+# Sentiment Analysis Settings
+SENTIMENT_THRESHOLD = float(os.getenv("SENTIMENT_THRESHOLD", "0.2"))
+CONFIDENCE_THRESHOLD = float(os.getenv("CONFIDENCE_THRESHOLD", "0.6"))
+ENSEMBLE_AGREEMENT_THRESHOLD = float(os.getenv("ENSEMBLE_AGREEMENT_THRESHOLD", "0.7"))
+
+# Entity Recognition Settings
+ENTITY_TYPES = ["ORG", "PERSON", "GPE", "MONEY", "DATE", "TIME", "PERCENT", "QUANTITY"]
+ENTITY_CONFIDENCE_THRESHOLD = float(os.getenv("ENTITY_CONFIDENCE_THRESHOLD", "0.5"))
+
+# Sentiment Target Settings
+TARGET_CONTEXT_WINDOW = int(os.getenv("TARGET_CONTEXT_WINDOW", "100"))  # characters
+TARGET_CONFIDENCE_THRESHOLD = float(os.getenv("TARGET_CONFIDENCE_THRESHOLD", "0.5"))
+
 # Cache Settings
-CACHE_DIR = ".cache"
-CACHE_EXPIRY = 3600  # 1 hour
-CACHE_DURATION = 300  # 5 minutes in seconds
+CACHE_DIR = os.getenv("CACHE_DIR", ".cache")
+CACHE_EXPIRY = int(os.getenv("CACHE_EXPIRY", "3600"))  # 1 hour
+CACHE_DURATION = int(os.getenv("CACHE_DURATION", "300"))  # 5 minutes in seconds
 
 # Audio Settings
-AUDIO_OUTPUT_DIR = "audio_output"
-DEFAULT_LANG = "hi"  # Hindi
+AUDIO_OUTPUT_DIR = os.getenv("AUDIO_OUTPUT_DIR", "audio_output")
+DEFAULT_LANG = os.getenv("DEFAULT_LANG", "hi")  # Hindi
+AUDIO_SPEED = float(os.getenv("AUDIO_SPEED", "1.0"))
+AUDIO_VOLUME = float(os.getenv("AUDIO_VOLUME", "1.0"))
 
 # News Sources
 NEWS_SOURCES = {
@@ -83,3 +98,19 @@ HEADERS = {
     "Accept-Language": "en-US,en;q=0.5",
     "Connection": "keep-alive"
 }
+
+# Visualization Settings
+CHART_COLORS = {
+    "positive": "#28a745",
+    "negative": "#dc3545",
+    "neutral": "#6c757d",
+    "financial": "#17a2b8",
+    "emotional": "#ffc107",
+    "esg": "#20c997"
+}
+
+CHART_DIMENSIONS = {
+    "width": 600,
+    "height": 300,
+    "margin": 20
+}