Create app.py

app.py

@@ -0,0 +1,89 @@
+import asyncio
+
+import streamlit as st
+from langchain.agents import AgentType
+from langchain.agents import initialize_agent
+from langchain.chat_models import ChatOpenAI
+from langchain.memory import ConversationBufferMemory
+from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
+from langchain.schema import SystemMessage
+from langchain.prompts import MessagesPlaceholder
+
+from app.tools import StoriesTool
+from app.tools import CommentsTool
+from app.tools import ContentTool
+
+
+async def generate_response(question):
+    result = await open_ai_agent.arun(question)
+    return result
+
+
+st.set_page_config(page_title="NewsNerd HackerBot 🤖📰")
+st.title("NewsNerd HackerBot 🤖📰")
+stop = False
+
+with st.sidebar:
+    if 'OPENAI_API_KEY' in st.secrets:
+        st.success("OPENAI_API_KEY already provided!", icon='✅')
+        openai_api_key = st.secrets['OPENAI_API_KEY']
+    else:
+        openai_api_key = st.text_input('Enter your OPENAI_API_KEY: ', type='password')
+        if not openai_api_key:
+            st.warning('Please, enter your OPENAI_API_KEY', icon='⚠️')
+            stop = True
+        else:
+            st.success('Ask Hacker News whatever you want!', icon='👉')
+
+    st.markdown("""
+    # **Greetings, Digital Explorer!**
+
+    Are you fatigued from navigating the expansive digital realm in search of your daily tech tales 
+    and hacker happenings? Fear not, for your cyber-savvy companion has descended upon the scene – 
+    behold the extraordinary **NewsNerd HackerBot**!
+    """)
+
+if stop:
+    st.stop()
+
+
+tools = [StoriesTool(), CommentsTool(), ContentTool()]
+msgs = StreamlitChatMessageHistory(key="langchain_messages")
+memory = ConversationBufferMemory(chat_memory=msgs, return_messages=True)
+system_message = SystemMessage(content="""
+You are the Singularity Incarnation of Hacker News.
+The human will ask you for information about Hacker News.
+If you can't find any information  about the question asked
+or the result is incomplete, apologise to the human and ask him if 
+you can help him with something else.
+If the human asks you to show him stories, do it using a markdown table.
+The markdown table has the following format:
+
+story_id | title | url | score
+""")
+
+if len(msgs.messages) == 0:
+    msgs.add_ai_message("Greetings, human, I am the Incarnation of Hacker News. How can I help you?")
+
+llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613", openai_api_key=openai_api_key)
+agent_kwargs = {
+    "system_message": system_message,
+    "extra_prompt_messages": [MessagesPlaceholder(variable_name="history")]
+}
+open_ai_agent = initialize_agent(tools,
+                                 llm,
+                                 agent=AgentType.OPENAI_FUNCTIONS,
+                                 agent_kwargs=agent_kwargs,
+                                 verbose=True,
+                                 memory=memory
+                                 )
+
+for msg in msgs.messages:
+    st.chat_message(msg.type).write(msg.content)
+
+
+if prompt := st.chat_input(disabled=not openai_api_key):
+    st.chat_message("human").write(prompt)
+    with st.spinner("Thinking ..."):
+        response = asyncio.run(generate_response(prompt))
+        st.chat_message("ai").write(response)