Update app.py

app.py

@@ -5,7 +5,7 @@ from io import BytesIO
 from PyPDF2 import PdfReader
 import docx
 from textwrap import wrap
-from pptx import Presentation
+import json
 
 # Configure Streamlit
 st.set_page_config(
@@ -38,8 +38,7 @@ task = st.sidebar.selectbox("What would you like to do?", [
     "Adapt Lesson for Grades",
     "Generate Conceptual Assignment",
     "Provide Learning Resources",
-    "Generate Short Conceptual Questions",
-    "Generate Presentation (PPT)"
+    "Generate Conceptual Short Questions"
 ])
 
 # Helper functions
@@ -71,27 +70,13 @@ def save_to_docx(text, filename="download.docx"):
     byte_io.seek(0)
     return byte_io
 
-def save_to_ppt(slides_content, filename="presentation.pptx"):
-    presentation = Presentation()
-    for slide_title, slide_content in slides_content:
-        slide = presentation.slides.add_slide(presentation.slide_layouts[1])
-        title = slide.shapes.title
-        body = slide.shapes.placeholders[1]
-
-        title.text = slide_title
-        tf = body.text_frame
-        for point in slide_content:
-            p = tf.add_paragraph()
-            p.text = point
-
-    byte_io = BytesIO()
-    presentation.save(byte_io)
-    byte_io.seek(0)
-    return byte_io
-
 # Main App Layout
 st.title("EduAI Assistant for Teachers")
-st.markdown("""Welcome to your AI-powered teaching assistant!""")
+st.markdown("""
+Welcome to your AI-powered teaching assistant!  
+- Upload lesson files or input text.  
+- Perform actions like summarizing topics, generating quizzes, and adapting lessons.
+""")
 
 if uploaded_files or manual_input:
     lesson_text = ""
@@ -110,63 +95,85 @@ if uploaded_files or manual_input:
 
     text_chunks = chunk_text(lesson_text)
 
-    if task == "Generate Presentation (PPT)":
-        topic = st.text_input("Enter the topic for the presentation:")
-        
-        if topic:
-            # Generate slide content based on the topic
-            slide_content = []
+    if task == "Summarize a Topic":
+        topic = st.text_input("Enter the topic or keywords:")
+        if st.button("Summarize"):
+            summaries = [process_with_groq([
+                {"role": "system", "content": "Summarize the following lesson content."},
+                {"role": "user", "content": f"Context: {chunk}\n\nSummarize the topic: {topic}"}
+            ]) for chunk in text_chunks]
+            st.write("### Summary")
+            summary_text = "\n\n".join(summaries)
+            st.write(summary_text)
+            docx_file = save_to_docx(summary_text)
+            st.download_button("Download Summary as DOCX", docx_file, file_name="summary.docx")
+
+    elif task == "Ask Questions":
+        question = st.text_input("Enter your question:")
+        if st.button("Get Answer"):
+            answers = [process_with_groq([
+                {"role": "system", "content": "You are a helpful teaching assistant."},
+                {"role": "user", "content": f"Context: {chunk}\n\nQuestion: {question}"}
+            ]) for chunk in text_chunks]
+            st.write("### Answer")
+            st.write("\n\n".join(answers))
+
+    elif task == "Generate MCQs":
+        num_questions = st.slider("Number of questions to generate:", 1, 10, 5)
+        if st.button("Generate MCQs"):
+            mcqs = []
             for chunk in text_chunks:
-                content = process_with_groq([{
-                    "role": "system", "content": "Generate content for a presentation slide."
-                }, {
-                    "role": "user", "content": f"Topic: {topic}\n\nProvide a few bullet points for a slide on the following content: {chunk}"
-                }])
-
-                slides = content.split("\n")
-                title = slides[0]  # First line as the slide title
-                slide_content_points = slides[1:]  # Remaining lines as bullet points
-                slide_content.append((title, slide_content_points))
-
-            # Show the generated slide content for review
-            st.subheader("Generated Presentation Content")
-            for i, (slide_title, slide_content_points) in enumerate(slide_content):
-                st.write(f"### {slide_title}")
-                for point in slide_content_points:
-                    st.write(f"- {point}")
-                st.markdown("---")
-
-            # Option to regenerate the content
-            regenerate = st.button("Regenerate Content")
-            if regenerate:
-                st.write("Content regenerated. Review again!")
-                # Regenerate the content by calling the same process again
-                slide_content = []
-                for chunk in text_chunks:
-                    content = process_with_groq([{
-                        "role": "system", "content": "Generate content for a presentation slide."
-                    }, {
-                        "role": "user", "content": f"Topic: {topic}\n\nProvide a few bullet points for a slide on the following content: {chunk}"
-                    }])
-
-                    slides = content.split("\n")
-                    title = slides[0]  # First line as the slide title
-                    slide_content_points = slides[1:]  # Remaining lines as bullet points
-                    slide_content.append((title, slide_content_points))
-
-                # Show regenerated content
-                st.subheader("Regenerated Presentation Content")
-                for i, (slide_title, slide_content_points) in enumerate(slide_content):
-                    st.write(f"### {slide_title}")
-                    for point in slide_content_points:
-                        st.write(f"- {point}")
-                    st.markdown("---")
-
-            # Allow download once the user is satisfied
-            if st.button("Generate PowerPoint"):
-                ppt_file = save_to_ppt(slide_content)
-                st.write("### Generated Presentation")
-                st.download_button("Download Presentation (PPT)", ppt_file, file_name="presentation.pptx")
-
+                mcqs.extend(process_with_groq([
+                    {"role": "system", "content": "Generate multiple-choice questions for a lesson."},
+                    {"role": "user", "content": f"Context: {chunk}\n\nGenerate {num_questions} MCQs as a JSON array."}
+                ]))
+            st.write("### Multiple Choice Questions")
+            for mcq in mcqs:
+                st.write(f"**{mcq}**")
+
+    elif task == "Adapt Lesson for Grades":
+        grade = st.slider("Select Grade:", 1, 16, 9)
+        if st.button("Adapt Lesson"):
+            adaptations = [process_with_groq([
+                {"role": "system", "content": "Adapt the lesson content for a specific grade."},
+                {"role": "user", "content": f"Context: {chunk}\n\nAdapt this lesson for grade {grade}."}
+            ]) for chunk in text_chunks]
+            st.write("### Adapted Lesson")
+            st.write("\n\n".join(adaptations))
+
+    elif task == "Generate Conceptual Assignment":
+        topic = st.text_input("Enter the topic for the assignment:")
+        if st.button("Generate Assignment"):
+            assignments = [process_with_groq([
+                {"role": "system", "content": "Generate a conceptual-based assignment."},
+                {"role": "user", "content": f"Context: {chunk}\n\nTopic: {topic}"}
+            ]) for chunk in text_chunks]
+            st.write("### Conceptual Assignment")
+            assignment_text = "\n\n".join(assignments)
+            docx_file = save_to_docx(assignment_text)
+            st.download_button("Download Assignment as DOCX", docx_file, file_name="assignment.docx")
+
+    elif task == "Provide Learning Resources":
+        topic = st.text_input("Enter the topic:")
+        if st.button("Generate Resources"):
+            resources = process_with_groq([
+                {"role": "system", "content": "Provide a list of learning resources for a topic."},
+                {"role": "user", "content": f"Topic: {topic}"}
+            ])
+            st.write("### Learning Resources")
+            st.write(resources)
+
+    elif task == "Generate Conceptual Short Questions":
+        topic = st.text_input("Enter the topic for conceptual short questions:")
+        if st.button("Generate Conceptual Short Questions"):
+            conceptual_questions = [process_with_groq([
+                {"role": "system", "content": "Generate conceptual short questions based on the provided lesson content."},
+                {"role": "user", "content": f"Context: {chunk}\n\nGenerate deep conceptual questions for the topic: {topic}"}
+            ]) for chunk in text_chunks]
+            st.write("### Conceptual Short Questions")
+            conceptual_questions_text = "\n\n".join(conceptual_questions)
+            st.write(conceptual_questions_text)
+            docx_file = save_to_docx(conceptual_questions_text)
+            st.download_button("Download Conceptual Short Questions as DOCX", docx_file, file_name="conceptual_short_questions.docx")
 else:
     st.info("Please upload files or enter lesson text to get started.")