Update app.py

app.py

@@ -59,7 +59,8 @@ def main():
         with model_1:
             language = st.selectbox('Qual a linguagem do seu modelo?', ('Português (pt)', 'Inglês (en)', 'Outra'))
             link = st.text_area(label="Coloque o link do seu vídeo do YouTube!", height=25, placeholder="Digite seu link...")
-            submit_1 = st.button('Gerar sumário!')
+            questions = st.text_area(label="Coloque suas perguntas separadas por vírgula!", height=50, placeholder="Digite suas perguntas...").split(", ")
+            submit_1 = st.button('Gerar soluções!')
 
         with model_2:
             if submit_1:
@@ -70,32 +71,15 @@ def main():
     
                     elif language == 'Inglês (en)':
                         outputs = english_sum_pipeline(link)
+                        answers = english_qa_pipeline(question, outputs[0])
                     
                     else:
                         st.write("Erro na seleção de linguagem.")
 
-                st.write("Sumário.....................................................................: \n {} \n \n".format(outputs[1]))
-
-    with qa:
-        st.header("Modelo para resposta de perguntas")
-        
-        with qa_1:
-            question = st.text_area(label="Digite sua pergunta para o modelo", height=25, placeholder="Digite sua pergunta...")
-            submit_2 = st.button('Gerar resposta!')
-
-        with qa_2:
-            with st.spinner('Wait for it...'):
-                if submit_2:
-                    if language == 'Português (pt)':
-                        #answer = portuguese_qa_pipeline(link)
-                        st.write("Modelo ainda não implementado.")
-    
-                    elif language == 'Inglês (en)':
-                        answer = english_qa_pipeline(question, outputs[0])
+                    st.write("Sumário.....................................................................: \n {} \n \n".format(outputs[1]))
+                    st.write("Resposta....................................................................: \n")
                     
-                    else:
-                        st.write("Erro na seleção de linguagem.")
-
-                    st.write("Resposta.....................................................................: \n {} \n \n".format(answer))
+                    for i in range(len(answers)):
+                        st.write(questions[i] + ": " + answers[i])
 
 main()