# -*- coding: utf-8 -*-
"""app.py.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1zk7xuWSf7ii7zowOqNVLy0FwXYVHYE2V
"""

# https://huggingface.co/spaces/eHemink/assessment3_part2

# Here are the imports
import PyPDF2
import re
import transformers
import scipy
from transformers import pipeline
from bark import SAMPLE_RATE, generate_audio, preload_models
from scipy.io.wavfile import write as write_wav
from IPython.display import Audio
import gradio as gr

# Here is the code
def abstract_to_audio(insert_pdf):
    # Extracting the abstract text from the article pdf
    def extract_abstract(pdf_file):
        # Open the PDF file in read-binary mode
        with open(pdf_file, 'rb') as file:
            # Create a PDF reader object
            pdf_reader = PyPDF2.PdfReader(file)

            # Initialize an empty string to store abstract content
            abstract_text = ''

            # Loop through each page in the PDF
            for page_num in range(len(pdf_reader.pages)):
                # Get the text from the current page
                page = pdf_reader.pages[page_num]
                text = page.extract_text()

                # Use regular expression to find the "Abstract" section
                abstract_match = re.search(r'\bAbstract\b', text, re.IGNORECASE)
                if abstract_match:
                    # Get the text after the "Abstract" heading until the next section, indicated by "Introduction" heading
                    start_index = abstract_match.end()
                    next_section_match = re.search(r'\bIntroduction\b', text[start_index:])
                    if next_section_match:
                        end_index = start_index + next_section_match.start()
                        abstract_text = text[start_index:end_index]
                    else:
                        # If no next section found, extract text till the end
                        abstract_text = text[start_index:]
                    break  # Exit loop once abstract is found

            return abstract_text.strip()


    abstract = extract_abstract(insert_pdf)

    # Creating a summarization pipeline
    model = "lidiya/bart-large-xsum-samsum"
    pipeline1 = pipeline(task = "summarization", model = model)

    # Summarizing the extracted abstract
    summarized = pipeline1(abstract)
    print(summarized[0]['summary_text'])
    tss_prompt = summarized[0]['summary_text']

    # Generate audio file that speaks the generated sentence using Bark
    # download and load all models
    #preload_models()

    # generate audio from text
    #text_prompt = tss_prompt
    #audio_array = generate_audio(text_prompt)
    #return (15500, audio_array)

    tss_pipeline = pipeline("text-to-speech", "suno/bark")
    speech = tss_pipeline(tss_prompt, forward_params={"do_sample": True})
    return (rate=speech["sampling_rate"], data=speech["audio"])






my_app = gr.Interface(fn=abstract_to_audio, inputs='file', outputs='audio', title="PDF Abstract Summarizer",
    description="Extracts abstracts from PDFs and generates audio summaries. This app only accepts PDFs with abstracts.")
my_app.launch()