Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import json 
 import requests
 import time
+import pandas as pd 
  
 # AssemblyAI transcript endpoint (where we submit the file)
 transcript_endpoint = "https://api.assemblyai.com/v2/transcript"
@@ -87,13 +88,10 @@ def get_transcript_url(url):
     print(type(res))
     sentiment_analysis_result = ''
 
-    print(res[0])
-    for elt in res:
-      sentiment_analysis_result = sentiment_analysis_result +  "TEXT: "+ elt['text']+ "\n"
-      sentiment_analysis_result = sentiment_analysis_result + "SENTIMENT: " + elt['sentiment'] + "\n"
-      sentiment_analysis_result = sentiment_analysis_result + "CONFIDENCE: " + str(round(float(elt['confidence']), 2)) + "\n"
-  
-    return transcription_result['text'], transcription_result['summary'], sentiment_analysis_result
+    df = pd.DataFrame(res)
+    df = df.loc[:, ["text", "sentiment", "confidence"]]
+    
+    return transcription_result['text'], transcription_result['summary'], df
 
 # def get_transcript_file(filename):
 
@@ -188,7 +186,7 @@ with gr.Blocks() as demo:
         with gr.TabItem('Summary', visible = False) as summary_tab: 
            summary = gr.Textbox(label = "Summary Result")
         with gr.TabItem('Sentiment Analysis', visible = False) as sentiment_tab:
-            sentiment_analysis = gr.Textbox(label = "Sentiment Analysis Result" )
+            sentiment_analysis = gr.Dataframe(label = "Sentiment Analysis Result" )
         
     
     inputs.submit(get_transcript_url, [inputs], [transcript, summary, sentiment_analysis])