Update app.py

app.py

@@ -101,7 +101,6 @@ def get_news(symbol, data):
     for end_date, row in data.iterrows():
         start_date = row['Start Date'].strftime('%Y-%m-%d')
         end_date = row['End Date'].strftime('%Y-%m-%d')
-#         print(symbol, ': ', start_date, ' - ', end_date)
         time.sleep(1) # control qpm
         weekly_news = finnhub_client.company_news(symbol, _from=start_date, to=end_date)
         if len(weekly_news) == 0:
@@ -114,7 +113,7 @@ def get_news(symbol, data):
             } for n in weekly_news
         ]
         weekly_news.sort(key=lambda x: x['date'])
-        news_list.append(json.dumps(weekly_news))
+        news_list.append(weekly_news)
     
     data['News'] = news_list
     
@@ -163,7 +162,7 @@ def sample_news(news, k=5):
     return [news[i] for i in sorted(random.sample(range(len(news)), k))]
 
 
-def latest_news(news, k=10):
+def latest_news(news, k=5):
     # Sort news by date in descending order and select the latest k items
     sorted_news = sorted(news, key=lambda x: x['date'], reverse=True)
     return sorted_news[:k]
@@ -201,17 +200,18 @@ def get_all_prompts_online(symbol, data, curday, with_basics=True):
 
     for row_idx, row in data.iterrows():
         head, news, _ = get_prompt_by_row(symbol, row)
+        news = json.loads(news)  # Ensure news is parsed as list of dictionaries
         prev_rows.append((head, news, None))
         
     prompt = ""
     for i in range(-len(prev_rows), 0):
         prompt += "\n" + prev_rows[i][0]
-        sampled_news = latest_news(
+        latest_news_items = latest_news(
             prev_rows[i][1],
             min(5, len(prev_rows[i][1]))
         )
-        if sampled_news:
-            prompt += "\n".join(sampled_news)
+        if latest_news_items:
+            prompt += "\n".join(latest_news_items)
         else:
             prompt += "No relative news reported."