Update app.py

app.py

@@ -3,6 +3,8 @@ from PyPDF2 import PdfReader
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 from gtts import gTTS
 from io import BytesIO
+import pydub
+from pydub import AudioSegment
 import base64
 import re
 
@@ -41,14 +43,13 @@ def summarize_pdf_abstract(pdf_file):
 
     # Generate audio
     speech = gTTS(text=summary_sentence, lang="en")
-    speech_bytes = BytesIO()
-    speech.write_to_fp(speech_bytes)
 
-    # Encode audio data with Base64
-    audio_data = base64.b64encode(speech_bytes.getvalue()).decode("utf-8")
+    # Save audio file
+    audio_file = AudioSegment.from_file(speech, format="mp3")
+    audio_file.export("summary.mp3", format="mp3")
 
-    # Return individual output values
-    return summary_sentence, audio_data
+    # Return summary and audio filename
+    return summary_sentence, "summary.mp3"
 
   except Exception as e:
     raise Exception(str(e))
@@ -59,4 +60,4 @@ interface = gr.Interface(
   outputs=[gr.Textbox(label="Summary"), gr.Audio()],
 )
 
-interface.launch(share=True)
+interface.launch(share=True)