Update load_rock.py

load_rock.py

@@ -1,8 +1,7 @@
 from app_vars import *
 from bs4 import BeautifulSoup as bs
 import random as r
-from urllib.request import urlretrieve
-
+import requests
 
 def get_gallery_image_list_from_rock_id(chosen_rock_id):
     system(
@@ -33,5 +32,8 @@ def url_of_rock(rock_name):
 def get_rock_name_url():
     name = rand_rock_key()
     url = url_of_rock(name)
+    path = f"./assets/img.{url.split()[-1]}"
+    with open(path, "wb") as f:
+        f.write(requests.get(url).content)
     # url, _ = urlretrieve(url)
-    return name, url
+    return name, path