Update driveapi/drive.py

driveapi/drive.py

@@ -50,10 +50,11 @@ def download_file(file_id):
 
 # Function to process a PDF file
 def process_pdf(file_stream):
-    print(f"File stream is of type: {type(file_stream)}")
-    print(f"File stream is : {file_stream}")
-    file_path = file_stream['name'] # Use 'path' for local testing and 'name' for Gradio
-    pdf_reader = PyPDF2.PdfReader(file_path)
+    if isinstance(file_stream, dict): # Check if PDF was obtained using Drag and Drop or Drive link
+        file_path = file_stream['name'] # Use 'path' for local testing and 'name' for Gradio
+        pdf_reader = PyPDF2.PdfReader(file_path)
+    else:
+        pdf_reader = PyPDF2.PdfReader(file_stream)
     text = ""
     for page_num in range(len(pdf_reader.pages)):
         page = pdf_reader.pages[page_num]