Bug fix

app.py

@@ -18,13 +18,11 @@ system_prompt = config['SYSTEM_PROMPT']
 use_sage = config['USE_SAGE']
 sage_token = os.environ['SAGE_TOKEN']
 
-client = poe.Client(sage_token)
-
-def get_answer(question: str) -> Dict[str, Any]:
-
+def get_answer(question: str, client: poe.Client=None) -> Dict[str, Any]:
     if use_sage:
         for chunk in client.send_message('capybara', question, with_chat_break=True):
             pass
+        client.delete_message(chunk['messageId'])
         return {
             'status': True,
             'content': chunk['text']
@@ -101,12 +99,15 @@ def find_answers(tags: str, questions: str, progress=gr.Progress()) -> str:
 
     tags_str = ''.join([f'[{tag}]' for tag in tags])
 
+    if use_sage:
+        client = poe.Client(sage_token)
+
     results = []
     for question in progress.tqdm(questions):
         time.sleep(wait_time)
         tagged_question = f'{tags_str} {question}'
         for attempt in range(max_attempts):
-            answer = get_answer(tagged_question)
+            answer = get_answer(tagged_question, client)
             if answer['status']:
                 results.append((question, answer['content']))
                 break