Update app.py

app.py

@@ -1,5 +1,11 @@
 import gradio as gr
 from predictor import predict, key_already_generated, pre_process_encrypt_send_purchase, decrypt_prediction
+import base64
+
+def encode_image_to_base64(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode("utf-8")
+
 
 def key_generated():
     """
@@ -18,19 +24,23 @@ def key_generated():
         print("Keys have been generated ✅")
         return {gen_key_btn: gr.update(value="Keys have been generated ✅")}
     
-
 demo = gr.Blocks(css=".markdown-body { font-size: 18px; }")
 
 with demo:
+    zama_base64 = encode_image_to_base64("./Img/zama.png")
+    epita_base64 = encode_image_to_base64("./Img/Epita.png")
+
     gr.Markdown(
-        """
+        f"""
         <div style="display: flex; justify-content: center; align-items: center;">
-            <img style="margin-right: 50px;" width=200 src="Img/zama.png">
-            <img width=200 src="Img/Epita.png">
+            <img style="margin-right: 50px;" width=200 src="data:image/png;base64,{zama_base64}">
+            <img width=200 src="data:image/png;base64,{epita_base64}">
         </div>
         """
     )
 
+
+
     gr.Markdown(
         """
         <h1 style="text-align: center;">Fraud Detection with FHE Model</h1>
@@ -93,10 +103,11 @@ with demo:
         """
     )
 
+    schema_base64 = encode_image_to_base64("./Img/schema.png")
     gr.Markdown(
-        """
+        f"""
         <p align="center">
-            <img width="75%" height="30%" src="Img/schema.png">
+            <img width="75%" height="30%" src="data:image/png;base64,{schema_base64}">
         </p>
         """
     )
@@ -120,7 +131,7 @@ with demo:
         key_generated,
         inputs=[],
         outputs=[gen_key_btn],
-    )
+    )#547
 
     gr.Markdown("<hr />")
 
@@ -253,6 +264,7 @@ with demo:
         label="Total FHE execution time (in seconds):", max_lines=1, interactive=False
     )
 
+    # Button to send the encodings to the server using post method
     execute_fhe_button.click(predict, outputs=[fhe_execution_time, execute_fhe_button])
 
     gr.Markdown("<hr />")
@@ -274,6 +286,7 @@ with demo:
         decrypt_prediction,
         outputs=[prediction_output, get_output_button],
     )
+    
 
     gr.Markdown(
         """
@@ -288,4 +301,4 @@ with demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()