Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
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:
|