Update app.py

app.py

@@ -16,17 +16,43 @@ 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_")
-
-def update_message(request: gr.Request):
-    return f"Welcome, {request.username}"
+#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
+    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="")
+    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="")
+    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="")
+    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="")
+    else:
+        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():
+    return gr.update(visible=False)  # Hide Panel 2
+
+# Function to show Panel 1
+def show_panel_1():
+    return gr.update(visible=True)  # 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()
+
+#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}"
@@ -132,25 +158,39 @@ 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)
-        range_input = gr.Slider(1, 100, 6, label="Window / Moving Average Period")
+    # Create the first panel (visible from the start)
+    with gr.Column(visible=True) as panel_1:
+        user_text = gr.Textbox(label="User name:", placeholder="Type here...", max_lines=1)
+        input_text = gr.Textbox(label="Password:", placeholder="Type here...", max_lines=1, type="password")
     
-    table_ = gr.DataFrame(value=df, label="Predictions (consult 'revenue' column):")
-    btn_actualizar = gr.Button("Run Forecast")
+        # First button, visible from the start
+        btn_1 = gr.Button("Login")
 
-    #with gr.Row():
-    #    file_name_input = gr.Textbox(label="CSV Name", placeholder="Predictions.csv")
-    #    btn_guardar = gr.Button("Save CSV")
     
-    btn_actualizar.click(fn=update_table, inputs=[start_input, prediction_input, range_input], outputs=table_)
-    #btn_guardar.click(fn=save_csv, inputs=file_name_input, outputs=gr.Textbox(label="File Status:"))
+    with gr.Column(visible=False) as panel_2:
+        #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)
+            range_input = gr.Slider(1, 10, 6, label="Window / Moving Average Period")
+        
+        table_ = gr.DataFrame(value=df, label="Predictions (consult 'revenue' column):")
+        btn_update = gr.Button("Run Forecast")
+    
+        #with gr.Row():
+        #    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_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
+
 
 demo.launch()
-#demo.launch(share=True, auth=[(user_1, pass_1), (user_2, pass_2), (user_3, pass_3), (user_4, pass_4)])
-#demo.launch(auth=[("Pete", "123"), ("Da", "abc")])
+