Update app.py

app.py

@@ -23,7 +23,7 @@ def _read_file(filename, chunk_size=5242880):
                 break
             yield data
             
-def get_transcript_url(url):
+def get_transcript_url(url, audio_intelligence_options):
     
     # JSON that tells the API which file to trancsribe
     json={
@@ -82,7 +82,9 @@ def get_transcript_url(url):
         raise RuntimeError(f"Transcription failed: {transcription_result['error']}")
       else:
         time.sleep(3)
-    return transcription_result['text']
+
+    
+    return transcription_result['text'], transcription_result['summary']
 
 def get_transcript_file(filename):
 
@@ -99,7 +101,7 @@ def get_transcript_file(filename):
     # Turn on speaker labels
     "speaker_labels": True,
 
-    # Turn on cusom vocabulary
+    # Turn on custom vocabulary
     "word_boost": ["assembly ai"],
 
     # Turn on custom spelling
@@ -151,20 +153,24 @@ def get_transcript_file(filename):
         time.sleep(3)
     return transcription_result['text']
 
+audio_intelligence_list = [
+    "Summarization",
+    "Sentiment Analysis"
+]
 
 title = """<h1 align="center">🔥Conformer-1 API </h1>"""
 description = """
 ### In this demo, you can explore the outputs of a Conformer-1 Speech Recognition Model from AssemblyAI.
 """
                 
-with gr.Blocks(css = """#col_container {width: 1000px; margin-left: auto; margin-right: auto;}
-                """) as demo:
+with gr.Blocks() as demo:
     gr.HTML(title)
     gr.Markdown(description)
 
     with gr.Column(elem_id = "col_container"):
         with gr.Tab("Audio URL file"):
-            inputs = gr.Textbox(label = "Enter the url for the audio file")        
+            inputs = gr.Textbox(label = "Enter the url for the audio file")
+            audio_intelligence_options = gr.CheckboxGroup(audio_intelligence_list, label="Audio Intelligence Options")
             b1 = gr.Button('Transcribe')
         
         with gr.Tab("Upload Audio as File"):
@@ -172,12 +178,14 @@ with gr.Blocks(css = """#col_container {width: 1000px; margin-left: auto; margin
             transcribe_audio_u = gr.Button('Transcribe')
 
         transcript = gr.Textbox(label = "Transcript Result" )
+        summary = gr.Textbox(label = "Summary Result" )
+        
     
-    inputs.submit(get_transcript_url, [inputs], [transcript])
-    b1.click(get_transcript_url, [inputs], [transcript])
+    inputs.submit(get_transcript_url, [inputs, audio_intelligence_options], [transcript, summary])
+    b1.click(get_transcript_url, [inputs, audio_intelligence_options], [transcript])
     transcribe_audio_u.click(get_transcript_file, [audio_input_u], [transcript])
 
-    examples = gr.Examples(examples = [['https://bit.ly/3yxKEIY']], inputs = inputs, outputs=transcript, cache_examples = True, fn = get_transcript_url)
+    #examples = gr.Examples(examples = [['https://bit.ly/3yxKEIY']], inputs = inputs, outputs=transcript, cache_examples = True, fn = get_transcript_url)
                     
     
     demo.queue().launch(debug=True)