Update app.py

app.py

@@ -4,15 +4,16 @@ from langchain.chains import LLMChain
 from langchain_google_genai import ChatGoogleGenerativeAI
 import fitz
 import json
+import docx  # Added for Word file handling
 
 # Title
-st.title("📄 PDF-based MCQ Generator")
+st.title("📄 File-based MCQ Generator")
 
 # Sidebar
 st.sidebar.title("Upload & Settings")
 
-# Upload PDF
-pdf_file = st.sidebar.file_uploader("Upload a PDF file", type=["pdf","docx"])
+# Upload file
+uploaded_file = st.sidebar.file_uploader("Upload a file (PDF or Word)", type=["pdf", "docx"])
 
 # Number of questions
 number_of_questions = st.sidebar.slider("Number of questions", min_value=1, max_value=20, value=5)
@@ -28,7 +29,7 @@ if "quiz_finished" not in st.session_state:
     st.session_state.quiz_finished = False
 
 # Gemini setup
-GOOGLE_API_KEY = "AIzaSyCB5NLx39vOAlfRQBDmnEG3uLBgLraGvH4"
+GOOGLE_API_KEY = "your_api_key_here"
 llm = ChatGoogleGenerativeAI(
     model="gemini-2.0-flash",
     google_api_key=GOOGLE_API_KEY,
@@ -60,28 +61,23 @@ prompt = PromptTemplate(
 
 mcq_chain = LLMChain(llm=llm, prompt=prompt)
 
-# PDF or word text extractor
+# Extract text from PDF or Word
 def extract_text(file):
     if file.name.endswith(".pdf"):
         doc = fitz.open(stream=file.read(), filetype="pdf")
-        full_text = ""
-        for page in doc:
-            full_text += page.get_text()
-        doc.close()
-        return full_text
+        return "\n".join([page.get_text() for page in doc])
     elif file.name.endswith(".docx"):
         doc = docx.Document(file)
         return "\n".join([para.text for para in doc.paragraphs])
-    else:
-        return ""
+    return ""
 
 # Generate MCQs
 if st.sidebar.button("Generate MCQs"):
-    if pdf_file is None:
-        st.error("Please upload a PDF file.")
+    if uploaded_file is None:
+        st.error("Please upload a file.")
     else:
         with st.spinner("Extracting text and generating MCQs..."):
-            text = extract_text(pdf_file)
+            text = extract_text(uploaded_file)
             try:
                 response = mcq_chain.run(text=text, number=str(number_of_questions))
                 mcqs_json = json.loads(response[8:-3])
@@ -99,20 +95,18 @@ if st.session_state.mcqs and not st.session_state.quiz_finished:
     q_data = st.session_state.mcqs[idx]
 
     st.subheader(f"Question {idx + 1}: {q_data['question']}")
-    selected_option = st.radio(
-        "Choose an answer:",
-        q_data["options"],
-        key=f"radio_{idx}"
-    )
-
-    if st.button("Next"):
-        st.session_state.user_answers[idx] = selected_option
-
-        if st.session_state.current_q < len(st.session_state.mcqs) - 1:
-            st.session_state.current_q += 1
-        else:
-            st.session_state.quiz_finished = True
-            st.success("🎉 Quiz completed!")
+    
+    with st.form(key=f"form_{idx}"):
+        selected_option = st.radio("Choose an answer:", q_data["options"], key=f"radio_{idx}")
+        submitted = st.form_submit_button("Next")
+
+        if submitted:
+            st.session_state.user_answers[idx] = selected_option
+            if idx < len(st.session_state.mcqs) - 1:
+                st.session_state.current_q += 1
+            else:
+                st.session_state.quiz_finished = True
+                st.success("🎉 Quiz completed!")
 
 # Show result
 if st.session_state.quiz_finished: