Update Gradio_UI.py

Gradio_UI.py

@@ -207,10 +207,12 @@ class GradioUI:
             "application/pdf",
             "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
             "text/plain",
+            "text/csv",
+            "application/vnd.ms-excel"
         ],
     ):
         """
-        Handle file uploads, default allowed types are .pdf, .docx, and .txt
+        Handle file uploads, default allowed types are .pdf, .docx, .csv, .xlsx and .txt
         """
         import gradio as gr
 
@@ -218,31 +220,30 @@ class GradioUI:
             return gr.Textbox("No file uploaded", visible=True), file_uploads_log
 
         try:
-            mime_type, _ = mimetypes.guess_type(file.name)
+            if file.name.endswith(".csv"):
+                mime_type = "text/csv"
+            else:
+                mime_type, _ = mimetypes.guess_type(file.name)
         except Exception as e:
             return gr.Textbox(f"Error: {e}", visible=True), file_uploads_log
 
         if mime_type not in allowed_file_types:
             return gr.Textbox("File type disallowed", visible=True), file_uploads_log
 
-        # Sanitize file name
+        # Extract name and extension safely
         original_name = os.path.basename(file.name)
-        sanitized_name = re.sub(
-            r"[^\w\-.]", "_", original_name
-        )  # Replace any non-alphanumeric, non-dash, or non-dot characters with underscores
-
-        type_to_ext = {}
-        for ext, t in mimetypes.types_map.items():
-            if t not in type_to_ext:
-                type_to_ext[t] = ext
-
-        # Ensure the extension correlates to the mime type
-        sanitized_name = sanitized_name.split(".")[:-1]
-        sanitized_name.append("" + type_to_ext[mime_type])
-        sanitized_name = "".join(sanitized_name)
-
-        # Save the uploaded file to the specified folder
-        file_path = os.path.join(self.file_upload_folder, os.path.basename(sanitized_name))
+        name, ext = os.path.splitext(original_name)
+
+        # Sanitize the filename (only alphanumeric, dash, and dot allowed)
+        sanitized_name = re.sub(r"[^\w\-.]", "_", name) + ext
+
+        # Ensure extension matches mime type
+        expected_ext = mimetypes.guess_extension(mime_type)
+        if expected_ext and ext.lower() != expected_ext:
+            sanitized_name = sanitized_name.replace(ext, expected_ext)
+
+        # Save the uploaded file
+        file_path = os.path.join(self.file_upload_folder, sanitized_name)
         shutil.copy(file.name, file_path)
 
         return gr.Textbox(f"File uploaded: {file_path}", visible=True), file_uploads_log + [file_path]