Update app.py

app.py

@@ -12,13 +12,24 @@ url_timeGpt =  "https://ofirmatzlawi-fin-proj-docker-1.hf.space/ticker/"
 
 
 def get_max_sentiment(row):
-    max_value = max(row['neg'], row['neu'], row['pos'])
-    if max_value == row['neg']:
-        return 'neg'
-    elif max_value == row['neu']:
-        return 'neu'
+    if row["sentiment_score"] > 0.05:  # Threshold for positive sentiment
+        return "positive"
+    elif row["sentiment_score"] < -0.05:  # Threshold for negative sentiment
+        return "negative"
     else:
-        return 'pos'
+        return "neutral"
+
+
+
+
+    
+    #max_value = max(row['neg'], row['neu'], row['pos'])
+    #if max_value == row['neg']:
+    #    return 'neg'
+    #elif max_value == row['neu']:
+    #    return 'neu'
+    #else:
+    #    return 'pos'
 
 
 def get_sentiment_data(stock_info):