Add encrypted output representation

app.py

@@ -139,15 +139,15 @@ with demo:
     )
     get_output_button = gr.Button("Receive the encrypted output from the server.")
 
-    # encrypted_output_representation = gr.Textbox(
-    #     label="Encrypted output representation: ", max_lines=1, interactive=False
-    # )
+    encrypted_output_representation = gr.Textbox(
+        label="Encrypted output representation: ", max_lines=1, interactive=False
+    )
 
     gr.Markdown("### Step 8: Decrypt the output.")
     decrypt_button = gr.Button("Decrypt the output")
 
     prediction_output = gr.Textbox(
-        label="Credit card approval decision: ", max_lines=1, interactive=False
+        label="Credit card approval high risk: ", max_lines=1, interactive=False
     )
 
     # Button to pre-process, generate the key, encrypt and send the user inputs from the client 
@@ -184,7 +184,7 @@ with demo:
     get_output_button.click(
         get_output, 
         inputs=[user_id, bank_id, third_party_id], 
-        # outputs=[encrypted_output_representation]
+        outputs=[encrypted_output_representation],
     )
 
     # TODO : ID should be unique

backend.py

@@ -342,6 +342,9 @@ def get_output(user_id, bank_id, third_party_id):
         user_id (int): The user ID to consider.
         bank_id (int): The bank ID to consider.
         third_party_id (int): The third party ID to consider.
+    
+    Returns:
+        encrypted_output_short (bytes): A byte short representation of the encrypted output. 
     """
     data = {
         "user_id": user_id,
@@ -360,19 +363,15 @@ def get_output(user_id, bank_id, third_party_id):
 
             # Save the encrypted output to bytes in a file as it is too large to pass through regular
             # Gradio buttons (see https://github.com/gradio-app/gradio/issues/1877)
-            # TODO : check if output to user is relevant
             encrypted_output_path = _get_client_file_path("encrypted_output", user_id + bank_id + third_party_id, "output")
 
             with encrypted_output_path.open("wb") as encrypted_output_file:
                 encrypted_output_file.write(encrypted_output)
             
-            # TODO 
-            # Decrypt the output using a different (wrong) key for display
-            # output_encrypted_representation = decrypt_output_with_wrong_key(encrypted_output, client_type)
-
-            # return output_encrypted_representation
+            # Create a truncated version of the encrypted inputs for display
+            encrypted_output_short = shorten_bytes_object(encrypted_output)
 
-            return None
+            return encrypted_output_short
         else:
             raise gr.Error("Please wait for the FHE execution to be completed.")