Update app.py

app.py

@@ -5,102 +5,98 @@ import streamlit as st
 from graph import EssayWriter
 import os
 import base64
-from readability import Readability
 
-# Set up the page configuration
-st.set_page_config(page_title="Multi-Agent Essay Writer", page_icon="✍️", layout="wide")
+st.set_page_config(page_title="Multi-Agent Essay Writer", page_icon="🤖")
+
+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>
+'''
 
-# Initialize session state variables
 if "messages" not in st.session_state:
-    st.session_state.messages = [{"role": "assistant", "content": "Hello! How can I assist you with your essay?"}]
+    st.session_state.messages = [{"role": "assistant", "content": "Hello!"}]
     st.session_state.app = None
     st.session_state.chat_active = True
-    st.session_state.saved_essays = {}
 
-# Sidebar for settings and initialization
 with st.sidebar:
-    st.title("Settings")
-    st.info("🔑 Provide your OpenAI API key to enable functionality.")
+    st.info(" * This app uses the OpenAI API to generate text, please provide your API key."
+            "\n\n * This app uses the 'gpt-4o-mini-2024-07-18' model. Cost effective and efficient."
+            "\n\n * Writing essays may take some time, please be patient. 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", type="password")
+    openai_key = st.secrets.get("OPENAI_API_KEY", "") if not developer_mode else st.text_input("OpenAI API Key (Developer Mode)", type="password")
 
-    if st.button("Initialize AI Agents"):
-        if len(openai_key) > 0:
-            os.environ["OPENAI_API_KEY"] = openai_key
-            st.session_state.app = EssayWriter()
-            st.success("Agents initialized successfully!")
-            st.session_state.chat_active = False
-        else:
-            st.error("OpenAI API key is missing! Please provide a valid key.")
+def initialize_agents():
+    if len(openai_key) < 1:
+        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
+    essay_writer = EssayWriter().graph
+    st.success("Agents successfully initialized!")
+    st.session_state.chat_active = False
+    return essay_writer
+
+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)
 
-    # Display saved essays for easy access
-    st.markdown("### Saved Essays")
-    for title, content in st.session_state.saved_essays.items():
-        if st.button(f"Load Essay: {title}"):
-            st.session_state.messages.append({"role": "assistant", "content": f"Loaded Essay: {title}"})
-            st.session_state.loaded_essay = content
+app = st.session_state.app
 
-# Function to generate AI responses
 def generate_response(topic):
-    if st.session_state.app:
-        return st.session_state.app.graph.invoke({"topic": topic})
-    return {"response": "Error: AI agents are not initialized."}
+    return app.invoke(input={"topic": topic}) if app else {"response": "Error: Agents not initialized."}
 
-# Display chat history
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"], unsafe_allow_html=True)
 
-# Chat input for user queries
-response = None 
-if topic := st.chat_input(placeholder="Ask me to write, edit, format, or analyze an essay!"):
+if topic := st.chat_input(placeholder="Ask a question", disabled=st.session_state.chat_active):
     st.chat_message("user").markdown(topic)
     st.session_state.messages.append({"role": "user", "content": topic})
 
-    with st.spinner("Processing your request..."):
+    with st.spinner("Thinking..."):
         response = generate_response(topic)
 
     with st.chat_message("assistant"):
-        if response and "pdf_name" in response:
+        if "pdf_name" in response:
             with open(f"./{response['pdf_name']}", "rb") as file:
                 file_bytes = file.read()
                 b64 = base64.b64encode(file_bytes).decode()
-            href = f"<a href='data:application/octet-stream;base64,{b64}' download='{response['pdf_name']}'>Download Your Essay</a>"
-            st.markdown(f"{response['response']} {href}", unsafe_allow_html=True)
-            st.session_state.messages.append({"role": "assistant", "content": f"{response['response']} {href}"})
-        elif response:
+            href = f"<a href='data:application/octet-stream;base64,{b64}' download='{response['pdf_name']}'>Click here to download the PDF</a>"
+            st.markdown(f"{response['response']}: {href}", unsafe_allow_html=True)
+            st.session_state.messages.append({"role": "assistant", "content": f"{response['response']}: {href}"})
+        else:
             st.markdown(response["response"])
             st.session_state.messages.append({"role": "assistant", "content": response["response"]})
-        else:
-            st.error("Failed to process your request. Please try again.")
-
-    # Save the essay for future reference
-    if response and "essay" in response:
-        save_title = st.text_input("Save this essay as:", value="Untitled Essay")
-        if st.button("Save Essay"):
-            st.session_state.saved_essays[save_title] = response["essay"]
-            st.success(f"Essay saved as '{save_title}'.")
-
-# Add readability analysis for the generated essay
-if response and "essay" in response:
-    st.markdown("---")
-    st.markdown("### Essay Analysis")
-
-    essay_text = response["essay"]
-    r = Readability(essay_text)
-    try:
-        flesch_score = r.flesch()
-        st.metric("Flesch Reading Ease", f"{flesch_score.score:.2f}")
-    except Exception:
-        st.warning("Could not calculate readability. The essay may not be in a valid format.")
 
-# Footer for acknowledgments
+# Add acknowledgment at the bottom
 st.markdown("---")
 st.markdown(
     """
-    <div style="text-align: center; font-size: 12px; color: #777;">
-        <strong>Multi-Agent Essay Writer:</strong> A tool for writing, editing, and analyzing essays.
+    <div style="text-align: center; font-size: 14px; color: #555;">
+        <strong>Acknowledgment:</strong> This project 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
+        </a>
     </div>
-    """,
-    unsafe_allow_html=True,
-)
+    """, unsafe_allow_html=True)