Update app.py

app.py

@@ -433,18 +433,23 @@ def main_ui():
         map_output = gr.HTML(label="Prediction Map")
 
         clear_btn = gr.Button("Clear Session")
-
+        point_counter = gr.Textbox(label="Current # of Points", interactive=False)
+        
         def on_add_point(date, time_inp, loc_text):
-            return add_point(date, time_inp, loc_text)
+            msg = add_point(date, time_inp, loc_text)
+            return msg, f"{len(SESSION_POINTS)}"
 
         def on_train(trace_str):
+            global SESSION_POINTS
             traces = [parse_trace_text(trace_str)]
+            SESSION_POINTS = traces[0] if traces and len(traces[0]) >= SEQ_LEN else []
             model = train_model_on_traces(traces)
             if model:
                 return "Training completed."
             else:
                 return "Training failed or insufficient data."
 
+
         def on_predict():
             if not SESSION_POINTS or len(SESSION_POINTS) < SEQ_LEN:
                 return "Not enough points for prediction.", ""
@@ -457,10 +462,10 @@ def main_ui():
             map_html = make_map_html(last_lat, last_lon, preds)
             return explanation, map_html
 
-        add_btn.click(on_add_point, inputs=[date_input, time_input, loc_input], outputs=add_status)
+        add_btn.click(on_add_point, inputs=[date_input, time_input, loc_input], outputs=[add_status, point_counter])
         train_btn.click(on_train, inputs=[trace_text], outputs=train_status)
         predict_btn.click(on_predict, outputs=[predict_output, map_output])
-        clear_btn.click(lambda: clear_session(), outputs=add_status)
+        clear_btn.click(lambda: (clear_session(), "0"), outputs=[add_status, point_counter])
 
     return demo