2nd

app.py

@@ -181,7 +181,11 @@ def process_file(model_name,inc_slider,progress=Progress(track_tqdm=True)):
             filtered_data = filtered_data[filtered_data[6] == task_type]  # Ensure test_info[6] matches
 
             # Define filename dynamically
-            filename = f"output_task{task_type}_label{label}.csv"
+            task_type_map = {0: "ER", 1: "ME"}
+            label_map = {0: "unsuccessful", 1: "successful"}
+
+            filename = f"{task_type_map[task_type]}-{label_map[label]}-strategies.csv"
+
 
             # Write to CSV
             process_and_write_csv(filtered_data, filename)
@@ -1096,12 +1100,27 @@ FILE_DIR = "fileHandler"
 
 # Function to get list of files
 def list_files():
-    return ['output_task0_label0.csv', 'output_task0_label1.csv', 'output_task1_label0.csv', 'output_task1_label1.csv']
+    return ['Unsuccessful Strategies (ER)', 'Successful Strategies (ER)', 'Unsuccessful Strategies (ME)', 'Successful Strategies (ME)']
+    # return ['output_task0_label0.csv', 'output_task0_label1.csv', 'output_task1_label0.csv', 'output_task1_label1.csv']
     # return [f for f in os.listdir(FILE_DIR) if os.path.isfile(os.path.join(FILE_DIR, f))]
-
+label_to_filename = {
+    'Unsuccessful Strategies (ER)': 'ER-unsuccessful-strategies.csv',
+    'Successful Strategies (ER)': 'ER-successful-strategies.csv',
+    'Unsuccessful Strategies (ME)': 'ME-unsuccessful-strategies.csv',
+    'Successful Strategies (ME)': 'ME-successful-strategies.csv'
+}
 # Function to provide the selected file path
-def provide_file_path(file_name):
-    return f"{FILE_DIR}/{file_name}" if file_name else None
+def provide_file_path(label_name):
+    file_name=label_to_filename.get(label_name, None)
+    dynamic_text = (
+        "🔍 [Visualize the strategies](https://path-analysis.vercel.app/)\n\n"
+        
+        "To use the demo, please follow these steps:\n"
+        "📄 Generate strategies from the AI model and visualize the strategies `.csv` files "
+        "using the path analysis web-app. Refer to the user guide for more details on how to interpret "
+        "the strategy visualization."
+    )
+    return f"{FILE_DIR}/{file_name}" if file_name else None, gr.update(visible=True,value=dynamic_text)
     # file_path = os.path.join(FILE_DIR, file_name)
     # return file_path
 
@@ -1152,13 +1171,14 @@ with gr.Blocks(theme='gstaff/sketch', css=custom_css) as demo:
     #     output_text_sampled_auc = gr.Textbox(label="")
   
     with gr.Row():
-        file_dropdown = gr.Dropdown(choices=list_files(), label="Generate File")
-        download_button = gr.Button("Generate files")
-
+        file_dropdown = gr.Dropdown(choices=list_files(), label="Select a strategy from the dropdown")
+        download_button = gr.Button("Generate Strategies")
+    generated_textbox = gr.Markdown( visible=False)
+    
     download_button.click(
         fn=provide_file_path,
         inputs=[file_dropdown],
-        outputs=[gr.File(label="Your Download is ready, click on the right side to download")]
+        outputs=[gr.File(label=""), generated_textbox]
     )
 
     btn.click(

fileHandler/result.txt

@@ -3,5 +3,9 @@ total_acc: 69.00702106318957
 precisions: 0.7236623191454734
 recalls: 0.6900702106318957
 f1_scores: 0.6802420656474512
+<<<<<<< HEAD
 time_taken_from_start: 2.1567678451538086
+=======
+time_taken_from_start: 39.65812015533447
+>>>>>>> 890efcc1f23279edda21b3cc2f50174aea7ddb43
 auc_score: 0.7457100293916334

fileHandler/roc_data.pkl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2483f35aa06ef8983623602f690eb3fe006654c79d448f4f82a913b4862e34e9
-size 9437

fileHandler/roc_data2.pkl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0578bbc08b428a1f54707fc3aca6aa1063e045033cb007f8ba3361f1aace43df
-size 28023