Update app.py

app.py

@@ -76,116 +76,14 @@ def chat_with_ai(user_input, chat_history):
 def clear_history():
     return [], ""
 
-import os
-import PyPDF2
-import docx
-import pandas as pd
-
-def extract_text_from_file(file_path):
-    """
-    Extracts text from the file based on its extension.
-    Supports: PDF, DOC/DOCX, TXT, XLS/XLSX.
-    """
-    ext = os.path.splitext(file_path)[1].lower()
-    text = ""
-    
-    if ext == ".pdf":
-        try:
-            with open(file_path, "rb") as f:
-                pdf_reader = PyPDF2.PdfReader(f)
-                for page in pdf_reader.pages:
-                    page_text = page.extract_text()
-                    if page_text:
-                        text += page_text + "\n"
-        except Exception as e:
-            text = f"Error processing PDF: {e}"
-    
-    elif ext in [".doc", ".docx"]:
-        try:
-            doc = docx.Document(file_path)
-            text = "\n".join([para.text for para in doc.paragraphs])
-        except Exception as e:
-            text = f"Error processing Word document: {e}"
-    
-    elif ext == ".txt":
-        try:
-            with open(file_path, "r", encoding="utf-8") as f:
-                text = f.read()
-        except Exception as e:
-            text = f"Error processing TXT file: {e}"
-    
-    elif ext in [".xls", ".xlsx"]:
-        try:
-            # Read the first sheet of the Excel file
-            df = pd.read_excel(file_path)
-            # Convert the dataframe to CSV format (or any format you prefer)
-            text = df.to_csv(index=False)
-        except Exception as e:
-            text = f"Error processing Excel file: {e}"
-    
-    else:
-        text = "Unsupported file type for text extraction."
-    
-    return text
-
 def upload_file(file):
-    """
-    Handles file upload from Gradio.
-    Saves the file to the "new_file" directory and extracts text content based on file type.
-    Supports file-like objects, dictionaries, or file paths.
-    """
-    # Check if a file was uploaded
-    if file is None:
-        return "No file uploaded!"
-    
-    # If file is a list (multiple files), take the first one
-    if isinstance(file, list):
-        file = file[0]
-    
-    # Initialize file_name and file_data based on the type of 'file'
-    if hasattr(file, 'read'):
-        # file is a file-like object
-        file_data = file.read()
-        file_name = getattr(file, 'name', "uploaded_file")
-    elif isinstance(file, dict):
-        # file is a dictionary with "name" and "data" keys
-        file_name = file.get("name", "uploaded_file")
-        file_data = file.get("data")
-    elif isinstance(file, str):
-        # file is a string (e.g., a NamedString representing a file path)
-        file_name = os.path.basename(file)
-        try:
-            with open(file, "rb") as f:
-                file_data = f.read()
-        except Exception as e:
-            return f"Error reading file from path: {e}"
-    else:
-        return "Uploaded file format not recognized."
-    
-    # Validate that file_data is available
-    if file_data is None:
-        return "Uploaded file data not found!"
-    
-    # Ensure the "new_file" directory exists
+    # Save the uploaded file to the "new_file" directory
     if not os.path.exists("new_file"):
         os.makedirs("new_file")
-    
-    # Save the file to the "new_file" directory
-    file_path = os.path.join("new_file", file_name)
-    try:
-        with open(file_path, "wb") as f:
-            f.write(file_data)
-    except Exception as e:
-        return f"Error saving file: {e}"
-    
-    # Extract text from the file for further processing
-    extracted_text = extract_text_from_file(file_path)
-    
-    # Create a preview of the extracted text
-    preview = extracted_text[:200] + "..." if len(extracted_text) > 200 else extracted_text
-    return f"File {file_name} uploaded and processed successfully!\nExtracted text preview:\n{preview}"
-
-
+    file_path = os.path.join("new_file", file.name)
+    with open(file_path, "wb") as f:
+        f.write(file.read())
+    return f"File {file.name} uploaded successfully!"
 
 def gradio_chatbot():
     with gr.Blocks() as demo: