Update app.py

app.py

@@ -53,13 +53,14 @@ def get_wiki_article(topic):
         choices = [x for x in e.options if ('disambiguation' not in x) and ('All pages' not in x) and (x!=topic)]
         s = random.choice(choices)
         p = wikipedia.page(s)
-        saved = AIMemory(p, s)
+        saved = AIMemory(p.content, p.url)
     return p.content, p.url
 
 def get_answer(topic, question):
     w_art, w_url=get_wiki_article(topic)
     qa = {'question': question, 'context': w_art}
     res = nlp(qa)
+    saved = AIMemory(res['answer'], w_url)
     return res['answer'], w_url, {'confidence':res['score']}