import gradio as gr
import requests
import re
import fitz

def extract_text_from_pdf(pdf_file_path):
    doc = fitz.open(pdf_file_path) 
    text = ""
    for page in doc:
        text+=page.get_text() 

    return text

API_URL = "https://api-inference.huggingface.co/models/potsawee/t5-large-generation-squad-QuestionAnswer"
headers = {"Authorization": "Bearer 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"}

def query(payload):
	response = requests.post(API_URL, headers=headers, json=payload)
	return response.json()
    
def generate_question_answer_pairs(input_file):
    if input_file is None:
        return "Please upload a file"

    pdf_text = extract_text_from_pdf(input_file)

    sentences = re.split(r'(?<=[.!?])', pdf_text)
    outputs = [] 

    for sentence in sentences:
        if sentence.strip():
            output = query({
                "inputs": sentence,
            })
            outputs.append(output)
            
    return outputs

title = "Question-Answer Pairs Generation"
input_file = gr.File(label="Upload a PDF file")
output_text = gr.Textbox()

interface = gr.Interface(
    fn=generate_question_answer_pairs,
    inputs=input_file,
    outputs=output_text,
    title=title,
)
interface.launch()