Update app.py

app.py

@@ -65,7 +65,7 @@ def convert_to_wav(audio_file):
     return wav_file
 
     
-@spaces.GPU(duration=120, queue=False)
+@spaces.GPU(duration=21600, queue=False)
 def transcribe_audio(audio_file, batch_size=4):
     start_time = time.time()
     # Convert .m4a to .wav
@@ -128,25 +128,6 @@ image_html = """
 </div>
 """
 
-# Gradio UI
-iface = gr.Blocks()
-
-with iface:
-    gr.HTML(image_html)
-    gr.Markdown("# Upload an audio file to get the transcription")
-    audio_input = gr.Audio(type="filepath")
-    batch_size_input = gr.Slider(minimum=1, maximum=16, step=1, label="Batch Size")
-    transcription_output = gr.Textbox("Transcription | nb-whisper-large-semantic")
-    summary_output = gr.Textbox("Summary | TextRank, graph-based")
-    transcribe_button = gr.Button("Transcribe and Summarize")
-
-    def transcribe_and_summarize(audio_file, batch_size):
-        transcription, result = transcribe_audio(audio_file, batch_size)
-        summary = summarize_text(transcription)
-        return result, summary
-
-    transcribe_button.click(fn=transcribe_and_summarize, inputs=[audio_input, batch_size_input], outputs=[transcription_output, summary_output])
-    
 def save_to_pdf(transcription, summary):
     pdf = FPDF()
     pdf.add_page()
@@ -165,5 +146,36 @@ def save_to_pdf(transcription, summary):
     pdf.output(pdf_output_path)
     return pdf_output_path
 
+# Gradio UI
+iface = gr.Blocks()
+
+with iface:
+    gr.HTML(image_html)
+    gr.Markdown("# Upload an audio file to get the transcription")
+    audio_input = gr.Audio(type="filepath")
+    batch_size_input = gr.Slider(minimum=7, maximum=16, step=1, label="Batch Size")
+    transcription_output = gr.Textbox("Transcription | nb-whisper-large-semantic")
+    summary_output = gr.Textbox("Summary | TextRank, graph-based")
+    pdf_output = gr.File(label="Download PDF")
+    transcribe_button = gr.Button("Transcribe and Summarize")
+
+    def transcribe_and_summarize(audio_file, batch_size):
+        transcription, result = transcribe_audio(audio_file, batch_size)
+        summary = summarize_text(transcription)
+        pdf_path = save_to_pdf(transcription, summary)
+        return result, summary, pdf_path
+
+    transcribe_button.click(
+        fn=transcribe_and_summarize,
+        inputs=[audio_input, batch_size_input],
+        outputs=[transcription_output, summary_output, pdf_output]
+    )
+
+# run
+iface.launch(share=True, debug=True)
+
+    
+
+
 # run
 iface.launch(share=True, debug=True)