updates

app.py

@@ -3,28 +3,32 @@ from json import JSONDecodeError
 import logging
 
 import streamlit as st
+
 from utils.haystack import run_agent, start_haystack
 from utils.ui import reset_results, set_initial_state, sidebar
-from utils.config import TWITTER_BEARER_TOKEN, SERPER_KEY, OPENAI_API_KEY
+from utils.config import TWITTER_BEARER_TOKEN, OPENAI_API_KEY, SERPER_KEY
 
 set_initial_state()
 
 sidebar()
 
 st.write("# 👩 What would they have tweeted about this?")
+st.write("## About this App")
+st.write("This app uses an [Agent](https://docs.haystack.deepset.ai/docs/agent) (using GPT-4) with a `WebSearch` and `TwitterRetriever` tool")
+st.write("The `TwitterRetriever` is set to retrieve the latest 15 tweets by the given twitter username to construct an understanding of their style of tweeting")
+st.write("### Instructions")
+st.write("For best results, please provide a Twitter username as it appears on twietter. E.g.: dog_feelings 🐶")
 
-# if st.session_state.get("OPENAI_API_KEY"):
-if "last_k_tweets" not in st.session_state:
-    st.session_state["last_k_tweets"] = st.slider("How many tweets should we retrieve?", value=10, step=10, max_value=100, min_value=10, on_change = reset_results)
+# if st.session_state.get("OPENAI_API_KEY") and st.session_state.get("SERPER_KEY"):
 
-agent = start_haystack(openai_key=OPENAI_API_KEY, twitter_bearer=TWITTER_BEARER_TOKEN, serper_key=SERPER_KEY, last_k_tweets=st.session_state["last_k_tweets"])
+agent = start_haystack(openai_key=OPENAI_API_KEY, twitter_bearer=TWITTER_BEARER_TOKEN, serper_key=SERPER_KEY)
 # st.session_state["api_keys_configured"] = True
 
 # Search bar
 question = st.text_input("Example: If the twitter account tuanacelik were to write a tweet in their style about climate change, what would it be?", on_change=reset_results) 
 run_pressed = st.button("Generate tweet")
 # else:
-#     st.write("Please provide your OpenAI Key to start using the application")
+#     st.write("Please provide your OpenAI and SerperDev Keys to start using the application")
 #     st.write("If you are using a smaller screen, open the sidebar from the top left to provide your OpenAI Key 🙌")
 
 # if st.session_state.get("api_keys_configured"):
@@ -46,5 +50,4 @@ if run_query and question:
             
 if st.session_state.result:
     result = st.session_state.result
-    st.write(result["answers"][0].answer)
-            
+    st.write(result["answers"][0].answer)

utils/haystack.py

@@ -6,7 +6,7 @@ from haystack.nodes import PromptNode, PromptTemplate, WebRetriever
 from haystack.pipelines import WebQAPipeline
 
 
-def start_haystack(openai_key, twitter_bearer, serper_key, last_k_tweets):
+def start_haystack(openai_key, twitter_bearer, serper_key):
     prompt_node = PromptNode(
         "text-davinci-003",
         default_prompt_template=PromptTemplate(prompt="./prompts/lfqa.yaml"),
@@ -17,9 +17,9 @@ def start_haystack(openai_key, twitter_bearer, serper_key, last_k_tweets):
     web_retriever = WebRetriever(api_key=serper_key, top_search_results=2, mode="preprocessed_documents")
     web_pipeline = WebQAPipeline(retriever=web_retriever, prompt_node=prompt_node)
 
-    twitter_retriver = TwitterRetriever(bearer_token=twitter_bearer, last_k_tweets=last_k_tweets)
+    twitter_retriver = TwitterRetriever(bearer_token=twitter_bearer, last_k_tweets=15)
 
-    pn = PromptNode(model_name_or_path="gpt-4", api_key=openai_key, stop_words=["Observation:"])
+    pn = PromptNode(model_name_or_path="gpt-4", api_key=openai_key, stop_words=["Observation:"], max_length=400)
     agent = Agent(prompt_node=pn, prompt_template="./prompts/twitter_agent.yaml")
 
     tweet_retriver_tool = Tool(name="TwitterRetriever", pipeline_or_node=twitter_retriver, 

utils/ui.py

@@ -16,6 +16,9 @@ def reset_results(*args):
 # def set_openai_api_key(api_key: str):
 #     st.session_state["OPENAI_API_KEY"] = api_key
 
+# def set_serper_dev_key(api_key: str):
+#     st.session_state["SERPER_KEY"] = api_key
+
 def sidebar():
     with st.sidebar:
         image = Image.open('logo/haystack-logo-colored.png')
@@ -27,12 +30,12 @@ def sidebar():
 
         st.markdown(
             "## How to use\n"
-            # "1. Enter your [OpenAI API key](https://platform.openai.com/account/api-keys) below\n"
+            # "1. Enter your [OpenAI API](https://platform.openai.com/account/api-keys) and [SerperDev API](https://serper.dev/) keys below\n"
             "1. Enter a query that includes a twitter username and be descriptive about wanting a tweet as a result.\n"
             "2. Enjoy 🤗\n"
         )
 
-        # api_key_input = st.text_input(
+        # openai_api_key_input = st.text_input(
         #     "OpenAI API Key",
         #     type="password",
         #     placeholder="Paste your OpenAI API key here (sk-...)",
@@ -40,8 +43,19 @@ def sidebar():
         #     value=st.session_state.get("OPENAI_API_KEY", ""),
         # )
 
-        # if api_key_input:
-        #     set_openai_api_key(api_key_input)
+        # serper_api_key_input = st.text_input(
+        #     "SerperDev API Key",
+        #     type="password",
+        #     placeholder="Paste your SerperDev API key here (sk-...)",
+        #     help="You can get your API key from https://serper.dev.",
+        #     value=st.session_state.get("SERPER_KEY", ""),
+        # )
+
+        # if openai_api_key_input:
+        #     set_openai_api_key(openai_api_key_input)
+        
+        # if serper_api_key_input:
+        #     set_serper_dev_key(serper_api_key_input)
 
         st.markdown("---")
         st.markdown(