from sumy.parsers.plaintext import PlaintextParser
from sumy.parsers.html import HtmlParser
from sumy.nlp.tokenizers import Tokenizer
from sumy.nlp.stemmers import Stemmer
from sumy.utils import get_stop_words

import gradio as gr

import nltk
nltk.download('punkt')


def summarize(algorithm, language, sentence_count, input_type, input_):
  if algorithm == 'LSA':
    from sumy.summarizers.lsa import LsaSummarizer as Summarizer
  if algorithm =='text-rank':
    from sumy.summarizers.text_rank import TextRankSummarizer as Summarizer
  if algorithm =='lex-rank':
    from sumy.summarizers.lex_rank import LexRankSummarizer as Summarizer
  if algorithm =='edmundson':
    from sumy.summarizers.edmundson import EdmundsonSummarizer as Summarizer
  if algorithm =='luhn':
    from sumy.summarizers.luhn import LuhnSummarizer as Summarizer
  if algorithm =='kl-sum':
    from sumy.summarizers.kl import KLSummarizer as Summarizer
  if algorithm =='random':
    from sumy.summarizers.random import RandomSummarizer as Summarizer
  if algorithm =='reduction':
    from sumy.summarizers.reduction import ReductionSummarizer as Summarizer

  if input_type=="URL":
    parser = HtmlParser.from_url(input_, Tokenizer(language))
  if input_type=="text":
    parser = PlaintextParser.from_string(input_, Tokenizer(language))
    
  stemmer = Stemmer(language)
  summarizer = Summarizer(stemmer)
  stop_words = get_stop_words(language)

  if algorithm=='edmundson':
    summarizer.null_words = stop_words
    summarizer.bonus_words = parser.significant_words
    summarizer.stigma_words = parser.stigma_words
  else:
    summarizer.stop_words = stop_words

  summary_sentences = summarizer(parser.document, sentence_count)
  summary = ' '.join([str(sentence) for sentence in summary_sentences])
  
  return summary


iface = gr.Interface(
    summarize,
    [
      gr.inputs.Dropdown(["LSA", "luhn", "edmundson", "text-rank", "lex-rank", "random", "reduction", "kl-sum"]),
      gr.inputs.Textbox(1, default="english"),
      gr.inputs.Number(default=5),
      gr.inputs.Radio(["URL", "text"], default="URL"),
      gr.inputs.Textbox(5),
    ],
    "text",
    examples=[
        ["LSA", 'english', 5, "URL", "https://en.wikipedia.org/wiki/Automatic_summarization"]
    ],
)

iface.launch()