Update test_app.py

test_app.py

@@ -1,126 +1,139 @@
 import streamlit as st
-from graph import EssayWriter
+from graph import EssayWriter, RouteQuery, GraphState
+from crew import *
 import os
 import base64
 
-st.set_page_config(page_title="Multi-Agent Essay Writer", page_icon="🤖")
-
-# Button HTML (if needed for external links)
-button_html = f'''
-    <div style="display: flex; justify-content: center;">
-        <a href=" " target="_blank">
-            <button style="
-                background-color: #FFDD00;
-                border: none;
-                color: black;
-                padding: 10px 20px;
-                text-align: center;
-                text-decoration: none;
-                display: inline-block;
-                font-size: 16px;
-                margin: 4px 2px;
-                cursor: pointer;
-                border-radius: 10px;
-                box-shadow: 2px 2px 10px rgba(0, 0, 0, 0.2);
-            ">
-            </button>
-        </a>
-    </div>
-'''
+#st.title("Multi-Agent Essay Writing Assistant")
+st.set_page_config(page_title="Multi-Agent Essay Writing Assistant", page_icon="🤖🤖🤖✍️", layout = "centered")
 
-# Initialize session state
+# Ensure session state variables are initialized properly
 if "messages" not in st.session_state:
-    st.session_state.messages = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]
-    st.session_state.app = None
-    st.session_state.chat_active = True
+    st.session_state["messages"] = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]
+
+if "app" not in st.session_state:
+    st.session_state["app"] = None
 
-# Sidebar with instructions and API key input
+if "chat_active" not in st.session_state:
+    st.session_state["chat_active"] = True
+
+# Sidebar with essay settings and user-defined length
 with st.sidebar:
+    st.subheader("About:")
     st.info(
-        "* ⚠️ Initialize the agents to start using the application."
         "\n\n* This app uses the 'gpt-4o-mini-2024-07-18' model."
         "\n\n* Writing essays may take some time, approximately 1-2 minutes."
     )
-    developer_mode = st.checkbox("Developer Mode", value=False)
-    openai_key = st.secrets.get("OPENAI_API_KEY", "") if not developer_mode else st.text_input("OpenAI API Key (Developer Mode)", type="password")
+
+    # API Key Retrieval
+    openai_key = st.secrets.get("OPENAI_API_KEY", "")
+
+    st.divider()
+
+    # User-defined essay length selection
+    st.subheader("📝 Configure Essay Settings:")
+    essay_length = st.number_input(
+        "Select Essay Length (words):",
+        min_value=150,
+        max_value=400,
+        value=250,
+        step=50
+    )
+
+    st.divider()
+
+    # Reference section
+    st.subheader("📖 References:")
+    st.markdown(
+        "[1. Multi-Agent System with CrewAI and LangChain](https://discuss.streamlit.io/t/new-project-i-have-build-a-multi-agent-system-with-crewai-and-langchain/84002)",
+        unsafe_allow_html=True
+    )
 
 # Initialize agents function
 def initialize_agents():
     if not openai_key:
-        st.error("OpenAI API key is missing! Please provide a valid key through Hugging Face Secrets or Developer Mode.")
-        st.session_state.chat_active = True
+        st.error("⚠️ OpenAI API key is missing! Please provide a valid key through Hugging Face Secrets.")
+        st.session_state["chat_active"] = True
         return None
 
     os.environ["OPENAI_API_KEY"] = openai_key
     try:
+        if st.session_state["app"] is not None:
+            return st.session_state["app"]  # Prevent re-initialization
+        
         essay_writer = EssayWriter().graph
-        st.success("Agents successfully initialized!")
-        st.session_state.chat_active = False
+        st.session_state["chat_active"] = False
+
+        # Success message
+        #st.success("✅ Agents successfully initialized!")
         return essay_writer
     except Exception as e:
-        st.error(f"Error initializing agents: {e}")
-        st.session_state.chat_active = True
+        st.error(f"❌ Error initializing agents: {e}")
+        st.session_state["chat_active"] = True
         return None
 
-# Sidebar button for initializing agents
-with st.sidebar:
-    if st.button("Initialize Agents", type="primary"):
-        st.session_state.app = initialize_agents()
-    st.divider()
-    st.markdown(button_html, unsafe_allow_html=True)
+# Automatically initialize agents on app load
+if st.session_state["app"] is None:
+    st.session_state["app"] = initialize_agents()
 
-# Access the initialized app
-app = st.session_state.app
+if st.session_state["app"] is None:
+    st.error("⚠️ Failed to initialize agents. Please check your API key and restart the app.")
+
+app = st.session_state["app"]
 
 # Function to invoke the agent and generate a response
-def generate_response(topic):
-    return app.invoke(input={"topic": topic}) if app else {"response": "Error: Agents not initialized."}
+def generate_response(topic, length):
+    if not app:
+        st.error("⚠️ Agents are not initialized. Please check the system or restart the app.")
+        return {"response": "Error: Agents not initialized."}
+
+    return app.invoke(input={"topic": topic, "length": length})
 
 # Display chat messages from the session
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"], unsafe_allow_html=True)
+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):
+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})
+    st.session_state["messages"].append({"role": "user", "content": topic})
 
-    with st.spinner("Thinking..."):
-        response = generate_response(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("### Essay Preview:")
+            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("**🖊️ 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>"
+                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": "Here is your essay preview and the download link."}
+            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"]})
+            st.session_state["messages"].append({"role": "assistant", "content": response["response"]})
 
-# Add acknowledgment at the bottom
-st.markdown("---")
+# Acknowledgment Section
 st.markdown(
     """
-    <div style="text-align: center; font-size: 14px; color: #555;">
-        <strong>Acknowledgment:</strong> This project is based on Mesut Duman's work: 
+    <div style="text-align: center; font-size: 14px; color: #555; padding-top: 250px; margin-top: 250px;">
+        <strong>Acknowledgment:</strong> This app is based on Mesut Duman's work: 
         <a href="https://github.com/mesutdmn/Autonomous-Multi-Agent-Systems-with-CrewAI-Essay-Writer/tree/main" 
            target="_blank" style="color: #007BFF; text-decoration: none;">
            CrewAI Essay Writer
@@ -128,4 +141,4 @@ st.markdown(
     </div>
     """,
     unsafe_allow_html=True,
-)
+    )