Update main.py

main.py

@@ -83,55 +83,6 @@ def process_document(image, questions):
 
     return answers
 
-@app.post("/pdfQA/", description=description)
-async def pdf_question_answering(
-    file: UploadFile = File(...),
-    questions: str = Form(...),
-):
-    try:
-        # Read the uploaded file as bytes
-        contents = await file.read()
-
-        # Initialize an empty string to store the text content of the PDF
-        all_text = ""
-
-        # Use PyMuPDF to process the PDF and extract text
-        pdf_document = fitz.open_from_bytes(contents)
-        
-        # Loop through each page and perform OCR
-        for page_num in range(pdf_document.page_count):
-            page = pdf_document.load_page(page_num)
-            print(f"Processing page {page_num + 1}...")
-            text = page.get_text()
-            all_text += text + '\n'
-
-        # Print or do something with the collected text
-        print(all_text)
-
-        # List of questions
-        question_list = questions.split(',')
-
-        # Initialize an empty dictionary to store questions and answers
-        qa_dict = {}
-
-        # Get answers for each question with the same context
-        for question in question_list:
-            result = nlp_qa({
-                'question': question,
-                'context': all_text
-            })
-
-            # Access the 'answer' key from the result
-            answer = result['answer']
-
-            # Store the question and answer in the dictionary
-            qa_dict[question] = answer
-
-        return qa_dict
-
-    except Exception as e:
-        return JSONResponse(content=f"Error processing PDF file: {str(e)}", status_code=500)
-
 # Set up CORS middleware
 origins = ["*"]  # or specify your list of allowed origins
 app.add_middleware(