Update app.py

app.py

@@ -34,10 +34,31 @@ from pdfminer.high_level import extract_text
 def pdf_to_text(file_obj):
    text = extract_text(file_obj.name)
    
+   output_text= bert_legal_model(text,  min_length = 8, ratio = 0.05)
+   output_text = output_text.replace('   ',' ')
+   output_text = output_text .replace(',.',',')
+   output_text = output_text .replace('\n','  ')
+   output_text = output_text .replace('..','.')
    
-   myobj = gTTS(text=text, lang='en', slow=False)
-   myobj.save("test.wav")
-   return 'test.wav'
+   pdf = FPDF()  
+
+  # Add a page
+   pdf.add_page()
+
+   pdf.set_font("Times", size = 12)
+
+  # open the text file in read mode
+   f = output_text
+  # insert the texts in pdf
+   pdf.multi_cell(190, 10, txt = f, align = 'C')
+  # save the pdf with name .pdf
+   pdf.output("legal.pdf")  
+   all_text
+   
+   myobj = gTTS(text=output_text, lang='en', slow=False)
+   myobj.save("legal.wav")
+   
+   return 'legal.wav', output_text, "legal.pdf"
 
 
     
@@ -46,7 +67,7 @@ def pdf_to_text(file_obj):
   #  return path 
     #pageObject.extractText() 
 iface = gr.Interface(fn = pdf_to_text,
-   inputs =  "file", outputs="audio" )
+   inputs =  "file", outputs=["audio","text", "file"] )
 
 if __name__ == "__main__":
     iface.launch(share=True)