app updated

wiki_game_st_bs4.py

@@ -23,7 +23,10 @@ def most_similar_sentence(target_topic, labels_list):
     return labels_list[most_similar_index], similarities[most_similar_index], most_similar_index
 
 def get_wikipedia_page(query):
-    response = requests.get(f"https://en.wikipedia.org/wiki/{query}")
+    if "wikipedia" not in query:
+        response = requests.get(f"https://en.wikipedia.org/wiki/{query}")
+    else:
+        response = requests.get(query)
     return response.text
 
 def get_topic_context(page_source):
@@ -54,7 +57,7 @@ def play_wiki_game(starting_topic: str, target_topic: str, limit: int = 100):
 
         if num_pages > 1:
             # load url to new page of next topic
-            page_source = get_wikipedia_page(topic)
+            page_source = get_wikipedia_page(used_links[-1])
 
             # create backup list of links and texts from previous page in case new current page fails
             prev_links_texts = links_texts.pop(loc_idx) # removes the previously used topic
@@ -80,6 +83,7 @@ def play_wiki_game(starting_topic: str, target_topic: str, limit: int = 100):
                         links_texts.append((link_url, link_text))
 
         # st.write(f'links_texts length: {len(links_texts)}')
+        prev_links_texts = []
         if len(links_texts) == 0 and num_pages > 1: # if no links
             links_texts = prev_links_texts
 
@@ -118,10 +122,6 @@ def play_wiki_game(starting_topic: str, target_topic: str, limit: int = 100):
         used_links.append(next_link)
         used_topics.append(topic)
 
-starting_topic = "soulja boy"
-target_topic = "game"
-play_wiki_game(starting_topic = starting_topic, target_topic = target_topic, limit = 100)
-
 # starting_topic = "soulja boy"
-# target_topic = "test"
-# play_wiki_game(starting_topic = starting_topic, target_topic = target_topic, limit = 50)
+# target_topic = "game"
+# play_wiki_game(starting_topic = starting_topic, target_topic = target_topic, limit = 100)