Update app.py

app.py

@@ -11,20 +11,10 @@ import os
 pd.set_option('display.max_columns', None)
 pd.set_option('display.max_rows', None)
 
-
 DATABRICKS_INSTANCE = 'https://136286767906206.6.gcp.databricks.com'
 ACCESS_TOKEN = os.getenv("ACCESS_TOKEN_")
 JOB_ID = os.getenv("JOB_ID_")
 
-#user_1 = os.getenv("USER_1_")
-#user_2 = os.getenv("USER_2_")
-#user_3 = os.getenv("USER_3_")
-#user_4 = os.getenv("USER_4_")
-#pass_1 = os.getenv("PASS_1_")
-#pass_2 = os.getenv("PASS_2_")
-#pass_3 = os.getenv("PASS_3_")
-#pass_4 = os.getenv("PASS_4_")
-
 # Function that controls the visibility of the panels
 def check_words(user_text, input_text):
     # If the entered text is "user" and "casa", show Panel 2 and hide Panel 1
@@ -147,7 +137,6 @@ def update_table(start_date, end_date, window):
     df = df_predictions.copy()
     return df
 
-
 def save_csv(file_name):
     global df  # Use global variable
     if file_name:
@@ -156,7 +145,6 @@ def save_csv(file_name):
     else:
         return "Please set file name!"
 
-
 with gr.Blocks() as demo:
     # Create the first panel (visible from the start)
     with gr.Column(visible=True) as panel_1:
@@ -186,11 +174,13 @@ with gr.Blocks() as demo:
         
         btn_update.click(fn=update_table, inputs=[start_input, prediction_input, range_input], outputs=table_)
         #btn_saving.click(fn=save_csv, inputs=file_name_input, outputs=gr.Textbox(label="File Status:"))
-
+        
+        # Add a second button inside Panel 2 to hide the panel and show Panel 1 again
+        btn_2 = gr.Button("Logout", visible=True)
+        btn_2.click(hide_and_show_panel, outputs=[panel_2, panel_1])  # Hide Panel 2 and show Panel 1
 
     # 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
 
-
 demo.launch()