Added progress bar

app.py

@@ -58,7 +58,7 @@ def validate_questions(questions: str) -> None:
     if len(questions) > MAX_QUESTIONS_COUNT:
         raise gr.Error(f'Validation error. The maximum allowed number of questions is {MAX_QUESTIONS_COUNT}.')
 
-def find_answers(tags: str, questions: str) -> str:
+def find_answers(tags: str, questions: str, progress=gr.Progress()) -> str:
     tags = tags.split('\n')
     questions = questions.split('\n')
 
@@ -68,7 +68,7 @@ def find_answers(tags: str, questions: str) -> str:
     tags_str = ''.join([f'[{tag}]' for tag in tags])
 
     results = []
-    for question in questions:
+    for question in progress.tqdm(questions):
         tagged_question = f'{tags_str} {question}'
         for attempt in range(MAX_ATTEMPS):
             answer = get_answer(tagged_question)
@@ -100,4 +100,4 @@ with gr.Blocks(theme='soft', title='AnswerMate') as blocks:
     outputs = gr.Textbox(label='Output', placeholder='Output will appear here')
     process_button.click(fn=find_answers, inputs=[tags_input, questions_input], outputs=outputs)
 
-blocks.launch()
+blocks.queue(concurrency_count=1).launch()