from transformers import pipeline
import streamlit as st
def generate(str):
  pipe = pipeline("text-generation", model="huggingtweets/cryptoanglio")
  story_txt = pipe(str)[0]['generated_text']

  print(story_txt)
  return story_txt

def classify(str):
  pipe = pipeline("text-classification", model="yzhangqs/Finace_news")
  result=pipe(str)
  print(result[0]['label'])
  return result[0]['label']

def main():
  st.set_page_config(page_title="A Crypto News Generation Robots", page_icon="🦜")
  #st.header("Choose Positive or Negative")
  #choice = st.text_input("Import your choice")
  st.header("Import text here: ")
  input_text =st.text_input("Import your text")

  if input_text is not None:
      print(input_text)
      st.text('Generating the news....')
      news = generate(input_text)
      st.write(news)

      st.text('The predicted Trend is')
      label = classify(news)
      st.write(label)

if __name__ == "__main__":
    main()