Update app.py

changed the visualize button

app.py

@@ -14,7 +14,6 @@ from matplotlib.figure import Figure
 import csv
 import os
 from huggingface_hub import login
-
 # import os
 # Define the function to process the input file and model selection
 
@@ -245,13 +244,15 @@ def process_file(model_name,inc_slider,progress=Progress(track_tqdm=True)):
         print(f"CSV file '{filename}' created successfully.")
         full_path = os.path.join("fileHandler", filename)
         # upload_to_dataset(full_path)
-
+        
+        upload_to_dataset(full_path)
     task_type_map = {0: "ER", 1: "ME"}
     label_map = {0: "unsuccessful", 1: "successful"}
 
     # -------------------------------
     # 1. Where tlb == plb
     # -------------------------------
+    delete_files()
     for label in [0, 1]:
         # All strategies
         matching_indices = [i for i in range(len(tlb)) if tlb[i] == plb[i] == label]
@@ -844,7 +845,7 @@ def process_file(model_name,inc_slider,progress=Progress(track_tqdm=True)):
 
 
     # return text_output,text_output_sampled_auc,fig,fig_task1,fig_task2,fig_hist1,fig_hist2
-    return text_output,fig,fig_task1,fig_task2
+    return text_output,fig,fig_task1,fig_task2,gr.update(visible=True)
 # List of models for the dropdown menu
 
 # models = ["ASTRA-FT-HGR", "ASTRA-FT-LGR", "ASTRA-FT-FULL"]
@@ -1321,48 +1322,22 @@ with gr.Blocks(theme='gstaff/sketch', css=custom_css) as demo:
         opt1_pie = gr.Plot(label="ER")
         opt2_pie = gr.Plot(label="ME")
 
+    
     with gr.Row():
         with gr.Column():
-            # gr.Markdown("Select strategy filters and click Generate")
-            task_type_radio = gr.Dropdown(
-                choices=["ER", "ME", "All"],
-                label="Filter by Problem Type",
-                interactive=True
-            )
-            source_radio = gr.Checkbox(
-                label="Predicted Labels",
-                value=True
-            )
-            generate_button = gr.Button("Generate Strategies")
-
-    # with gr.Row():
-        with gr.Column():
-            with gr.Group(visible=True) as file_output_group:
-          
-                visualize_markdown = gr.Markdown(visible=True)
-
-
-    def handle_generate(task_type_dropdown, use_predicted):
-        label_source = "Predicted" if use_predicted else "Ground Truth"
-        file_success_path, file_unsuccess_path,file_all_path, viz_link = provide_file_paths(task_type_dropdown, label_source)
-        delete_files()
-        print("writing file to the dataset",file_success_path)
-        # full_path = os.path.join("fileHandler", filename)
-        
-        upload_to_dataset(file_success_path)
-        print("writing file to the dataset",file_unsuccess_path)
-        upload_to_dataset(file_unsuccess_path)
-        print("writing file to the dataset",file_all_path)
-        upload_to_dataset(file_all_path)
-        return viz_link,gr.update(visible=True)
-
-
-    generate_button.click(
-        fn=handle_generate,
-        inputs=[task_type_radio, source_radio],
-        outputs=[ visualize_markdown,file_output_group]
+            redirect_button = gr.Button("Click to Visualization Strategies", visible=False)
+
+    redirect_button.click(
+        None,
+        inputs=[],
+        outputs=[],
+         js="""() => { 
+            window.open("https://path-analysis.vercel.app", "_blank"); 
+            return "ok";  // must return something
+        }"""
     )
 
+
     btn.click(
         fn=lambda model, increment: (
             *process_file(model, increment),  # Unpack all outputs from process_file
@@ -1374,7 +1349,7 @@ with gr.Blocks(theme='gstaff/sketch', css=custom_css) as demo:
             gr.update(visible=False)           # Hide visualize markdown
         ),
         inputs=[model_dropdown, increment_slider],
-        outputs=[output_text, plot_output, opt1_pie, opt2_pie, task_type_radio,source_radio, visualize_markdown]
+        outputs=[output_text, plot_output, opt1_pie, opt2_pie,redirect_button]
     )