Initial commit
Browse files
app.py
CHANGED
@@ -23,17 +23,6 @@ def resizeImage(image):
|
|
23 |
|
24 |
|
25 |
|
26 |
-
def free_port(port):
|
27 |
-
try:
|
28 |
-
result = subprocess.check_output(f"lsof -t -i:{port}", shell=True).decode().strip()
|
29 |
-
if result:
|
30 |
-
for pid in result.split("\n"):
|
31 |
-
subprocess.call(["kill", "-9", pid])
|
32 |
-
except Exception as e:
|
33 |
-
print(f"Could not free port {port}: {e}")
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
|
38 |
|
39 |
SECURITYLEVELS = ["128", "196", "256"]
|
@@ -196,7 +185,7 @@ with gr.Blocks() as demo:
|
|
196 |
key_button = gr.Button("Generate the FHE public and secret keys")
|
197 |
key_status = gr.Checkbox(label="FHE Public and Secret keys generated.", value=False)
|
198 |
time_output = gr.HTML()
|
199 |
-
key_button.click(fn=runBinFile, inputs=[gr.State("./genKeys.bin"), securityLevel, gr.State("genkeys")], outputs=[key_status,time_output])
|
200 |
|
201 |
|
202 |
with gr.Row():
|
@@ -206,12 +195,12 @@ with gr.Blocks() as demo:
|
|
206 |
key_button = gr.Button("Encrypt")
|
207 |
key_status = gr.Checkbox(label="Reference embedding encrypted.", value=False)
|
208 |
time_output = gr.HTML()
|
209 |
-
key_button.click(fn=runBinFile, inputs=[gr.State("./encReference.bin"), securityLevel, gr.State("encrypt")], outputs=[key_status,time_output])
|
210 |
|
211 |
with gr.Column():
|
212 |
key_button = gr.Button("Display")
|
213 |
output_text = gr.Text(label="Encrypted embedding", lines=3, interactive=False)
|
214 |
-
key_button.click(fn=runBinFile, inputs=[gr.State("./encReference.bin"), securityLevel, gr.State("print")], outputs=output_text)
|
215 |
|
216 |
|
217 |
with gr.Row():
|
@@ -244,11 +233,11 @@ with gr.Blocks() as demo:
|
|
244 |
key_button = gr.Button("Protect")
|
245 |
key_status = gr.Checkbox(label="Probe embedding protected.", value=False)
|
246 |
time_output = gr.HTML()
|
247 |
-
key_button.click(fn=runBinFile, inputs=[gr.State("./encProbe.bin"), securityLevel, gr.State("encrypt")], outputs=[key_status,time_output])
|
248 |
with gr.Column():
|
249 |
key_button = gr.Button("Display")
|
250 |
output_text = gr.Text(label="Protected embedding", lines=3, interactive=False)
|
251 |
-
key_button.click(fn=runBinFile, inputs=[gr.State("./encProbe.bin"), securityLevel, gr.State("print")], outputs=output_text)
|
252 |
|
253 |
with gr.Row():
|
254 |
gr.Markdown("### Step 4: 🔒 Compute biometric recognition decision using the threshold under FHE.")
|
@@ -263,11 +252,11 @@ with gr.Blocks() as demo:
|
|
263 |
key_button = gr.Button("Biometric recognition under FHE")
|
264 |
key_status = gr.Checkbox(label="Recognition decision encrypted.", value=False)
|
265 |
time_output = gr.HTML()
|
266 |
-
key_button.click(fn=runBinFile, inputs=[gr.State("./recDecision.bin"), securityLevel, gr.State("decision"), number_threshold], outputs=[key_status,time_output])
|
267 |
with gr.Column():
|
268 |
key_button = gr.Button("Display")
|
269 |
output_text = gr.Text(label="Encrypted decision", lines=3, interactive=False)
|
270 |
-
key_button.click(fn=runBinFile, inputs=[gr.State("./recDecision.bin"), securityLevel, gr.State("print")], outputs=output_text)
|
271 |
|
272 |
|
273 |
with gr.Row():
|
@@ -277,14 +266,14 @@ with gr.Blocks() as demo:
|
|
277 |
decision_button = gr.Button("Decrypt")
|
278 |
decision_status = gr.Checkbox(label="Recognition decision decrypted.", value=False)
|
279 |
time_output = gr.HTML()
|
280 |
-
decision_button.click(fn=runBinFile, inputs=[gr.State("./decDecision.bin"), securityLevel, gr.State("decision")], outputs=[decision_status, time_output])
|
281 |
with gr.Column(scale=3):
|
282 |
with gr.Row():
|
283 |
check_button = gr.Button("Check")
|
284 |
with gr.Row():
|
285 |
with gr.Column(scale=1):
|
286 |
final_output = gr.HTML()
|
287 |
-
check_button.click(fn=runBinFile, inputs=[gr.State("./decDecision.bin"), securityLevel, gr.State("styledPrint")], outputs=final_output)
|
288 |
with gr.Column(scale=1):
|
289 |
image_output_enroll = gr.Image(label="Reference", sources="upload")
|
290 |
image_input_enroll.change(fn=display_image, inputs=image_input_enroll, outputs=image_output_enroll)
|
@@ -294,13 +283,4 @@ with gr.Blocks() as demo:
|
|
294 |
|
295 |
|
296 |
|
297 |
-
|
298 |
-
|
299 |
-
# preferred_port = 8080
|
300 |
-
# free_port(preferred_port)
|
301 |
-
|
302 |
-
#
|
303 |
-
# demo.launch(debug=True,server_port=preferred_port)
|
304 |
-
|
305 |
-
|
306 |
demo.launch()
|
|
|
23 |
|
24 |
|
25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
|
27 |
|
28 |
SECURITYLEVELS = ["128", "196", "256"]
|
|
|
185 |
key_button = gr.Button("Generate the FHE public and secret keys")
|
186 |
key_status = gr.Checkbox(label="FHE Public and Secret keys generated.", value=False)
|
187 |
time_output = gr.HTML()
|
188 |
+
key_button.click(fn=runBinFile, inputs=[gr.State("./bin/genKeys.bin"), securityLevel, gr.State("genkeys")], outputs=[key_status,time_output])
|
189 |
|
190 |
|
191 |
with gr.Row():
|
|
|
195 |
key_button = gr.Button("Encrypt")
|
196 |
key_status = gr.Checkbox(label="Reference embedding encrypted.", value=False)
|
197 |
time_output = gr.HTML()
|
198 |
+
key_button.click(fn=runBinFile, inputs=[gr.State("./bin/encReference.bin"), securityLevel, gr.State("encrypt")], outputs=[key_status,time_output])
|
199 |
|
200 |
with gr.Column():
|
201 |
key_button = gr.Button("Display")
|
202 |
output_text = gr.Text(label="Encrypted embedding", lines=3, interactive=False)
|
203 |
+
key_button.click(fn=runBinFile, inputs=[gr.State("./bin/encReference.bin"), securityLevel, gr.State("print")], outputs=output_text)
|
204 |
|
205 |
|
206 |
with gr.Row():
|
|
|
233 |
key_button = gr.Button("Protect")
|
234 |
key_status = gr.Checkbox(label="Probe embedding protected.", value=False)
|
235 |
time_output = gr.HTML()
|
236 |
+
key_button.click(fn=runBinFile, inputs=[gr.State("./bin/encProbe.bin"), securityLevel, gr.State("encrypt")], outputs=[key_status,time_output])
|
237 |
with gr.Column():
|
238 |
key_button = gr.Button("Display")
|
239 |
output_text = gr.Text(label="Protected embedding", lines=3, interactive=False)
|
240 |
+
key_button.click(fn=runBinFile, inputs=[gr.State("./bin/encProbe.bin"), securityLevel, gr.State("print")], outputs=output_text)
|
241 |
|
242 |
with gr.Row():
|
243 |
gr.Markdown("### Step 4: 🔒 Compute biometric recognition decision using the threshold under FHE.")
|
|
|
252 |
key_button = gr.Button("Biometric recognition under FHE")
|
253 |
key_status = gr.Checkbox(label="Recognition decision encrypted.", value=False)
|
254 |
time_output = gr.HTML()
|
255 |
+
key_button.click(fn=runBinFile, inputs=[gr.State("./bin/recDecision.bin"), securityLevel, gr.State("decision"), number_threshold], outputs=[key_status,time_output])
|
256 |
with gr.Column():
|
257 |
key_button = gr.Button("Display")
|
258 |
output_text = gr.Text(label="Encrypted decision", lines=3, interactive=False)
|
259 |
+
key_button.click(fn=runBinFile, inputs=[gr.State("./bin/recDecision.bin"), securityLevel, gr.State("print")], outputs=output_text)
|
260 |
|
261 |
|
262 |
with gr.Row():
|
|
|
266 |
decision_button = gr.Button("Decrypt")
|
267 |
decision_status = gr.Checkbox(label="Recognition decision decrypted.", value=False)
|
268 |
time_output = gr.HTML()
|
269 |
+
decision_button.click(fn=runBinFile, inputs=[gr.State("./bin/decDecision.bin"), securityLevel, gr.State("decision")], outputs=[decision_status, time_output])
|
270 |
with gr.Column(scale=3):
|
271 |
with gr.Row():
|
272 |
check_button = gr.Button("Check")
|
273 |
with gr.Row():
|
274 |
with gr.Column(scale=1):
|
275 |
final_output = gr.HTML()
|
276 |
+
check_button.click(fn=runBinFile, inputs=[gr.State("./bin/decDecision.bin"), securityLevel, gr.State("styledPrint")], outputs=final_output)
|
277 |
with gr.Column(scale=1):
|
278 |
image_output_enroll = gr.Image(label="Reference", sources="upload")
|
279 |
image_input_enroll.change(fn=display_image, inputs=image_input_enroll, outputs=image_output_enroll)
|
|
|
283 |
|
284 |
|
285 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
286 |
demo.launch()
|