Update app.py

app.py

@@ -153,8 +153,6 @@ tab1, tab2 = st.tabs(["📜 Essay Generation", "📊 Workflow Viz"])
 
 # 📜 Tab 1: Essay Generation
 with tab1:
-    #st.subheader("📝 Generate an Essay")
-
     # Display chat messages from the session
     if "messages" not in st.session_state:
         st.session_state["messages"] = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]
@@ -177,7 +175,7 @@ with tab1:
 
         # Handle the assistant's response
         with st.chat_message("assistant"):
-            if response and "essay" in response:  # Display essay preview and download link
+            if response and "essay" in response:  # Display essay preview and allow editing
                 essay = response["essay"]
                 st.markdown(f"### 📝 Essay Preview ({essay_length} words)")
                 st.markdown(f"#### {essay['header']}")
@@ -190,12 +188,51 @@ with tab1:
                 st.markdown("**🖊️ Conclusion:**")
                 st.markdown(essay["conclusion"])
 
-                # Provide download link for the PDF (only if available)
+                # Editable Section
+                st.markdown("## ✍️ Edit Your Essay Below:")
+
+                # Combine all parts of the essay into one editable text field
+                full_essay_text = f"## {essay['header']}\n\n{essay['entry']}\n\n"
+                for para in essay["paragraphs"]:
+                    full_essay_text += f"### {para['sub_header']}\n{para['paragraph']}\n\n"
+                full_essay_text += f"**Conclusion:**\n{essay['conclusion']}"
+
+                # Editable text area for the user
+                edited_essay = st.text_area("Edit Your Essay:", value=full_essay_text, height=400)
+
+                # Save and Download buttons
+                col1, col2 = st.columns(2)
+
+                with col1:
+                    if st.button("💾 Save as TXT"):
+                        with open("edited_essay.txt", "w", encoding="utf-8") as file:
+                            file.write(edited_essay)
+                        with open("edited_essay.txt", "rb") as file:
+                            st.download_button(label="⬇️ Download TXT", data=file, file_name="edited_essay.txt", mime="text/plain")
+
+                with col2:
+                    if st.button("📄 Save as PDF"):
+                        from fpdf import FPDF
+                        
+                        pdf = FPDF()
+                        pdf.set_auto_page_break(auto=True, margin=15)
+                        pdf.add_page()
+                        pdf.set_font("Arial", size=12)
+                        
+                        for line in edited_essay.split("\n"):
+                            pdf.cell(200, 10, txt=line, ln=True, align='L')
+
+                        pdf.output("edited_essay.pdf")
+
+                        with open("edited_essay.pdf", "rb") as file:
+                            st.download_button(label="⬇️ Download PDF", data=file, file_name="edited_essay.pdf", mime="application/pdf")
+
+                # Provide download link for the original PDF (only if available)
                 pdf_name = response.get("pdf_name")
                 if pdf_name and os.path.exists(pdf_name):
                     with open(pdf_name, "rb") as pdf_file:
                         b64 = base64.b64encode(pdf_file.read()).decode()
-                        href = f"<a href='data:application/octet-stream;base64,{b64}' download='{pdf_name}'>📄 Click here to download the PDF</a>"
+                        href = f"<a href='data:application/octet-stream;base64,{b64}' download='{pdf_name}'>📄 Click here to download the original PDF</a>"
                         st.markdown(href, unsafe_allow_html=True)
 
                 # Save response in session state
@@ -209,6 +246,7 @@ with tab1:
                 st.error("⚠️ No response received. Please try again.")
 
 
+
 # 📊 Tab 2: Workflow Visualization
 with tab2:
     #st.subheader("📊 Multi-Agent Essay Writer Workflow Viz")