Spaces:
Sleeping
Sleeping
MartinSumera
commited on
Commit
•
12cd134
1
Parent(s):
b82213b
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,106 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import json
|
2 |
+
import gradio as gr
|
3 |
+
from validate import StyleImageDatabase
|
4 |
+
|
5 |
+
JSON_FILE_NAME = 'validate_results_images.json'
|
6 |
+
|
7 |
+
def get_grouped_images_from_database():
|
8 |
+
database = StyleImageDatabase()
|
9 |
+
images = database.get_all_style_images()
|
10 |
+
grouped_images = {}
|
11 |
+
|
12 |
+
for image in images:
|
13 |
+
style_id = image[1]
|
14 |
+
style_name = image[2]
|
15 |
+
request_image_url = image[3]
|
16 |
+
result_image_url = image[4]
|
17 |
+
|
18 |
+
if style_id not in grouped_images:
|
19 |
+
grouped_images[style_id] = {'style_name': style_name, 'images': []}
|
20 |
+
|
21 |
+
image_data = {'result_image_url': result_image_url, 'request_image_url': request_image_url}
|
22 |
+
grouped_images[style_id]['images'].append(image_data)
|
23 |
+
|
24 |
+
return grouped_images
|
25 |
+
|
26 |
+
def save_images_to_json_file(grouped_images):
|
27 |
+
with open(JSON_FILE_NAME, 'w') as json_file:
|
28 |
+
json.dump(grouped_images, json_file, indent=4)
|
29 |
+
|
30 |
+
def load_images_from_json_file():
|
31 |
+
with open(JSON_FILE_NAME, 'r') as json_file:
|
32 |
+
return json.load(json_file)
|
33 |
+
|
34 |
+
def get_test_images():
|
35 |
+
return {
|
36 |
+
"Lukas": "https://cc.cz/wp-content/uploads/2020/10/lukas_strandel_ceo_futured_01-1.jpg",
|
37 |
+
"Jana": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-10-36-19-IMG_7058 2 1.jpg",
|
38 |
+
"Honzík1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-10-33-39-IMG_20210401_210045_869.jpg",
|
39 |
+
"Radek1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-27-12-34-30-IMG_0422.jpg",
|
40 |
+
"Radek2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-08-31-18-radek_selfie.jpeg",
|
41 |
+
"Bezo1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-10-47-10-IMG_3670 1.jpeg",
|
42 |
+
"Bezo2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-10-38-40-IMG_3845.jpeg",
|
43 |
+
"David": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-10-38-40-56798776 1.jpg",
|
44 |
+
"Kalman": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-08-31-18-kalman.jpg",
|
45 |
+
"Lenka": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-08-31-18-lenka.jpg",
|
46 |
+
"Kotal": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-08-31-18-kotal.png",
|
47 |
+
"Sumera1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-09-02-18-martin.jpg",
|
48 |
+
"Sumera2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-08-31-18-martin_sucks.jpg",
|
49 |
+
"Lada": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-03-28-08-31-18-lada.jpg",
|
50 |
+
"Vladka1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-08-8264DE74-B9CE-477C-8883-CD5699C41B45.JPG",
|
51 |
+
"Radek3": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-10-IMG_0558.jpg",
|
52 |
+
"Pollak": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-13-PXL_20211217_185315324.jpg",
|
53 |
+
"Martina1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-19-IMG_7157.jpg",
|
54 |
+
"Vox1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-23-1000044348.jpg",
|
55 |
+
"Sarka1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-26-IMG_6809.jpg",
|
56 |
+
"Ruda1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-30-IMG_0104.JPG",
|
57 |
+
"Matsem": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-32-PXL_20240412_133830638.MP.jpg",
|
58 |
+
"Martina2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-36-IMG_6007.jpg",
|
59 |
+
"Radek4": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-41-IMG_9661.jpg",
|
60 |
+
"Matej G. 1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-44-IMG_8423.jpg",
|
61 |
+
"Marie1": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-45-IMG_2294.jpg",
|
62 |
+
"Sarka2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-46-IMG_7474.jpg",
|
63 |
+
"Matej G. 2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-51-IMG_6556.jpg",
|
64 |
+
"Ruda2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-52-IMG_0586.jpg",
|
65 |
+
"Marie2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-55-IMG_0381.jpg",
|
66 |
+
"Vox2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-56-1000052742.jpg",
|
67 |
+
"Vladka2": "https://mdevcamp-ai-upload-script.s3.amazonaws.com/images/2024-04-15-10-50-59-51E641DC-CEB1-4786-BE5B-56D443790907.JPG",
|
68 |
+
"Honzik2": "https://storage.googleapis.com/mdevcamp-avatars/source/af146193-f350-11ee-9eb3-4200a9fe0102",
|
69 |
+
"Honzik3": "https://storage.googleapis.com/mdevcamp-avatars/source/9386c6f4-f654-11ee-9eb3-4200a9fe0102",
|
70 |
+
"Simon1": "https://storage.googleapis.com/mdevcamp-avatars/source/bcd9af2b-f6ac-11ee-9eb3-4200a9fe0102",
|
71 |
+
"Simon2": "https://storage.googleapis.com/mdevcamp-avatars/source/f798bc1b-f70d-11ee-9eb3-4200a9fe0102",
|
72 |
+
"Simon3": "https://storage.googleapis.com/mdevcamp-avatars/source/b922a19c-f70a-11ee-9eb3-4200a9fe0102",
|
73 |
+
"Patrik": "https://storage.googleapis.com/mdevcamp-avatars/source/9c178993-f71e-11ee-9eb3-4200a9fe0102",
|
74 |
+
"Patrik2": "https://storage.googleapis.com/mdevcamp-avatars/source/f3bd8696-f71a-11ee-9eb3-4200a9fe0102",
|
75 |
+
"Patrik3": "https://storage.googleapis.com/mdevcamp-avatars/source/af3b30ef-f713-11ee-9eb3-4200a9fe0102",
|
76 |
+
}
|
77 |
+
|
78 |
+
if __name__ == "__main__":
|
79 |
+
# database = StyleImageDatabase()
|
80 |
+
# grouped_images = get_grouped_images_from_database()
|
81 |
+
# save_images_to_json_file(grouped_images)
|
82 |
+
|
83 |
+
loaded_images = load_images_from_json_file()
|
84 |
+
|
85 |
+
test_images_with_names = get_test_images()
|
86 |
+
inverted_test_images_with_names = {v: k for k, v in test_images_with_names.items()}
|
87 |
+
|
88 |
+
with gr.Blocks() as demo:
|
89 |
+
for style_id, data in loaded_images.items():
|
90 |
+
with gr.Accordion(label=f"Style {data['style_name']} - {style_id}", open=False):
|
91 |
+
with gr.Row():
|
92 |
+
images = [(image['result_image_url'], inverted_test_images_with_names.get(image['request_image_url'], image['request_image_url'])) for image in data['images']]
|
93 |
+
gr.Gallery(
|
94 |
+
label="Results",
|
95 |
+
value=images,
|
96 |
+
show_label=False,
|
97 |
+
elem_id="gallery",
|
98 |
+
columns=[5],
|
99 |
+
rows=[10],
|
100 |
+
object_fit="contain",
|
101 |
+
height="auto",
|
102 |
+
)
|
103 |
+
|
104 |
+
demo.launch()
|
105 |
+
|
106 |
+
|