Update app.py

app.py

@@ -30,7 +30,7 @@ def YTVideoToText(video_link):
         summarized_text2 = ' '.join(summarized_text)
 
     # returning summary
-    return summarized_text2;
+    return [result, summarized_text2];
 
 
 # How to use: postSummaryWithBart("https://ethereum.org/en/what-is-ethereum/")
@@ -80,7 +80,7 @@ def postSummaryWithBart(blog_link):
     text = ''.join([summ['summary_text'] for summ in res])
 
     # returning summary
-    return text;
+    return [sentences, text];
 
 
 # How to use: abstractiveSummaryWithPegasus("""Sample text to be summarized""")
@@ -99,10 +99,12 @@ def abstractiveSummaryWithPegasus(words):
     print(actual_summ)
 
 
-
 # Main logic of the program
 def process(uri, mode):
-    return ["full text","resumed text"] ;
+    if mode == "Youtube"
+        return YTVideoToText(uri)
+    if mode == "Blog"
+        return postSummaryWithBart(uri)
 
 
 import gradio as gr
@@ -122,7 +124,6 @@ with gr.Blocks() as ui:
             TRANSCRIPT = gr.Textbox(
                 label="URI à résumer",
                 lines=10,
-                interactive=False,
                 placeholder="https://youtube|website.ext",
             )
             RESUME = gr.Textbox(
@@ -131,13 +132,18 @@ with gr.Blocks() as ui:
                 interactive=False,
                 placeholder="https://youtube|website.ext",
             )
-            
-        MODE = gr.Radio(choices=["First Choice", "Second Choice", "Third Choice"])
-        gr.Button("Process").click(
-            fn=process,
-            inputs=[URI, MODE],
-            outputs=[TRANSCRIPT, RESUME]
-        )
+        with gr.Column():
+            MODE = gr.Radio(choices=["Youtube", "Blog", "Text"])
+            gr.Button("Process URI").click(
+                fn=process,
+                inputs=[URI, MODE],
+                outputs=[TRANSCRIPT, RESUME]
+            )
+            gr.Button("Process TEXT").click(
+                fn=abstractiveSummaryWithPegasus,
+                inputs=[TRANSCRIPT],
+                outputs=[RESUME]
+            )
     
 #translator_fr = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-fr-en")
 #summarizer = gr.Interface.load("huggingface/sshleifer/distilbart-cnn-12-6")