Update app.py

app.py

@@ -197,6 +197,109 @@ def getAnalysis(score):
     return 'Neutral'
   else:
     return 'Positive'
+def Original_Image(path):
+  img= cv2.imread(path)
+  img= cv2.cvtColor(img, cv2.COLOR_BGR2RGB)  
+  return img
+
+def Image_Processed(path):
+  '''
+  Reading the image file
+  '''
+  img= cv2.imread(path)
+  img= cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+  
+  #Thresholding
+  ret, bw_img = cv2.threshold(img, 124, 255, cv2.THRESH_BINARY)
+
+  return bw_img
+ 
+def wordCloud(orgIm,mask_img,text_Party_pr,maxWord=2000,colorGener=True,contCol='white',bckColor='white'):
+  '''
+  Generating word cloud
+  '''
+  
+  mask =mask_img
+  # Create and generate a word cloud image:
+  wordcloud = WordCloud(max_words=maxWord, background_color=bckColor,
+                        mask=mask,
+                        colormap='nipy_spectral_r',
+                        contour_color=contCol,
+                        width=400, height=500,
+                        margin=2,
+                        contour_width=3).generate(text_Party_pr)
+
+  # create coloring from image
+
+  plt.figure(figsize=[10,12])
+  
+  if colorGener==True:
+    image_colors = ImageColorGenerator(orgIm)
+    plt.imshow(wordcloud.recolor(color_func= image_colors),interpolation="bilinear")
+    
+  
+  else:    
+    plt.imshow(wordcloud)
+
+  plt.axis("off")
+  
+def word_cloud_generator(parsed_text_name):
+  parsed=parsed_text_name.lower()
+
+  if 'bjp' in parsed:
+    orgImg=Original_Image('BJPMain.jpg')
+    bwImg=Image_Processed('BJPMain.jpg')
+    wordCloud(orgImg,bwImg,text_Party,maxWord=3000,colorGener=True,contCol='white',bckColor='black')
+    plt.tight_layout()
+    buf = BytesIO()
+    plt.savefig(buf)
+    buf.seek(0)
+    img1 = Image.open(buf)
+    plt.clf() 
+    return img1
+
+  
+  elif 'congress' in parsed:
+    orgImg=Original_Image('congrsMain.jpg')
+    bwImg=Image_Processed('congrsMain.jpg')
+
+    wordCloud(orgImg,bwImg,text_Party,maxWord=3000,colorGener=True)
+    plt.tight_layout()
+    buf = BytesIO()
+    plt.savefig(buf)
+    buf.seek(0)
+    img2 = Image.open(buf)
+    plt.clf() 
+    return img2
+    
+  
+  elif 'aap' in parsed:
+    orgImg=Original_Image('AAPMain.jpg')
+    bwImg=Image_Processed('AAPMain.jpg')
+    wordCloud(orgImg,bwImg,text_Party,maxWord=3000,colorGener=False,contCol='black')
+    plt.tight_layout()
+    buf = BytesIO()
+    plt.savefig(buf)
+    buf.seek(0)
+    img4 = Image.open(buf)
+    plt.clf() 
+    return img3
+  
+  else :
+    wordcloud = WordCloud(max_words=2000, background_color="white",mode="RGB").generate(text_Party)
+    plt.figure(figsize=(4,3))
+    plt.imshow(wordcloud, interpolation="bilinear")
+    plt.axis("off")   
+    plt.tight_layout()
+    buf = BytesIO()
+    plt.savefig(buf)
+    buf.seek(0)
+    img4 = Image.open(buf)
+    plt.clf()
+    return img4
+
+
+
 '''
 url = "http://library.bjp.org/jspui/bitstream/123456789/2988/1/BJP-Election-english-2019.pdf"
 path_input = "./Bjp_Manifesto_2019.pdf"
@@ -241,16 +344,7 @@ def analysis(Manifesto,Search):
   img2 = Image.open(buf)
   plt.clf()
   
-  wordcloud = WordCloud(max_words=2000, background_color="white",mode="RGB").generate(text_Party)
-  plt.figure(figsize=(4,3))
-  plt.imshow(wordcloud, interpolation="bilinear")
-  plt.axis("off")   
-  plt.tight_layout()
-  buf = BytesIO()
-  plt.savefig(buf)
-  buf.seek(0)
-  img3 = Image.open(buf)
-  plt.clf()
+  img3 = word_cloud_generator(Manifesto.name)
   
   fdist_Party=fDistance(text_Party)
   img4=fDistancePlot(text_Party)