Update app.py

app.py

@@ -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, user_text):
+def update_table(start_date, end_date, window, tu):
     #############################################################################################
     try:
         DATABRICKS_INSTANCE = 'https://136286767906206.6.gcp.databricks.com'
@@ -90,7 +90,7 @@ def update_table(start_date, end_date, window, user_text):
             "month_train_starting": str(start_date),
             "month_for_prediction": str(end_date),
             "moving_average_period": str(window),
-            "username": str(user_text) 
+            "username": str(tu) 
         }
         
         url = f"{DATABRICKS_INSTANCE}/api/2.1/jobs/run-now"
@@ -123,7 +123,7 @@ def update_table(start_date, end_date, window, user_text):
         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(str(user_text).split('@'))[0]+".csv")
+        data_predictions = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_predictions_"+str(tu).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(label="Usuario", interactive=False) 
+            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():