aminaB9 commited on
Commit
cc77879
·
1 Parent(s): f003e69

Initial commit

Browse files
Files changed (1) hide show
  1. app.py +9 -29
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()