Update app.py

app.py

@@ -19,15 +19,15 @@ JOB_ID = os.getenv("JOB_ID_")
 def check_words(user_text, input_text):
     # If the entered text is "user" and "casa", show Panel 2 and hide Panel 1
     if user_text == os.getenv("USER_1_") and input_text == os.getenv("PASS_1_"):
-        return gr.update(visible=True), gr.update(visible=False), gr.update(value=""), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)#, gr.update(value=""), gr.update(value="")
     elif user_text == os.getenv("USER_2_") and input_text == os.getenv("PASS_2_"):
-        return gr.update(visible=True), gr.update(visible=False), gr.update(value=""), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)#, gr.update(value=""), gr.update(value="")
     elif user_text == os.getenv("USER_3_") and input_text == os.getenv("PASS_3_"):
-        return gr.update(visible=True), gr.update(visible=False), gr.update(value=""), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)#, gr.update(value=""), gr.update(value="")
     elif user_text == os.getenv("USER_4_") and input_text == os.getenv("PASS_4_"):
-        return gr.update(visible=True), gr.update(visible=False), gr.update(value=""), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)#, gr.update(value=""), gr.update(value="")
     else:
-        return gr.update(visible=False), gr.update(visible=True), user_text, input_text  # Keep Panel 2 hidden, show Panel 1
+        return gr.update(visible=False), gr.update(visible=True)#, user_text, input_text  # Keep Panel 2 hidden, show Panel 1
 
 # Function to hide Panel 2
 def hide_panel():
@@ -39,7 +39,7 @@ def show_panel_1():
 
 # Wrapper function to hide Panel 2 and show Panel 1 when btn_2 is clicked
 def hide_and_show_panel():
-    return hide_panel(), show_panel_1()
+    return hide_panel(), show_panel_1(), gr.update(value=""), gr.update(value="")
 
 #def update_message(request: gr.Request):
 #    return f"Welcome, {request.username}"
@@ -76,7 +76,7 @@ df = pd.read_csv(StringIO(data_predictions_.decode('utf-8')))
 df['revenue'] = (df['vintage_unique_cases'] / df['predicted_monthly_payment_rate']).round(2)
 df['Month'] = [get_month_label(i) for i in range(len(df))]
     
-def update_table(start_date, end_date, window):
+def update_table(start_date, end_date, window, user_text):
     #############################################################################################
     try:
         DATABRICKS_INSTANCE = 'https://136286767906206.6.gcp.databricks.com'
@@ -88,7 +88,8 @@ def update_table(start_date, end_date, window):
             "events_api_base_url": "https://beta.datascore.ai",
             "month_train_starting": str(start_date),
             "month_for_prediction": str(end_date),
-            "moving_average_period": str(window)
+            "moving_average_period": str(window),
+            "username": str(user_text) 
         }
         
         url = f"{DATABRICKS_INSTANCE}/api/2.1/jobs/run-now"
@@ -162,7 +163,7 @@ with gr.Blocks(fill_height=True) as demo:
         #    logout_button = gr.Button("Logout", link="/logout", scale=0)
             # Add a second button inside Panel 2 to hide the panel and show Panel 1 again
             btn_2 = gr.Button("Logout", visible=True, scale=0)
-            btn_2.click(hide_and_show_panel, outputs=[panel_2, panel_1])  # Hide Panel 2 and show Panel 1
+            btn_2.click(hide_and_show_panel, outputs=[panel_2, panel_1, user_text, input_text])  # Hide Panel 2 and show Panel 1
         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)
@@ -175,12 +176,12 @@ with gr.Blocks(fill_height=True) as demo:
         #    file_name_input = gr.Textbox(label="CSV Name", placeholder="Predictions.csv")
         #    btn_saving = gr.Button("Save CSV")
         
-        btn_update.click(fn=update_table, inputs=[start_input, prediction_input, range_input], outputs=table_)
+        btn_update.click(fn=update_table, inputs=[start_input, prediction_input, range_input, user_text], outputs=table_)
         #btn_saving.click(fn=save_csv, inputs=file_name_input, outputs=gr.Textbox(label="File Status:"))
 
     
     # Configure the buttons and the panel visibility
-    btn_1.click(check_words, inputs=[user_text, input_text], outputs=[panel_2, panel_1, user_text, input_text])  # Hide Panel 1 and show Panel 2
+    btn_1.click(check_words, inputs=[user_text, input_text], outputs=[panel_2, panel_1])  # Hide Panel 1 and show Panel 2
 
 demo.launch()