Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import os
 import io
+import pdfplumber
+import docx
+from openpyxl import load_workbook
 import shutil
 import threading
 from datetime import datetime
@@ -146,15 +149,36 @@ with gr.Blocks() as demo:
         new_id = session_manager.create_session()
         return new_id, []
 
-    def user(message, file, history, session_id, save_history):
-        if file:
-            try:
-                with open(file, "r", encoding="utf-8") as f:
-                    file_content = f.read()
-                message += f"\nFile content: {file_content}"
-            except Exception as e:
-                message += f"\nError reading file: {str(e)}"
-            message += f"\nUploaded file: {file}"
+    def user(message, files, history, session_id, save_history):
+        if files:
+            for file_path in files:
+                try:
+                    file_extension = os.path.splitext(file_path)[1].lower()
+                    file_content = ""
+
+                    if file_extension == ".pdf":
+                        with pdfplumber.open(file_path) as pdf:
+                            for page in pdf.pages:
+                                file_content += page.extract_text()
+                    elif file_extension == ".docx":
+                        doc = docx.Document(file_path)
+                        for paragraph in doc.paragraphs:
+                            file_content += paragraph.text + "\n"
+                    elif file_extension == ".xlsx":
+                        workbook = load_workbook(file_path)
+                        for sheet in workbook.sheetnames:
+                            worksheet = workbook[sheet]
+                            for row in worksheet.iter_rows():
+                                row_values = [str(cell.value) for cell in row]
+                                file_content += ", ".join(row_values) + "\n"
+                    else:
+                        message += f"\nUnsupported file type: {file_extension}"
+                        continue
+
+                    message += f"\nFile content from {file_path}:\n{file_content}"
+
+                except Exception as e:
+                    message += f"\nError processing {file_path}: {str(e)}"
 
         if save_history:
             session = session_manager.load_session(session_id)
@@ -176,7 +200,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         msg = gr.Textbox(label="Message")
-        file_upload = gr.UploadButton("Upload file")
+        file_upload = gr.File(file_types=[".pdf", ".docx", ".xlsx"], file_count="multiple")
 
     msg.submit(user, [msg, file_upload, chatbot, session_id, save_history], [msg, chatbot]).then(
         bot, [chatbot, session_id], [chatbot]