Update app.py

app.py

@@ -1,5 +1,19 @@
 import gradio as gr
 
+
+# Function to list files with .csv and .txt extensions in the current directory
+def list_files(file_path):
+    import os
+    current_directory = os.getcwd()
+    file_list = []
+    for filename in os.listdir(current_directory):
+        if filename.endswith(".csv") or filename.endswith(".txt"):
+            file_list.append(filename)
+    if file_list:
+        return "\n".join(file_list)
+    else:
+        return "No .csv or .txt files found in the current directory."
+
 # Function to read a file
 def read_file(file_path):
     try:
@@ -45,11 +59,13 @@ with demo:
     
     label = gr.Label()
 
+    listFiles = gr.Button("List CSV and TXT File(s)")
     readFile = gr.Button("Read File")
     saveFile = gr.Button("Save File")
     deleteFile = gr.Button("Delete File")
     appendFile = gr.Button("Append File")
     
+    listFiles.click(list_files, inputs=fileName, outputs=fileContent)
     readFile.click(read_file, inputs=fileName, outputs=fileContent)
     saveFile.click(write_file, inputs=[fileName, fileContent], outputs=completedMessage)
     deleteFile.click(delete_file, inputs=fileName, outputs=completedMessage)