Update app.py

app.py

@@ -25,6 +25,9 @@ pass_2 = os.getenv("PASS_2_")
 pass_3 = os.getenv("PASS_3_")
 pass_4 = os.getenv("PASS_4_")
 
+def update_message(request: gr.Request):
+    return f"Welcome, {request.username}"
+
 def get_run_status(run_id):
     url = f"{DATABRICKS_INSTANCE}/api/2.1/jobs/runs/get?run_id={run_id}"
     headers = {
@@ -129,6 +132,10 @@ def save_csv(file_name):
 
 
 with gr.Blocks() as demo:
+    #m = gr.Markdown()
+    with gr.Row():
+        demo.load(update_message, None, gr.Markdown())
+        logout_button = gr.Button("Logout", link="/logout", scale=0)
     with gr.Row():
         start_input = gr.Textbox(label="Start Date:", placeholder="Enter date (e.g., 2023-09-01)", max_lines=1)
         prediction_input = gr.Text(label="Month Prediction:", placeholder="Enter date (e.g., 2024-10-31)", max_lines=1)