Create app.py

app.py

@@ -0,0 +1,68 @@
+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()