Update app.py

app.py

@@ -89,51 +89,54 @@ def generate_response(topic, length):
 
     return app.invoke(input={"topic": topic, "length": length})
 
-# Display chat messages from the session
-if "messages" in st.session_state:
-    for message in st.session_state["messages"]:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"], unsafe_allow_html=True)
-
-# Handle user input
-if topic := st.chat_input(placeholder="📝 Ask a question or provide an essay topic...", disabled=st.session_state["chat_active"]):
-    st.chat_message("user").markdown(topic)
-    st.session_state["messages"].append({"role": "user", "content": topic})
-
-    with st.spinner("⏳ Generating your essay..."):
-        response = generate_response(topic, essay_length)  
-
-    # Handle the assistant's response
-    with st.chat_message("assistant"):
-        if "essay" in response:  # Display essay preview and download link
-            st.markdown(f"### 📝 Essay Preview ({essay_length} words)")
-            st.markdown(f"#### {response['essay']['header']}")
-            st.markdown(response["essay"]["entry"])
-            for para in response["essay"]["paragraphs"]:
-                st.markdown(f"**{para['sub_header']}**")
-                st.markdown(para["paragraph"])
-            st.markdown("**🖊️ Conclusion:**")
-            st.markdown(response["essay"]["conclusion"])
-
-            # Provide download link for the PDF
-            with open(response["pdf_name"], "rb") as pdf_file:
-                b64 = base64.b64encode(pdf_file.read()).decode()
-                href = f"<a href='data:application/octet-stream;base64,{b64}' download='{response['pdf_name']}'>📄 Click here to download the PDF</a>"
-                st.markdown(href, unsafe_allow_html=True)
-
-            # Save the assistant's message to session state
-            st.session_state["messages"].append(
-                {"role": "assistant", "content": f"Here is your {essay_length}-word essay preview and the download link."}
-            )
-        else:  # For other responses (e.g., general answers)
-            st.markdown(response["response"])
-            st.session_state["messages"].append({"role": "assistant", "content": response["response"]})
-
-
-----------
-
-
-
+# Define Tabs
+tab1, tab2 = st.tabs(["📜 Essay Generation", "📊 Workflow Visualization"])
+
+# 📜 Tab 1: Essay Generation
+with tab1:
+    st.subheader("📝 Generate an Essay")
+
+    # Display chat messages from the session
+    if "messages" in st.session_state:
+        for message in st.session_state["messages"]:
+            with st.chat_message(message["role"]):
+                st.markdown(message["content"], unsafe_allow_html=True)
+
+    # Handle user input
+    if topic := st.chat_input(placeholder="📝 Ask a question or provide an essay topic...", disabled=st.session_state["chat_active"]):
+        st.chat_message("user").markdown(topic)
+        st.session_state["messages"].append({"role": "user", "content": topic})
+
+        with st.spinner("⏳ Generating your essay..."):
+            response = generate_response(topic, essay_length)  
+
+        # Handle the assistant's response
+        with st.chat_message("assistant"):
+            if "essay" in response:  # Display essay preview and download link
+                st.markdown(f"### 📝 Essay Preview ({essay_length} words)")
+                st.markdown(f"#### {response['essay']['header']}")
+                st.markdown(response["essay"]["entry"])
+                for para in response["essay"]["paragraphs"]:
+                    st.markdown(f"**{para['sub_header']}**")
+                    st.markdown(para["paragraph"])
+                st.markdown("**🖊️ Conclusion:**")
+                st.markdown(response["essay"]["conclusion"])
+
+                # Provide download link for the PDF
+                with open(response["pdf_name"], "rb") as pdf_file:
+                    b64 = base64.b64encode(pdf_file.read()).decode()
+                    href = f"<a href='data:application/octet-stream;base64,{b64}' download='{response['pdf_name']}'>📄 Click here to download the PDF</a>"
+                    st.markdown(href, unsafe_allow_html=True)
+
+                # Save the assistant's message to session state
+                st.session_state["messages"].append(
+                    {"role": "assistant", "content": f"Here is your {essay_length}-word essay preview and the download link."}
+                )
+            else:  # For other responses (e.g., general answers)
+                st.markdown(response["response"])
+                st.session_state["messages"].append({"role": "assistant", "content": response["response"]})
+
+# 📊 Tab 2: Workflow Visualization
 with tab2:
     st.subheader("📊 Multi-Agent Workflow Visualization")
 
@@ -160,4 +163,4 @@ st.markdown(
     </div>
     """,
     unsafe_allow_html=True,
-    )
+)