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):