Update app.py

app.py

@@ -129,29 +129,48 @@ def extract_text_from_file(file_path):
     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]
     
-    if hasattr(file, 'name'):
-        file_name = file.name
+    # 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
     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:
@@ -159,27 +178,13 @@ def upload_file(file):
     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)
-    if hasattr(file, "read"):
-        content = file.read()
-    elif isinstance(file, dict) and "data" in file:
-        content = file["data"]
-    else:
-        return "Uploaded file format not recognized."
-    
-    with open(file_path, "wb") as f:
-        f.write(content)
-    
-    return f"File {file_name} uploaded successfully!"
 
 
 def gradio_chatbot():