Upload 2 files

app.py

@@ -1,8 +1,8 @@
-from webbrowser import open as open_url
 from load_rock import get_rock_name_url
 from app_vars import *
 from nltk import edit_distance
 import gradio as g
+from urllib.request import urlretrieve
 
 state = g.State(
     {a: b for a, b in (zip(["name", "url"], get_rock_name_url()))}
@@ -62,7 +62,6 @@ Total | {state.value["correct"] + state.value["wrong"] + state.value["skipped"]}
     else:
         out = ans
         stats = feedback
-    print(state.value["url"])
     return (
         state.value["url"],
         out,

load_rock.py

@@ -1,19 +1,21 @@
 from app_vars import *
 from bs4 import BeautifulSoup as bs
 import random as r
+from urllib.request import urlretrieve
 
 
 def get_gallery_image_list_from_rock_id(chosen_rock_id):
     system(
-        f'{curl_invoker} -A "Mozilla/5.0 (platform; rv:geckoversion) Gecko/geckotrail Firefox/firefoxversion" -o "{rs}/app_data/html/index.html" https://www.mindat.org/gm/{chosen_rock_id}')
+        f'{curl_invoker} -A "Mozilla/5.0 (platform; rv:geckoversion) Gecko/geckotrail Firefox/firefoxversion" -o "{rs}/app_data/html/index.html" https://www.mindat.org/gm/{chosen_rock_id}'
+    )
 
     with open(f"{rs}\\app_data\\html\\index.html", "rb") as f:
-        soup = bs(f, 'html.parser')
+        soup = bs(f, "html.parser")
 
     image_list = []
-    for i in (soup.find_all("img")):
+    for i in soup.find_all("img"):
         image_list.append("https://www.mindat.org" + i.get("src"))
-    return (image_list)
+    return image_list
 
 
 rl = load(open(f"{rs}/app_data/rock_urls_list.json"))
@@ -31,4 +33,5 @@ def url_of_rock(rock_name):
 def get_rock_name_url():
     name = rand_rock_key()
     url = url_of_rock(name)
+    url, _ = urlretrieve(url)
     return name, url