Fixed download() to save file with reasonable filenames

dataLoad.py

@@ -57,9 +57,9 @@ def download(howhard, index):
     """
     r = sess.get(f'http://s0urce.io/client/img/word/{howhard}/{index}')
     if(r.status_code == 200):
-        with open(f'in/{index}.png', 'wb') as f:
+        with open(f'in/{howhard}_{index}.png', 'wb') as f:
             f.write(r.content)
-        return f'{index}.png ' + str(r.status_code)
+        return f'{howhard}_{index}.png ' + str(r.status_code)
     return r.status_code
 
 def img_point(x, y):
@@ -130,9 +130,9 @@ def save_image():
 
     return result
 
-def run(VAL):
+def run(difficult):
     for x in range(0, 70):
-        fname = f'in/{x}.png'
+        fname = f'in/{difficult}_{x}.png'
 
         try:
             with open(fname, 'r') as f:
@@ -164,40 +164,38 @@ def run(VAL):
         for l in img_s:
             print(l)
         img_s.append(input('Word: '))
-        y = x + VAL
-        with open(f'{DIR}/{y}.txt', 'w') as f:
+        with open(f'{DIR}/{difficult}_{x}.txt', 'w') as f:
             for i_s in img_s:
                 f.write(f'{i_s}\n')
 
-        print(f"Image saved to '{DIR}/{y}.txt' in byte string")
+        print(f"Image saved to '{DIR}/{difficult}_{x}.txt' in byte string")
         os.remove(f'{fname}')
         print(f"File '{fname}' automatically removed")
 
 if (DOWNLOAD == True):
     print("Downloading s0urce.io Words")
     print("EASY")
-    time.sleep(5)
+    # time.sleep(5)
     for e in range(0, 62):
         print(download('e', e))
         # time.sleep(random.randint(10, 15))
 
-    run(0)
-
     print("MEDIUM")
     # time.sleep(5)
     for m in range(0, 66):
         print(download('m', m))
         # time.sleep(random.randint(10, 15))
 
-    run(62)
-
     print("HARD")
     # time.sleep(5)
     for h in range(0, 55):
         print(download('h', h))
         # time.sleep(random.randint(10, 15))
 
-    run(128)
+    # Img Processing
+    run('e') # Answer the questions
+    run('m')
+    run('h')
 
 # ----------------------------------------------------------------------------------------
 # All below was in a seperate dataJS.py file... but now I have fixed it so it's 1 script!