Update app.py

app.py

@@ -155,16 +155,18 @@ 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"):
+# Function to save text to a docx file
+def save_to_docx(text, filename="download.docx"):
     doc = docx.Document()
     doc.add_paragraph(text)
-    doc.save(filename)
-    return filename
+    byte_io = BytesIO()
+    doc.save(byte_io)
+    byte_io.seek(0)
+    return byte_io
 
 # 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.
@@ -208,13 +210,12 @@ if manual_input or uploaded_files:
         if st.button("Summarize"):
             summaries = [summarize_topic(chunk, topic) for chunk in text_chunks]
             st.write("### Summary")
-            full_summary = "\n\n".join(summaries)
-            st.write(full_summary)
+            summary_text = "\n\n".join(summaries)
+            st.write(summary_text)
 
-            # 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}")
+            # Export to docx
+            docx_file = save_to_docx(summary_text)
+            st.download_button("Download Summary as DOCX", docx_file, file_name="summary.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
 
     elif task == "Ask Questions":
         question = st.text_input("Enter your question:")
@@ -248,14 +249,13 @@ if manual_input or uploaded_files:
         topic = st.text_input("Enter the topic for assignment generation:")
         if st.button("Generate Assignment"):
             assignments = [generate_assignment(chunk, topic) for chunk in text_chunks]
+            assignment_text = "\n\n".join(assignments)
             st.write("### Conceptual Assignment")
-            full_assignment = "\n\n".join(assignments)
-            st.write(full_assignment)
+            st.write(assignment_text)
 
-            # 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}")
+            # Export to docx
+            docx_file = save_to_docx(assignment_text)
+            st.download_button("Download Assignment as DOCX", docx_file, file_name="assignment.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
 
     elif task == "Provide Learning Resources":
         topic = st.text_input("Enter the topic for learning resources:")