Update app.py

app.py

@@ -159,15 +159,18 @@ with tab1:
         with st.chat_message(message["role"]):
             st.markdown(message["content"], unsafe_allow_html=True)
 
-    # Use text_input to allow a default value, but do not trigger generation immediately
+    # Input
     topic = st.text_input("📝 Provide an essay topic:", value="Write an essay on the cultural diversity of India")
 
-    # Add a button to trigger essay generation
-    if st.button("Generate Essay"):
-        if topic:
-            # Store user message in the chat
-            st.chat_message("user").markdown(topic)
-            st.session_state["messages"].append({"role": "user", "content": topic})
+    # Add spacing
+    st.write("")
+
+    # Generate button
+    if st.button("🚀 Generate Essay"):
+        if topic and topic.strip():  # Ensure it's not empty
+            # Store user message only if it's not already stored
+            if not any(msg["content"] == topic for msg in st.session_state["messages"]):
+                st.session_state["messages"].append({"role": "user", "content": topic})
 
             with st.spinner("⏳ Generating your essay..."):
                 response = None
@@ -176,86 +179,87 @@ with tab1:
                 else:
                     st.error("⚠️ Agents are not initialized. Please check the system or restart the app.")
 
-            # Store assistant response and display it
-            with st.chat_message("assistant"):
-                if response and "essay" in response:  # Display essay preview and allow editing
-                    essay = response["essay"]
-
-                    # Store response in session state
-                    assistant_response = f"Here is your {essay_length}-word essay preview and the download link."
-                    st.session_state["messages"].append({"role": "assistant", "content": assistant_response})
-
-                    # Create Two-Column Layout
-                    col1, col2 = st.columns(2)
-
-                    with col1:
-                        st.markdown(f"### 📝 Essay Preview ({essay_length} words)")
-                        st.markdown(f"#### {essay['header']}")
-                        st.markdown(essay["entry"])
-
-                        for para in essay["paragraphs"]:
-                            st.markdown(f"**{para['sub_header']}**")
-                            st.markdown(para["paragraph"])
-
-                        st.markdown("**🖊️ Conclusion:**")
-                        st.markdown(essay["conclusion"])
-
-                    with col2:
-                        st.markdown("### ✍️ Edit Your Essay:")
-
-                        # 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 Here:", value=full_essay_text, height=300)
-
-                        # Save and Download buttons
-                        save_col1, save_col2 = st.columns(2)
-
-                        with save_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 save_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 
-                    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 original PDF</a>"
-                            st.markdown(href, unsafe_allow_html=True)
-
-                    # Save response in session state
-                    st.session_state["messages"].append(
-                        {"role": "assistant", "content": f"Here is your {essay_length}-word essay preview and the download link."}
-                    )
-                elif response:
-                    st.markdown(response["response"])
-                    st.session_state["messages"].append({"role": "assistant", "content": response["response"]})
-                else:
-                    st.error("⚠️ No response received. Please try again.")
+            # Store and display assistant response
+            if response and "essay" in response:
+                essay = response["essay"]
+
+                assistant_response = f"Here is your {essay_length}-word essay preview and the download link."
+                st.session_state["messages"].append({"role": "assistant", "content": assistant_response})
+
+                st.chat_message("assistant").markdown(assistant_response)
+
+                # Create Two-Column Layout
+                col1, col2 = st.columns(2)
+
+                with col1:
+                    st.markdown(f"### 📝 Essay Preview ({essay_length} words)")
+                    st.markdown(f"#### {essay['header']}")
+                    st.markdown(essay["entry"])
+
+                    for para in essay["paragraphs"]:
+                        st.markdown(f"**{para['sub_header']}**")
+                        st.markdown(para["paragraph"])
+
+                    st.markdown("**🖊️ Conclusion:**")
+                    st.markdown(essay["conclusion"])
+
+                with col2:
+                    st.markdown("### ✍️ Edit Your Essay:")
+
+                    # 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 Here:", value=full_essay_text, height=300)
+
+                    # Save and Download buttons
+                    save_col1, save_col2 = st.columns(2)
+
+                    with save_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 save_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 
+                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 original PDF</a>"
+                        st.markdown(href, unsafe_allow_html=True)
+
+                # Save response in session state
+                st.session_state["messages"].append(
+                    {"role": "assistant", "content": f"Here is your {essay_length}-word essay preview and the download link."}
+                )
+            elif response:
+                st.markdown(response["response"])
+                st.session_state["messages"].append({"role": "assistant", "content": response["response"]})
+            else:
+                st.error("⚠️ No response received. Please try again.")
+
 
 
 # 📊 Tab 2: Workflow Visualization