Update app.py

app.py

@@ -5,22 +5,13 @@ from docx import Document
 import fitz  # PyMuPDF
 import google.generativeai as genai
 
-def extract_text_from_pdf_or_docx(file):
-    """Extract text from PDF or Word document."""
-    filename = file.name
+
+def extract_text_from_pdf(file):
+    """Extract text from PDF."""
     text = ""
-    if filename.endswith('.pdf'):
-        # Extract text from PDF
-        with fitz.open(file) as doc:
-            for page in doc:
-                text += page.get_text()
-    elif filename.endswith('.docx'):
-        # Extract text from Word document
-        doc = Document(file)
-        for paragraph in doc.paragraphs:
-            text += paragraph.text + "\n"
-    else:
-        text = "Unsupported file format. Please upload a PDF or Word document."
+    doc = fitz.open(stream=file.read(), filetype="pdf")
+    for page in doc:
+        text += page.get_text()
     return text
 
 def create_multiple_choice_prompt(num_questions, quiz_context, expertise):
@@ -204,9 +195,9 @@ def main():
     load_dotenv()
     genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
     
-    uploaded_file = st.file_uploader("Upload a PDF or Word document")
+    uploaded_file = st.file_uploader("Upload a PDF document", type=["pdf"])
     if uploaded_file is not None:
-        text = extract_text_from_pdf_or_docx(uploaded_file)
+        text = extract_text_from_pdf(uploaded_file)
         num_questions = st.number_input("Enter the number of questions", min_value=1, max_value=10, value=3)
         quiz_type = st.selectbox("Select the type of Question", ["multiple-choice", "true-false", "open-ended", "fill-in-the-blank", "mixed"])
         expertise = st.text_input("Enter the domain of the questions to be generated.")
@@ -235,4 +226,4 @@ def main():
         st.markdown(st.session_state.answers)
 
 if __name__ == "__main__":
-    main()
+    main()