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), user_text#, gr.update(value=""), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)#, user_text#, 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), user_text#, gr.update(value=""), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)#, user_text#, 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), user_text#, gr.update(value=""), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)#, user_text#, 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), user_text#, gr.update(value=""), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)#, user_text#, gr.update(value=""), gr.update(value="")
     else:
-        return gr.update(visible=False), gr.update(visible=True), user_text#, user_text, input_text  # Keep Panel 2 hidden, show Panel 1
+        return gr.update(visible=False), gr.update(visible=True)#, user_text#, user_text, input_text  # Keep Panel 2 hidden, show Panel 1
 
 # Function to hide Panel 2
 def hide_panel():
@@ -77,7 +77,7 @@ def get_month_label(i):
 #df['Month'] = [get_month_label(i) for i in range(len(df))]
 df = pd.DataFrame()
     
-def update_table(start_date, end_date, window, tu):
+def update_table(start_date, end_date, window, user_text):
     #############################################################################################
     try:
         DATABRICKS_INSTANCE = 'https://136286767906206.6.gcp.databricks.com'
@@ -90,7 +90,7 @@ def update_table(start_date, end_date, window, tu):
             "month_train_starting": str(start_date),
             "month_for_prediction": str(end_date),
             "moving_average_period": str(window),
-            "username": str(tu) 
+            "username": str(user_text) 
         }
         
         url = f"{DATABRICKS_INSTANCE}/api/2.1/jobs/run-now"
@@ -123,7 +123,7 @@ def update_table(start_date, end_date, window, tu):
         print(f"Run ID: {run_id}")
         #data_payments = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_payments.csv")
         #data_metrics = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_metrics.csv")
-        data_predictions = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_predictions_"+str(tu).split('@')[0]+".csv")
+        data_predictions = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_predictions_"+str(user_text).split('@')[0]+".csv")
         #df_payments = pd.read_csv(StringIO(data_payments.decode('utf-8')))
         #new_columns = ['vintage', 'vintage_unique_cases'] + [f'M{col}' for col in df_payments.columns[2:]] 
         #df_metrics = pd.read_csv(StringIO(data_metrics.decode('utf-8')))
@@ -161,7 +161,7 @@ with gr.Blocks(fill_height=True) as demo:
         #    demo.load(update_message, None, gr.Markdown())
         #    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
-            tu = gr.Textbox(interactive=False, scale=0) 
+            #tu = gr.Textbox(interactive=False, scale=0) 
             btn_2 = gr.Button("Logout", visible=True, scale=0)
             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():
@@ -172,10 +172,10 @@ with gr.Blocks(fill_height=True) as demo:
         table_ = gr.DataFrame(value=df, label="Predictions (consult 'revenue' column):")
         btn_update = gr.Button("Run Forecast")
         
-        btn_update.click(fn=update_table, inputs=[start_input, prediction_input, range_input, tu], outputs=table_)
+        btn_update.click(fn=update_table, inputs=[start_input, prediction_input, range_input, user_text], outputs=table_)
     
     # Configure the buttons and the panel visibility
-    btn_1.click(check_words, inputs=[user_text, input_text], outputs=[panel_2, panel_1, tu])  # Hide Panel 1 and show Panel 2
+    btn_1.click(check_words, inputs=[user_text, input_text], outputs=[panel_2, panel_1])
 
 demo.launch()