proKBD commited on
Commit
b49f52e
·
verified ·
1 Parent(s): 03d5192

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +37 -6
config.py CHANGED
@@ -13,7 +13,7 @@ API_BASE_URL = "http://localhost:8501" # Direct URL for local processing
13
 
14
  # News Scraping Settings
15
  ARTICLES_PER_SOURCE = int(os.getenv("ARTICLES_PER_SOURCE", "10"))
16
- 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"
17
 
18
  # RSS Feed Settings
19
  RSS_FEEDS = {
@@ -45,14 +45,29 @@ SENTIMENT_CATEGORIES = {
45
  "news_tone": ["positive", "negative", "neutral"]
46
  }
47
 
 
 
 
 
 
 
 
 
 
 
 
 
 
48
  # Cache Settings
49
- CACHE_DIR = ".cache"
50
- CACHE_EXPIRY = 3600 # 1 hour
51
- CACHE_DURATION = 300 # 5 minutes in seconds
52
 
53
  # Audio Settings
54
- AUDIO_OUTPUT_DIR = "audio_output"
55
- DEFAULT_LANG = "hi" # Hindi
 
 
56
 
57
  # News Sources
58
  NEWS_SOURCES = {
@@ -83,3 +98,19 @@ HEADERS = {
83
  "Accept-Language": "en-US,en;q=0.5",
84
  "Connection": "keep-alive"
85
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
 
14
  # News Scraping Settings
15
  ARTICLES_PER_SOURCE = int(os.getenv("ARTICLES_PER_SOURCE", "10"))
16
+ 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")
17
 
18
  # RSS Feed Settings
19
  RSS_FEEDS = {
 
45
  "news_tone": ["positive", "negative", "neutral"]
46
  }
47
 
48
+ # Sentiment Analysis Settings
49
+ SENTIMENT_THRESHOLD = float(os.getenv("SENTIMENT_THRESHOLD", "0.2"))
50
+ CONFIDENCE_THRESHOLD = float(os.getenv("CONFIDENCE_THRESHOLD", "0.6"))
51
+ ENSEMBLE_AGREEMENT_THRESHOLD = float(os.getenv("ENSEMBLE_AGREEMENT_THRESHOLD", "0.7"))
52
+
53
+ # Entity Recognition Settings
54
+ ENTITY_TYPES = ["ORG", "PERSON", "GPE", "MONEY", "DATE", "TIME", "PERCENT", "QUANTITY"]
55
+ ENTITY_CONFIDENCE_THRESHOLD = float(os.getenv("ENTITY_CONFIDENCE_THRESHOLD", "0.5"))
56
+
57
+ # Sentiment Target Settings
58
+ TARGET_CONTEXT_WINDOW = int(os.getenv("TARGET_CONTEXT_WINDOW", "100")) # characters
59
+ TARGET_CONFIDENCE_THRESHOLD = float(os.getenv("TARGET_CONFIDENCE_THRESHOLD", "0.5"))
60
+
61
  # Cache Settings
62
+ CACHE_DIR = os.getenv("CACHE_DIR", ".cache")
63
+ CACHE_EXPIRY = int(os.getenv("CACHE_EXPIRY", "3600")) # 1 hour
64
+ CACHE_DURATION = int(os.getenv("CACHE_DURATION", "300")) # 5 minutes in seconds
65
 
66
  # Audio Settings
67
+ AUDIO_OUTPUT_DIR = os.getenv("AUDIO_OUTPUT_DIR", "audio_output")
68
+ DEFAULT_LANG = os.getenv("DEFAULT_LANG", "hi") # Hindi
69
+ AUDIO_SPEED = float(os.getenv("AUDIO_SPEED", "1.0"))
70
+ AUDIO_VOLUME = float(os.getenv("AUDIO_VOLUME", "1.0"))
71
 
72
  # News Sources
73
  NEWS_SOURCES = {
 
98
  "Accept-Language": "en-US,en;q=0.5",
99
  "Connection": "keep-alive"
100
  }
101
+
102
+ # Visualization Settings
103
+ CHART_COLORS = {
104
+ "positive": "#28a745",
105
+ "negative": "#dc3545",
106
+ "neutral": "#6c757d",
107
+ "financial": "#17a2b8",
108
+ "emotional": "#ffc107",
109
+ "esg": "#20c997"
110
+ }
111
+
112
+ CHART_DIMENSIONS = {
113
+ "width": 600,
114
+ "height": 300,
115
+ "margin": 20
116
+ }