Update app.py

app.py

@@ -100,16 +100,7 @@ def abstractiveSummaryWithPegasus(words):
     return actual_summ
 
 
-# Main logic of the program
-def process(uri, mode):
-    if mode == "Youtube":
-        return YTVideoToText(uri)
-    elif mode == "Blog":
-        return postSummaryWithBart(uri)
-    else:
-        raise ValueError("Invalid mode")
-
-
+# Main UI
 with gr.Blocks() as ui:
     gr.Markdown("""
     ## Permet de faire le résumé d'une video youtube ou d'un article de blog
@@ -136,14 +127,15 @@ with gr.Blocks() as ui:
                 api_name="resume"
             )
         with gr.Column():
-            MODE = gr.Radio(
-                label="Mode pour URI", 
-                choices=["Youtube", "Blog"],
-                api_name="mode"
+            gr.Button("Process Youtube").click(
+                fn=YTVideoToText,
+                inputs=[URI],
+                outputs=[TRANSCRIPT, RESUME],
+                api_name="process_uri"
             )
-            gr.Button("Process URI").click(
-                fn=process,
-                inputs=[URI, MODE],
+            gr.Button("Process HTML").click(
+                fn=postSummaryWithBart,
+                inputs=[URI],
                 outputs=[TRANSCRIPT, RESUME],
                 api_name="process_uri"
             )