Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 import os
 import random
 import faker
+import datetime
 
 # Setting up API endpoints and headers
 face_api_url = "https://api-inference.huggingface.co/models/stablediffusionapi/realvis-xl-v40"
@@ -32,7 +33,7 @@ def generate_random_surname():
 # Function to generate a random birth date
 def generate_random_birth_date():
     fake = faker.Faker()
-    return fake.date_of_birth().strftime("%m/%d/%Y")
+    return fake.date_of_birth(minimum_age=18, maximum_age=80).strftime("%m/%d/%Y")
 
 # Function to generate a random 6-digit number for fingerprint generation
 def generate_random_number():
@@ -59,8 +60,14 @@ st.write(f"Name: {name}")
 st.write(f"Surname: {surname}")
 st.write(f"Birth Date: {birth_date}")
 
+# Calculate age based on birth year
+birth_year = int(birth_date[-4:])
+current_year = datetime.datetime.now().year
+age = current_year - birth_year
+
+st.write(f"Age: {age}")
+
 # Generate face image based on age
-age = 30
 st.write(f"Photo of {age} years old {gender} with serious face:")
 face_image = generate_face_image(age, gender)
 st.image(face_image, caption=f"{age} years old {gender}")