Update app.py

changed to appkly the path analysis tool function

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, HfApi
 import pickle
 from gradio import Progress
 import numpy as np
@@ -12,9 +12,39 @@ import plotly.graph_objects as go
 from sklearn.metrics import roc_auc_score
 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
 
+api = HfApi()
+DATASET_REPO = "suryadev1/generated-csvs" 
+def delete_files():
+    repo_files = api.list_repo_files(repo_id=DATASET_REPO, repo_type="dataset")
+
+    # Step 2: delete all CSV files
+    for f in repo_files:
+        if f.endswith(".csv"):
+            try:
+                api.delete_file(
+                    path_in_repo=f,
+                    repo_id=DATASET_REPO,
+                    repo_type="dataset"
+                )
+                print(f"Deleted old file: {f}")
+            except Exception as e:
+                print(f"Could not delete {f}: {e}")
+def upload_to_dataset(filepath):
+
+    api.upload_file(
+        path_or_fileobj=filepath,
+        path_in_repo=os.path.basename(filepath),
+        repo_id=DATASET_REPO,
+        repo_type="dataset"
+    )
+
+
 def process_file(model_name,inc_slider,progress=Progress(track_tqdm=True)):
     # progress = gr.Progress(track_tqdm=True)
 
@@ -213,6 +243,8 @@ def process_file(model_name,inc_slider,progress=Progress(track_tqdm=True)):
                     row_num += 1
 
         print(f"CSV file '{filename}' created successfully.")
+        full_path = os.path.join("fileHandler", filename)
+        # upload_to_dataset(full_path)
 
     task_type_map = {0: "ER", 1: "ME"}
     label_map = {0: "unsuccessful", 1: "successful"}
@@ -1305,26 +1337,30 @@ with gr.Blocks(theme='gstaff/sketch', css=custom_css) as demo:
 
     # with gr.Row():
         with gr.Column():
-            with gr.Group(visible=False) as file_output_group:
-            
-                gr.Markdown("**Download strategy descriptor files**")
-                file_output_success = gr.File(label=" ")
-                file_output_unsuccess = gr.File(label=" ")
-                file_output_all = gr.File(label=" ")
-                visualize_markdown = gr.Markdown(visible=False)
+            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)
         
-        return file_success_path, file_unsuccess_path,file_all_path, viz_link,gr.update(visible=True)
+        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=[file_output_success, file_output_unsuccess,file_output_all, visualize_markdown,file_output_group]
+        outputs=[ visualize_markdown,file_output_group]
     )
 
     btn.click(
@@ -1338,7 +1374,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,file_output_success,file_output_unsuccess,file_output_all, visualize_markdown]
+        outputs=[output_text, plot_output, opt1_pie, opt2_pie, task_type_radio,source_radio, visualize_markdown]
     )