Update app.py

app.py

@@ -37,7 +37,8 @@ task = st.sidebar.selectbox("What would you like to do?", [
     "Generate MCQs",
     "Adapt Lesson for Grades",
     "Generate Conceptual Assignment",
-    "Provide Learning Resources"
+    "Provide Learning Resources",
+    "Generate Short Questions"
 ])
 
 # Helper functions
@@ -161,5 +162,18 @@ if uploaded_files or manual_input:
             ])
             st.write("### Learning Resources")
             st.write(resources)
+
+    elif task == "Generate Short Questions":
+        topic = st.text_input("Enter the topic for short questions:")
+        if st.button("Generate Short Questions"):
+            short_questions = [process_with_groq([
+                {"role": "system", "content": "Generate short questions based on the provided lesson content."},
+                {"role": "user", "content": f"Context: {chunk}\n\nGenerate short questions for the topic: {topic}"}
+            ]) for chunk in text_chunks]
+            st.write("### Short Questions")
+            short_questions_text = "\n\n".join(short_questions)
+            st.write(short_questions_text)
+            docx_file = save_to_docx(short_questions_text)
+            st.download_button("Download Short Questions as DOCX", docx_file, file_name="short_questions.docx")
 else:
     st.info("Please upload files or enter lesson text to get started.")