Update app.py

app.py

@@ -593,7 +593,13 @@ with final_recs:
     recs['Date'] = recs['Date'].dt.date
     rec_dates = np.sort(list(recs['Date'].unique()))
     pred_date = st.selectbox("Pick the Test Date", rec_dates)
-    
+    fp = {}
+    for index, d in enumerate(rec_dates[:-1]):
+        fr = recs[recs['Date'] == rec_dates[index+1]]
+        following_price = fr['Price'][0]
+        fp[d] = following_price
+
+    fp[rec_dates[-1]] = 'Not Available'
     # date_value =  pd.to_datetime('2021-08-12')
     # pred_date = st.date_input("Pick Test Date", value=date_value, min_value=date_value, max_value=date_limit)
     role = st.radio(
@@ -627,7 +633,10 @@ with final_recs:
     # st.subheader("GenAI Recommendations")
     trade_recs.reset_index(inplace=True, drop=True)
     genrec_container = st.container(border=False)
-    genrec_container.write(f'<span style="font-size: large;"><b>Date:</b> <u>{pred_date}</u></span>', unsafe_allow_html=True)
+    rec_col1, rec_col2, rec_col3 = genrec_container.columns(3, gap='medium')
+    rec_col1.write(f'<span style="font-size: large;"><b>Current Date:</b> <u>{pred_date}</u></span>', unsafe_allow_html=True)
+    rec_col2.write(f'<span style="font-size: large;"><b>Current Close Price:</b> {trade_recs['Price'][0]}</span>', unsafe_allow_html=True)
+    rec_col3.write(f'<span style="font-size: large;"><b>Following Close Price:</b> {fp[pred_date]}</span>', unsafe_allow_html=True)
         
     # genrec = 'Recommendation for following day will go here.'
     genrec_container.write(trade_recs['Recommendations'][0])