Update app.py

app.py

@@ -10,24 +10,27 @@ def process_pdf(uploaded_file, qa_model, tokenizer):
         file_contents = uploaded_file.read()
 
         # Process the PDF file
-        doc = fitz.open(stream=uploaded_file.read(), filetype="pdf")
-        text = ""
-        for page in doc:
-            text += page.get_text()
-
-        # Tokenize the text
-        inputs = tokenizer(text, return_tensors="pt", max_length=512, truncation=True)
-
-        # Perform question answering
-        outputs = qa_model(**inputs)
-        start_scores = outputs.start_logits
-        end_scores = outputs.end_logits
-
-        # Display the generated questions and answers
-        for i, (start, end) in enumerate(zip(start_scores, end_scores)):
-            answer = tokenizer.decode(inputs["input_ids"][i][start.argmax():end.argmax()+1])
-            st.write("Answer:", answer)
-            st.write("---")
+        doc = fitz.open(file_contents, filetype="pdf")
+        if doc is not None:
+            text = ""
+            for page in doc:
+                text += page.get_text()
+
+            # Tokenize the text
+            inputs = tokenizer(text, return_tensors="pt", max_length=512, truncation=True)
+
+            # Perform question answering
+            outputs = qa_model(**inputs)
+            start_scores = outputs.start_logits
+            end_scores = outputs.end_logits
+
+            # Display the generated questions and answers
+            for i, (start, end) in enumerate(zip(start_scores, end_scores)):
+                answer = tokenizer.decode(inputs["input_ids"][i][start.argmax():end.argmax()+1])
+                st.write("Answer:", answer)
+                st.write("---")
+        else:
+            st.error("Error occurred while opening the PDF file.")
 
 # Main function
 def main():