add duration field

app.py

@@ -21,6 +21,7 @@ from tenacity import retry, retry_if_exception_type
 
 import pdfplumber
 import concurrent.futures
+from pydub import AudioSegment
 # from docx import Document
 # from pptx import Presentation
 
@@ -244,12 +245,14 @@ def generate_audio(file=None, url=None, openai_api_key: str = None) -> bytes:
     temporary_file.write(audio)
     temporary_file.close()
 
+    audio_segment = AudioSegment.from_file(temporary_file.name)
+    duration = len(audio_segment) / 1000.0  # duration in seconds
     # Delete any files in the temp directory that end with .mp3 and are over a day old
     for file in glob.glob(f"{temporary_directory}*.mp3"):
         if os.path.isfile(file) and time.time() - os.path.getmtime(file) > 24 * 60 * 60:
             os.remove(file)
 
-    return temporary_file.name, transcript
+    return temporary_file.name, transcript, duration
 
 
 demo = gr.Interface(

requirements.txt

@@ -17,3 +17,4 @@ langchain_openai==0.1.8
 langchain_community==0.2.4
 langchain==0.2.5
 beautifulsoup4==4.12.3
+pydub