Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from PIL import Image
-import os
 import pytesseract
 import torch
 import numpy as np
@@ -8,7 +7,6 @@ import nltk
 nltk.download('stopwords')
 nltk.download('punkt')
 from nltk.corpus import stopwords
-from nltk.tokenize import word_tokenize, sent_tokenize
 from nltk.cluster.util import cosine_distance
 import networkx as nx
 from transformers import pipeline
@@ -28,14 +26,11 @@ def read(filepath):
 def clean_text(text):
   article = text.split(".")
   article=[sentence for sentence in article if sentence!=""]
-  # print(article)
 
   sentences = []
 
   for sentence in article:
-      #print(sentence)
       sentence=sentence.replace(",", " , ").replace("'", " ' ").split(" ")
-      #sentence=sentence.replace("[^a-zA-Z]", " ").split(" ")
       sentence=[word for word in sentence if word!=""]
       sentences.append(sentence)
 
@@ -139,8 +134,8 @@ def important_sentences(filepath, no_of_sentences=5):
 def summarize(filepath):
   extractedInformation=read(filepath)
   extractedInformation=' '.join(extractedInformation.split('\n'))
-  output = summarizer(extractedInformation, max_length=int(len(extractedInformation)/6), min_length=int(len(extractedInformation)/10), do_sample=False)
-  return (gr.Textbox.update(output[0]["summary_text"]),gr.Button.update(visible=False),gr.Textbox.update(visible=False),gr.Dropdown.update(visible=False))
+  abstractive_summary = summarizer(extractedInformation, max_length=int(len(extractedInformation)/6), min_length=int(len(extractedInformation)/10), do_sample=False)
+  return (gr.Textbox.update(abstractive_summary[0]["summary_text"]),gr.Button.update(visible=False),gr.Textbox.update(visible=False),gr.Dropdown.update(visible=False))
 
 def Question_Answer(filepath,question,mod):
   extractedInformation=read(filepath)
@@ -152,9 +147,6 @@ def Question_Answer(filepath,question,mod):
   obj=question_answerer(question=question, context=extractedInformation)
   return obj['answer']
 
-def show_fn():
-    return (gr.Textbox.update(visible=True),gr.Button.update(visible=True),gr.Textbox.update(""))
-
 def show_fn():
     return (gr.Textbox.update(visible=True),gr.Button.update(visible=True),gr.Dropdown.update(visible=True),gr.Textbox.update(""))
 def dummy_fn(x):
@@ -166,18 +158,18 @@ with gr.Blocks() as demo:
     img=gr.components.Image(type="filepath", label="Input Image")
 
     with gr.Row():
-        summary = gr.Button(value="Summary")
-        sentence = gr.Button(value="Important Sentences")
-        quesAndAns = gr.Button(value="Question and Answers")
+        summary_btn = gr.Button(value="Summary")
+        sentence_btn = gr.Button(value="Important Sentences")
+        quesAndAns_btn = gr.Button(value="Question and Answers")
 
     mode=gr.Dropdown(["Roberta","DistilBert"],label="Model",info="Choose a model",visible=False)
     ques_box = gr.Textbox(label="Question",info="Enter a Question",interactive=True,visible=False)
-    submit= gr.Button(value="Submit",visible=False)
-    out=gr.Textbox(label="Generated Text")
-    summary.click(fn=summarize,inputs=[img],outputs=[out,submit,ques_box,mode])
-    sentence.click(fn=important_sentences,inputs=[img],outputs=[out,submit,ques_box,mode])
-    quesAndAns.click(fn=show_fn,outputs=[submit,ques_box,mode,out])
-    submit.click(fn=Question_Answer,inputs=[img,ques_box,mode],outputs=[out])
+    submit_btn= gr.Button(value="Submit",visible=False)
+    out_box=gr.Textbox(label="Generated Text")
+    summary_btn.click(fn=summarize,inputs=[img],outputs=[out_box,submit_btn,ques_box,mode])
+    sentence_btn.click(fn=important_sentences,inputs=[img],outputs=[out_box,submit_btn,ques_box,mode])
+    quesAndAns_btn.click(fn=show_fn,outputs=[submit_btn,ques_box,mode,out_box])
+    submit_btn.click(fn=Question_Answer,inputs=[img,ques_box,mode],outputs=[out_box])
     gr.Markdown("## Image Examples")
     with gr.Row():
         gr.Examples(