vancauwe commited on
Commit
8810ca7
·
1 Parent(s): 11e705e

fix: generate random md5 hash if None on image

Browse files
Files changed (1) hide show
  1. src/input_handling.py +9 -1
src/input_handling.py CHANGED
@@ -25,6 +25,14 @@ both the UI elements (setup_input_UI) and the validation functions.
25
  #allowed_image_types = ['webp']
26
  allowed_image_types = ['jpg', 'jpeg', 'png', 'webp']
27
 
 
 
 
 
 
 
 
 
28
 
29
  # autogenerated class to hold the input data
30
  class InputObservation:
@@ -101,7 +109,7 @@ class InputObservation:
101
  return {
102
  #"image": self.image,
103
  "image_filename": self.uploaded_filename.name if self.uploaded_filename else None,
104
- "image_md5": hashlib.md5(self.uploaded_filename.read()).hexdigest() if self.uploaded_filename else None,
105
  "latitude": self.latitude,
106
  "longitude": self.longitude,
107
  "author_email": self.author_email,
 
25
  #allowed_image_types = ['webp']
26
  allowed_image_types = ['jpg', 'jpeg', 'png', 'webp']
27
 
28
+ import random
29
+ import string
30
+ def generate_random_md5():
31
+ # Generate a random string
32
+ random_string = ''.join(random.choices(string.ascii_letters + string.digits, k=16))
33
+ # Encode the string and compute its MD5 hash
34
+ md5_hash = hashlib.md5(random_string.encode()).hexdigest()
35
+ return md5_hash
36
 
37
  # autogenerated class to hold the input data
38
  class InputObservation:
 
109
  return {
110
  #"image": self.image,
111
  "image_filename": self.uploaded_filename.name if self.uploaded_filename else None,
112
+ "image_md5": hashlib.md5(self.uploaded_filename.read()).hexdigest() if self.uploaded_filename else generate_random_md5(),
113
  "latitude": self.latitude,
114
  "longitude": self.longitude,
115
  "author_email": self.author_email,