joaco7172 commited on
Commit
0be50a9
Β·
verified Β·
1 Parent(s): f3072a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -135,14 +135,13 @@ def get_company_prompt(symbol):
135
 
136
 
137
  def get_prompt_by_row(symbol, row):
138
-
139
  start_date = row['Start Date'] if isinstance(row['Start Date'], str) else row['Start Date'].strftime('%Y-%m-%d')
140
  end_date = row['End Date'] if isinstance(row['End Date'], str) else row['End Date'].strftime('%Y-%m-%d')
141
  term = 'increased' if row['End Price'] > row['Start Price'] else 'decreased'
142
  head = "From {} to {}, {}'s stock price {} from {:.2f} to {:.2f}. Company news during this period are listed below:\n\n".format(
143
  start_date, end_date, symbol, term, row['Start Price'], row['End Price'])
144
 
145
- news = json.loads(row["News"])
146
  news = ["[Headline]: {}\n[Summary]: {}\n".format(
147
  n['headline'], n['summary']) for n in news if n['date'][:8] <= end_date.replace('-', '') and \
148
  not n['summary'].startswith("Looking for stock market analysis and research with proves results?")]
@@ -156,7 +155,6 @@ def get_prompt_by_row(symbol, row):
156
 
157
  return head, news, basics
158
 
159
-
160
  def sample_news(news, k=5):
161
 
162
  return [news[i] for i in sorted(random.sample(range(len(news)), k))]
@@ -200,7 +198,6 @@ def get_all_prompts_online(symbol, data, curday, with_basics=True):
200
 
201
  for row_idx, row in data.iterrows():
202
  head, news, _ = get_prompt_by_row(symbol, row)
203
- news = json.loads(news) # Ensure news is parsed as list of dictionaries
204
  prev_rows.append((head, news, None))
205
 
206
  prompt = ""
@@ -230,7 +227,6 @@ def get_all_prompts_online(symbol, data, curday, with_basics=True):
230
 
231
  return info, prompt
232
 
233
-
234
  def construct_prompt(ticker, curday, n_weeks, use_basics):
235
 
236
  try:
 
135
 
136
 
137
  def get_prompt_by_row(symbol, row):
 
138
  start_date = row['Start Date'] if isinstance(row['Start Date'], str) else row['Start Date'].strftime('%Y-%m-%d')
139
  end_date = row['End Date'] if isinstance(row['End Date'], str) else row['End Date'].strftime('%Y-%m-%d')
140
  term = 'increased' if row['End Price'] > row['Start Price'] else 'decreased'
141
  head = "From {} to {}, {}'s stock price {} from {:.2f} to {:.2f}. Company news during this period are listed below:\n\n".format(
142
  start_date, end_date, symbol, term, row['Start Price'], row['End Price'])
143
 
144
+ news = row["News"]
145
  news = ["[Headline]: {}\n[Summary]: {}\n".format(
146
  n['headline'], n['summary']) for n in news if n['date'][:8] <= end_date.replace('-', '') and \
147
  not n['summary'].startswith("Looking for stock market analysis and research with proves results?")]
 
155
 
156
  return head, news, basics
157
 
 
158
  def sample_news(news, k=5):
159
 
160
  return [news[i] for i in sorted(random.sample(range(len(news)), k))]
 
198
 
199
  for row_idx, row in data.iterrows():
200
  head, news, _ = get_prompt_by_row(symbol, row)
 
201
  prev_rows.append((head, news, None))
202
 
203
  prompt = ""
 
227
 
228
  return info, prompt
229
 
 
230
  def construct_prompt(ticker, curday, n_weeks, use_basics):
231
 
232
  try: