Update app.py

app.py

@@ -36,8 +36,6 @@ def get_answer(query,context):
     return res
 
 
-def update(name):
-    return f"Welcome to Gradio, {name}!"
 
 with gr.Blocks() as demo:
     gr.Markdown("<h1><center>Youtube-QA</center></h1>")
@@ -48,11 +46,21 @@ with gr.Blocks() as demo:
                     B -->C(Trascription)
                     C -->|Query| D(QA-model)
                     D -->E[Answer]
-""")
-    with gr.Row():
-        inp = gr.Textbox(placeholder="What is your name?")
-        out = gr.Textbox()
-    btn = gr.Button("Run")
-    btn.click(fn=update, inputs=inp, outputs=out)
+                    """)
+    with gr.Column():
+        with gr.Row():
+            in_yt = gr.Textbox(lines=1, placeholder="Enter Youtube URL", label="YouTube URL"))
+        transcribe_btn = gr.Button("Trascribe")
+    with gr.Column():
+        with gr.Row():
+            out_yt = ["html", "text"],
+    with gr.Column():
+        with gr.Row():
+            in_query = gr.Textbox(lines=1, placeholder="What's your Question", label="Query"))
+            ans_btn = gr.Button("Answer")
+            out_query = ["text"]
+            
+    ans_btn.click(fn=get_answer, inputs=[in_query,out_yt["text"]], outputs=out_query)
+    transcribe_btn.click(fn=yt_transcribe, inputs=in_yt, outputs=out_yt)
 
 demo.launch()