Update app.py

app.py

@@ -124,7 +124,7 @@ def summarize_text(text):
 # HTML syntax for imagery
 image_html = """
 <div style="text-align: center;">
-    <img src="https://huggingface.co/spaces/camparchimedes/ola_s-audioshop/raw/main/picture.png" alt="Banner" width="auto" height="auto">
+    <img src="https://huggingface.co/spaces/camparchimedes/ola_s-audioshop/raw/main/picture.png" alt="Banner" width="100%" height="auto">
 </div>
 """
 
@@ -151,31 +151,84 @@ iface = gr.Blocks()
 
 with iface:
     gr.HTML(image_html)
-    gr.Markdown("# Vi har nå muligheten til å oversette lydfiler til norsk skrift. Denne fungerer på norsk og oversetter norsk tale til tekst.")
-    audio_input = gr.Audio(type="filepath")
-    batch_size_input = gr.Slider(minimum=7, maximum=16, step=1, label="Batch Size")
-    transcription_output = gr.Textbox(label="Transcription | nb-whisper-large-semantic")
-    summary_output = gr.Textbox(label="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]
-    )
+    gr.Markdown("# Vi har nå muligheten til å oversette lydfiler til norsk skrift.")
+    
+    with gr.Tabs():
+        
+        # First Tab: Transcription
+        with gr.TabItem("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(label="Transcription | nb-whisper-large-semantic")
+            result_output = gr.Textbox(label="Time taken and Number of words")
+            transcribe_button = gr.Button("Transcribe")
+
+            def transcribe(audio_file, batch_size):
+                transcription, result = transcribe_audio(audio_file, batch_size)
+                return transcription, result
+
+            transcribe_button.click(
+                fn=transcribe,
+                inputs=[audio_input, batch_size_input],
+                outputs=[transcription_output, result_output]
+            )
 
-# run
-iface.launch(share=True, debug=True)
+        # Second Tab: Summary
+        with gr.TabItem("Summary"):
+            summary_output = gr.Textbox(label="Summary | TextRank, graph-based")
+            summarize_button = gr.Button("Summarize")
+
+            def summarize(transcription):
+                if not transcription:
+                    return "Warning: a transcription must be available."
+                summary = summarize_text(transcription)
+                return summary
+
+            summarize_button.click(
+                fn=summarize,
+                inputs=[transcription_output],  # Use the transcription from the first tab
+                outputs=summary_output
+            )
 
-    
+        # Third Tab: PDF Download Options
+        with gr.TabItem("Download PDF"):
+            pdf_transcription_only = gr.Button("Download PDF with Transcription Only")
+            pdf_summary_only = gr.Button("Download PDF with Summary Only")
+            pdf_both = gr.Button("Download PDF with Both")
+
+            pdf_output_transcription_only = gr.File(label="Download PDF")
+            pdf_output_summary_only = gr.File(label="Download PDF")
+            pdf_output_both = gr.File(label="Download PDF")
+
+            def generate_pdf_transcription_only(transcription):
+                return save_to_pdf(transcription, "")
+
+            def generate_pdf_summary_only(summary):
+                return save_to_pdf("", summary)
+
+            def generate_pdf_both(transcription, summary):
+                return save_to_pdf(transcription, summary)
+
+            pdf_transcription_only.click(
+                fn=generate_pdf_transcription_only,
+                inputs=[transcription_output],
+                outputs=[pdf_output_transcription_only]
+            )
+
+            pdf_summary_only.click(
+                fn=generate_pdf_summary_only,
+                inputs=[summary_output],
+                outputs=[pdf_output_summary_only]
+            )
+
+            pdf_both.click(
+                fn=generate_pdf_both,
+                inputs=[transcription_output, summary_output],
+                outputs=[pdf_output_both]
+            )
 
 
 # run
 iface.launch(share=True, debug=True)
+
+