File size: 1,849 Bytes
12c6e97
0f150a5
 
 
 
 
 
 
 
 
b1356e4
 
 
 
 
 
0f150a5
 
b1356e4
bff2a00
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b8e4e1b
 
bff2a00
 
98c9ae1
 
 
 
 
 
 
bff2a00
98c9ae1
 
 
 
 
 
 
bff2a00
98c9ae1
12c6e97
 
0f150a5
 
bff2a00
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
from gensim.parsing.preprocessing import STOPWORDS
import wikipedia
import gradio as gr
from gradio.mix import Parallel
import requests
import nltk
from nltk.tokenize import word_tokenize
from nltk.tokenize import sent_tokenize
import re
nltk.download('punkt')
import numpy as np
import pandas as pd
from os import path
from PIL import Image
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt


def opendomain(text,wikipedia_language="en"):
  try:
    question_words = STOPWORDS.union(set(['likes','play','.',',','like',"don't",'?','use','choose','important','better','?']))
    lower_text = text.lower()
    lower_text = word_tokenize(lower_text)
    new_text = [i for i in lower_text if i not in question_words]
    new_txt = "".join(new_text)
    if wikipedia_language:
      wikipedia.set_lang(wikipedia_language)

    et_page = wikipedia.page(new_txt.replace(" ", ""))
    title = et_page.title
    content = et_page.content
    page_url = et_page.url
    linked_pages = et_page.links

    text = content
    wordcloud = WordCloud(font_path="HelveticaWorld-Regular.ttf").generate(text)

    plt.imshow(wordcloud, interpolation='bilinear')
    plt.axis("off")
  except:
    return "Please write correct wikipedia article name OR question"


  final_out = re.sub(r'\=.+\=', '', text)
  result = list(filter(lambda x: x != '', final_out.split('\n\n')))
  answer = []
  for i in range(6):
      if len(result[i]) > 500:
          summary_point=result[i].split(".")[0:3]
          answer.append(summary_point)

  final = ""
  for value in answer:
    joint_value = ".".join(value)
    if final == "":
      final += joint_value
    else:
      final = f"{final}.\n\n{joint_value}"

  return final


iface = gr.Interface(fn=opendomain, inputs=[gr.inputs.Textbox(lines=5)], outputs="text")
iface.launch()