|
|
import streamlit as st |
|
|
import requests |
|
|
from utils.config import TWITTER_BEARER, OEPN_AI_KEY |
|
|
|
|
|
from haystack.nodes import PromptNode, PromptTemplate |
|
|
|
|
|
|
|
|
@st.cache_resource |
|
|
def start_haystack(): |
|
|
|
|
|
prompt_node = PromptNode(model_name_or_path="text-davinci-003", api_key=OEPN_AI_KEY) |
|
|
|
|
|
twitter_template = PromptTemplate(name="twitter-voice", prompt_text="""You will be given a twitter stream belonging to a specific profile. Tell us what they've lately been tweeting about and in what languages. |
|
|
You may go into some detail about what topics they tend to like tweeting about. Please also mention their overall tone, for example: positive, |
|
|
negative, political, sarcastic or something else. |
|
|
|
|
|
Example: |
|
|
|
|
|
Twitter stream: Many people in our community asked how to utilize LLMs in their NLP pipelines and how to modify prompts for their tasks.… |
|
|
RT @deepset_ai: We use parts of news articles from The Guardian as documents and create custom prompt templates to categorize these article |
|
|
|
|
|
Voice: This person has lately been tweeting about NLP and LLMs. Their tweets have been in Enlish |
|
|
|
|
|
Example: |
|
|
|
|
|
Twitter stream: I've directed my team to set sharper rules on how we deal with unidentified objects.\n\nWe will inventory, improve ca… |
|
|
the incursion by China’s high-altitude balloon, we enhanced radar to pick up slower objects.\n \nBy doing so, w… |
|
|
I gave an update on the United States’ response to recent aerial objects. |
|
|
|
|
|
Voice: This person has lately been tweeting about an unidentified object and an incursion by China with a high-altitude baloon. |
|
|
They have been tweeting about the USA. They have had a political tone. They mostly post in English. |
|
|
|
|
|
Twitter stream: $tweets. |
|
|
|
|
|
Voice: |
|
|
""") |
|
|
return prompt_node, twitter_template |
|
|
|
|
|
prompter, template = start_haystack() |
|
|
|
|
|
@st.cache_data(show_spinner=False) |
|
|
def query(username): |
|
|
|
|
|
bearer_token = TWITTER_BEARER |
|
|
|
|
|
headers = {"Authorization": "Bearer {}".format(bearer_token)} |
|
|
|
|
|
url = f"https://api.twitter.com/1.1/statuses/user_timeline.json?screen_name={username}&count={100}" |
|
|
try: |
|
|
response = requests.request("GET", url, headers = headers) |
|
|
twitter_stream = "" |
|
|
for tweet in response.json(): |
|
|
twitter_stream += (tweet["text"]) |
|
|
result = prompter.prompt(prompt_template=template, tweets=twitter_stream[0:4097]) |
|
|
except: |
|
|
result = ["Please make sure you are providing a correct, public twitter accout"] |
|
|
return result |
