joaco7172 commited on
Commit
b6b2e4f
·
verified ·
1 Parent(s): d360824

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -136,15 +136,15 @@ def get_company_prompt(symbol):
136
 
137
 
138
  def get_prompt_by_row(symbol, row):
139
-
140
  start_date = row['Start Date'] if isinstance(row['Start Date'], str) else row['Start Date'].strftime('%Y-%m-%d')
141
  end_date = row['End Date'] if isinstance(row['End Date'], str) else row['End Date'].strftime('%Y-%m-%d')
142
  term = 'increased' if row['End Price'] > row['Start Price'] else 'decreased'
143
  head = "From {} to {}, {}'s stock price {} from {:.2f} to {:.2f}. Company news during this period are listed below:\n\n".format(
144
  start_date, end_date, symbol, term, row['Start Price'], row['End Price'])
145
 
146
- news = json.loads(row["News"])
147
- news = ["[Headline]: {}\n[Summary]: {}\n".format(
148
  n['headline'], n['summary']) for n in news if n['date'][:8] <= end_date.replace('-', '') and \
149
  not n['summary'].startswith("Looking for stock market analysis and research with proves results?")]
150
 
@@ -205,8 +205,8 @@ def get_all_prompts_online(symbol, data, curday, with_basics=True):
205
  for i in range(-len(prev_rows), 0):
206
  prompt += "\n" + prev_rows[i][0]
207
  latest_news_items = latest_news(
208
- json.loads(prev_rows[i][1]),
209
- min(5, len(json.loads(prev_rows[i][1])))
210
  )
211
  if latest_news_items:
212
  prompt += "\n".join(latest_news_items)
@@ -229,7 +229,6 @@ def get_all_prompts_online(symbol, data, curday, with_basics=True):
229
  return info, prompt
230
 
231
 
232
-
233
  def construct_prompt(ticker, curday, n_weeks, use_basics):
234
 
235
  try:
 
136
 
137
 
138
  def get_prompt_by_row(symbol, row):
139
+
140
  start_date = row['Start Date'] if isinstance(row['Start Date'], str) else row['Start Date'].strftime('%Y-%m-%d')
141
  end_date = row['End Date'] if isinstance(row['End Date'], str) else row['End Date'].strftime('%Y-%m-%d')
142
  term = 'increased' if row['End Price'] > row['Start Price'] else 'decreased'
143
  head = "From {} to {}, {}'s stock price {} from {:.2f} to {:.2f}. Company news during this period are listed below:\n\n".format(
144
  start_date, end_date, symbol, term, row['Start Price'], row['End Price'])
145
 
146
+ news = row["News"] if isinstance(row["News"], list) else json.loads(row["News"])
147
+ news_formatted = ["[Headline]: {}\n[Summary]: {}\n".format(
148
  n['headline'], n['summary']) for n in news if n['date'][:8] <= end_date.replace('-', '') and \
149
  not n['summary'].startswith("Looking for stock market analysis and research with proves results?")]
150
 
 
205
  for i in range(-len(prev_rows), 0):
206
  prompt += "\n" + prev_rows[i][0]
207
  latest_news_items = latest_news(
208
+ prev_rows[i][1],
209
+ min(5, len(prev_rows[i][1]))
210
  )
211
  if latest_news_items:
212
  prompt += "\n".join(latest_news_items)
 
229
  return info, prompt
230
 
231
 
 
232
  def construct_prompt(ticker, curday, n_weeks, use_basics):
233
 
234
  try: