-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
89 lines (75 loc) · 4.01 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import streamlit as st
import QR_Gen
import pdf_create as pdf
from PIL import Image
import style
page_bg_img = style.stylespy() # used for styling the page
# Appname
st.set_page_config(page_title="Child Safety AI QR 🖌️", layout="wide")
st.markdown(page_bg_img, unsafe_allow_html=True)
st.markdown("<h1 style='text-align: center; color: #fff;'>Child Safety AI QR 🖌️</h1>", unsafe_allow_html=True)
# Split the page into two columns
col1, col2 = st.columns(2, gap='medium')
# Form UI in the first column
with st.form(key="form1"):
with col1:
uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
Name = st.text_input('Enter First name')
Address = st.text_input('Enter your address')
Birthday = st.date_input('Your birthday')
Gender = st.radio('Gender', ['Male', 'Female', 'Others'])
Contact = st.text_input('Enter your contact number')
anothercontact = st.text_input('Enter another contact number')
Contact_Email = st.text_input('Enter Contact email address')
FathersName = st.text_input("Enter Father's name")
with col2:
MothersName = st.text_input('Enter Mother"s name')
SchoolName = st.text_input('Enter your school name')
SchoolAddress = st.text_input('Enter your school address')
city = st.text_input('City')
state = st.text_input('State')
ZipCode = st.text_input('Zip code')
Country = st.text_input('Country')
Blood_Group = st.selectbox('Blood Group', ['A+', 'A-', 'B+', 'B-', 'AB+', 'AB-', 'O+', 'O-'])
Identification_mark = st.text_input('Identification mark')
Allergenes = st.text_input('Allergenes')
# Controlnet UI in the second column
with col2:
st.title("AI Configuration")
input_choice = st.radio("Select Input Type", ["Dropdown", "Text Field"])
p_prompt = ""
if input_choice == "Dropdown":
dropdown_choice = st.selectbox("Select promopt", [
"Sky view of highly aesthetic, ancient greek thermal baths in beautiful nature",
"Bright sunshine coming through the cracks of a wet, cave wall of big rocks",
"A sky view of colorful lakes and rivers flowing through the desert",
"Concept art: A surreal sight of a crystal-clear lake in the middle of the desert, reflecting the stars of the Milky Way. The juxtaposition of the arid landscape and the tranquil waters adds a dream-like quality to the scene",
"Sunlight filtering through a dense forest, creating a magical glow",
"In the heart of the tranquil woods, nature's embrace reveals a realm of enchanting radiance",
"Blended"
])
p_prompt = dropdown_choice
st.write("Selected:", dropdown_choice)
else:
text_input = st.text_input("Enter your prompt here")
p_prompt = text_input
st.write("Entered:", text_input)
seed = st.slider('Seed value', -1, 9999999999, 9635874126)
genarate = st.form_submit_button(label="Generate", use_container_width=True)
# QR code generation
if genarate:
st.subheader("QR Code Generated")
if uploaded_file is None:
st.error("Please upload an image to proceed.")
else:
save_path = "uploaded_image.png"
pil_image = Image.open(uploaded_file)
pil_image.save(save_path)
st.success("Please wait while we generate your QR code, it will take some time")
fileURL = str(pdf.create_qr_code_pdf(Name, Birthday, FathersName, MothersName, Address, Gender, Contact,
anothercontact, Contact_Email, SchoolName, SchoolAddress, city, state,
ZipCode, Country, Blood_Group, Identification_mark, Allergenes))
genimg = QR_Gen.generate_qr_code(fileURL, p_prompt, seed)
image = Image.open(genimg)
st.image(image, caption='Generated QR code')
st.success("QR code generated successfully")