Update app.py

app.py

@@ -1,9 +1,28 @@
 import gradio as gr
 
-
 # Function to list files with .csv and .txt extensions in the current directory
 def list_files(file_path):
     import os
+    
+    icon_csv = "📄 "
+    icon_txt = "📑 "
+    
+    current_directory = os.getcwd()
+    file_list = []
+    for filename in os.listdir(current_directory):
+        if filename.endswith(".csv"):
+            file_list.append(icon_csv + filename)
+        elif filename.endswith(".txt"):
+            file_list.append(icon_txt + filename)
+    if file_list:
+        return "\n".join(file_list)
+    else:
+        return "No .csv or .txt files found in the current directory."
+
+
+# Function to list files with .csv and .txt extensions in the current directory
+def list_filesold(file_path):
+    import os
     current_directory = os.getcwd()
     file_list = []
     for filename in os.listdir(current_directory):