Update app.py

app.py

@@ -1,50 +1,13 @@
-# Install required libraries
-# !pip install gradio
+from gradio import Interface, Input, Output, Text
 
-import gradio as gr
-import PyPDF2
-from transformers import pipeline
-from gtts import gTTS
-import io
+from utils import summarize_and_speak_pdf_abstract
 
-# Function to extract text from PDF using PyPDF2
-def extract_text_from_pdf(file_contents):
-    with io.BytesIO(file_contents) as file:
-        pdf_reader = PyPDF2.PdfFileReader(file)
-        text = ''
-        for page_num in range(pdf_reader.numPages):
-            text += pdf_reader.getPage(page_num).extractText()
-    return text
-
-# Function to summarize text
-def summarize_text(text):
-    summarizer = pipeline("summarization")
-    summary = summarizer(text, max_length=150, min_length=50, length_penalty=2.0, num_beams=4, early_stopping=True)
-    return summary[0]['summary_text']
-
-# Function to convert text to speech
-def text_to_speech(text, output_path='output.mp3'):
-    tts = gTTS(text, lang='en')
-    tts.save(output_path)
-
-# Custom function to handle file uploads and processing
-def process_file(input_file):
-    file_contents = input_file.read()
-    text = extract_text_from_pdf(file_contents)
-    summary = summarize_text(text)
-    text_to_speech(summary, output_path='output.mp3')
-    return {"text": summary, "audio": "output.mp3"}
-
-# Gradio Interface
-iface = gr.Interface(
-    fn=process_file,
-    inputs=gr.File(),
-    outputs=["text", "audio"],
-    live=True,
-    interpretation="default",
-    title="PDF Abstract Summarizer",
-    description="Summarize the abstract of a PDF file and generate audio.",
+# Define the Gradio interface
+interface = Interface(
+    fn=summarize_and_speak_pdf_abstract,
+    inputs=[Input(type="file", label="Upload PDF")],
+    outputs=[Output(type="text", label="One-sentence Summary")],
 )
 
-# Deploy the interface to Hugging Face Spaces
-iface.share(huggingface_spaces=True)
+# Launch the interface
+interface.launch(title="PDF Abstract Summarizer", description="Summarize the abstract of your PDF in one sentence.")