Update lab/layout2_v3_conversational_style.py

lab/layout2_v3_conversational_style.py

@@ -155,101 +155,104 @@ with tab1:
         with st.chat_message(message["role"]):
             st.markdown(message["content"], unsafe_allow_html=True)
 
-    # Use text_input instead of chat_input to allow a default value
+    # Use text_input to allow a default value, but do not trigger generation immediately
     topic = st.text_input("📝 Provide an essay topic:", value="Write an essay on the cultural diversity of India")
 
-    if topic:
-        # Store user message
-        st.chat_message("user").markdown(topic)
-        st.session_state["messages"].append({"role": "user", "content": topic})
-
-        with st.spinner("⏳ Generating your essay..."):
-            response = None
-            if app:
-                response = app.write_essay({"topic": topic})
-            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.")
+    # 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})
+
+            with st.spinner("⏳ Generating your essay..."):
+                response = None
+                if app:
+                    response = app.write_essay({"topic": topic})
+                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.")
+
 
 
 # 📊 Tab 2: Workflow Visualization
@@ -280,4 +283,4 @@ st.markdown(
     </div>
     """,
     unsafe_allow_html=True,
-)
+)