Update app.py

app.py

@@ -155,9 +155,16 @@ def provide_learning_resources(topic):
     response = client.chat.completions.create(messages=messages, model="llama-3.3-70b-versatile", stream=False)
     return response.choices[0].message.content
 
+# Function to export summary or assignment to DOCX
+def export_to_docx(text, filename="download.docx"):
+    doc = docx.Document()
+    doc.add_paragraph(text)
+    doc.save(filename)
+    return filename
+
 # Streamlit app layout
 st.title("EduAI Assistant for Teachers")
-st.markdown("""
+st.markdown("""  
 Welcome to the AI-powered teaching assistant!  
 - Upload your lesson files or input text.  
 - Ask questions, summarize topics, create quizzes, assignments, and access learning resources.
@@ -201,7 +208,13 @@ if manual_input or uploaded_files:
         if st.button("Summarize"):
             summaries = [summarize_topic(chunk, topic) for chunk in text_chunks]
             st.write("### Summary")
-            st.write("\n\n".join(summaries))
+            full_summary = "\n\n".join(summaries)
+            st.write(full_summary)
+
+            # Add export to DOCX button
+            if st.button("Export Summary to DOCX"):
+                filename = export_to_docx(full_summary)
+                st.success(f"Summary has been exported as {filename}")
 
     elif task == "Ask Questions":
         question = st.text_input("Enter your question:")
@@ -236,7 +249,13 @@ if manual_input or uploaded_files:
         if st.button("Generate Assignment"):
             assignments = [generate_assignment(chunk, topic) for chunk in text_chunks]
             st.write("### Conceptual Assignment")
-            st.write("\n\n".join(assignments))
+            full_assignment = "\n\n".join(assignments)
+            st.write(full_assignment)
+
+            # Add export to DOCX button
+            if st.button("Export Assignment to DOCX"):
+                filename = export_to_docx(full_assignment)
+                st.success(f"Assignment has been exported as {filename}")
 
     elif task == "Provide Learning Resources":
         topic = st.text_input("Enter the topic for learning resources:")