File size: 5,166 Bytes
03d4ea0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 |
import streamlit as st
from graph import EssayWriter
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>
'''
# Initialize session state
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
# Sidebar with instructions and API key input
with st.sidebar:
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")
# 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
return None
os.environ["OPENAI_API_KEY"] = openai_key
try:
essay_writer = EssayWriter().graph
st.success("Agents successfully initialized!")
st.session_state.chat_active = False
return essay_writer
except Exception as e:
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)
# Access the initialized 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."}
# 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)
# 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("Thinking..."):
response = generate_response(topic)
# 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"#### {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": "Here is your 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"]})
# Add acknowledgment at the bottom
st.markdown("---")
st.markdown(
"""
<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,
)
|