Update app.py

app.py

@@ -13,18 +13,27 @@ def read_pdf(pdf_path):
 
 # Process and retrieve answers
 def process_invoice(file, hf_token, questions):
- # Read the PDF content directly
- pdf_content = read_pdf(file.name)
-
- # Initialize the Hugging Face pipeline
- qa_pipeline = pipeline("question-answering", model="mistralai/Mixtral-8x7B-Instruct-v0.1", token=hf_token)
-
- answers = {}
- for question in questions.split(','):
- result = qa_pipeline(question=question.strip(), context=pdf_content)
- answers[question] = result['answer']
-
- return answers
+ try:
+ # Read the PDF content directly
+ print("Reading PDF content...")
+ pdf_content = read_pdf(file.name)
+ print(f"PDF Content: {pdf_content[:500]}...") # Print first 500 characters for verification
+
+ # Initialize the Hugging Face pipeline
+ print("Initializing the Hugging Face pipeline...")
+ qa_pipeline = pipeline("question-answering", model="mistralai/Mixtral-8x7B-Instruct-v0.1", use_auth_token=hf_token)
+
+ answers = {}
+ for question in questions.split(','):
+ print(f"Asking question: {question.strip()}")
+ result = qa_pipeline(question=question.strip(), context=pdf_content)
+ answers[question] = result['answer']
+ print(f"Answer: {result['answer']}")
+
+ return answers
+ except Exception as e:
+ print(f"Error: {e}")
+ return {"error": str(e)}
 
 # Gradio interface
 def gradio_interface(file, hf_token, questions):