Update app.py

app.py

@@ -1,51 +1,24 @@
 import os
 import re
-from shutil import copy
 import gradio as gr
-from os import makedirs, listdir
-from os.path import isdir, isfile, join, realpath
+from shutil import copy
 from tkinter import Tk
 from tkinter.filedialog import askdirectory
+from os import makedirs, listdir
+from os.path import isdir, isfile, join, realpath
 
 # --- Constants ---
 phases = ["iFAT", "(i)SAT"]
 inputPhases = {**{i: k for i, k in enumerate(phases)}, **{len(phases): "All"}}
-exitinput = {"no", "n", "0"}
 regTemplate = r'template'
 
 # --- Helper Functions ---
 
-# Function to print available options
-def printOptions():
-    print("\nchoose one of the following options;\n")
-    for key in inputPhases:
-        print("[%d] %s" % (key, inputPhases[key]))
-    print()
-
-# Function to get phase input
-def validatedPhaseInput():
-    inputPhase = None
-    while inputPhase is None:
-        printOptions()
-        inputPhase = input()
-
-        if inputPhase.isnumeric():
-            inputPhase = int(inputPhase)
-            if inputPhase not in range(len(inputPhases)):
-                print("\n", inputPhase, "is not a valid option")
-                inputPhase = None
-            else:
-                return inputPhases[inputPhase]
-        else:
-            inputPhase = inputPhase.lower()
-            if inputPhase not in phases:
-                print("\n", inputPhase, "is not a valid option")
-                inputPhase = None
-            else:
-                return inputPhases[inputPhase]
-
-    print("Something went wrong, please consult the maintainer of the codebase.")
-    return inputPhase
+# Function to open a folder selection dialog
+def select_folder():
+    Tk().withdraw()  # Hides the root window
+    folder_selected = askdirectory()
+    return folder_selected
 
 # Function to validate if a directory exists
 def validate_directory(directory_path):
@@ -169,13 +142,26 @@ def validate_and_run(phase, root_dir, templates_dir):
     
     return run_steps(phase, root_dir, templates_dir)
 
+# Functions to trigger folder selection for Gradio
+def select_root_directory():
+    return select_folder()
+
+def select_templates_directory():
+    return select_folder()
+
 # Gradio app components
 with gr.Blocks() as app:
     gr.Markdown("# Folder Creation Tool")
     
     phase_input = gr.Dropdown(choices=list(inputPhases.values()), label="Select Phase")
-    root_dir_input = gr.Textbox(label="Root Directory", placeholder="Enter the root directory path")
-    templates_dir_input = gr.Textbox(label="Templates Directory", placeholder="Enter the templates directory path")
+    
+    root_dir_input = gr.Textbox(label="Root Directory", placeholder="Select the root directory")
+    root_dir_button = gr.Button("Select Root Directory")
+    root_dir_button.click(fn=select_root_directory, outputs=root_dir_input)
+    
+    templates_dir_input = gr.Textbox(label="Templates Directory", placeholder="Select the templates directory")
+    templates_dir_button = gr.Button("Select Templates Directory")
+    templates_dir_button.click(fn=select_templates_directory, outputs=templates_dir_input)
     
     create_button = gr.Button("Create Folders")
     output = gr.Textbox(label="Output")