Update app.py

app.py

@@ -92,12 +92,36 @@ def rsa_keys():
     private_key, public_key = rsa_generate_keys()
     return private_key, public_key
 
-# Gradio interface
-caesar = gr.Interface(fn=caesar_encrypt_decrypt, inputs=["text", "number", gr.inputs.Radio(["Encrypt", "Decrypt"])], outputs="text", title="Caesar Cipher")
-aes = gr.Interface(fn=aes_encrypt_decrypt, inputs=["text", "text", gr.inputs.Radio(["Encrypt", "Decrypt"])], outputs="text", title="AES Encryption")
-rsa = gr.Interface(fn=rsa_encrypt_decrypt, inputs=["text", "text", gr.inputs.Radio(["Encrypt", "Decrypt"]), gr.inputs.Radio(["Private Key", "Public Key"])], outputs="text", title="RSA Encryption")
-rsa_key_gen = gr.Interface(fn=rsa_keys, inputs=None, outputs=["text", "text"], title="RSA Key Generation")
+# Gradio Blocks interface
+with gr.Blocks() as demo:
+    with gr.Tab("Caesar Cipher"):
+        with gr.Row():
+            caesar_text = gr.Textbox(label="Text")
+            caesar_shift = gr.Number(label="Shift")
+            caesar_mode = gr.Radio(["Encrypt", "Decrypt"], label="Mode")
+        caesar_output = gr.Textbox(label="Output")
+        gr.Button("Submit").click(caesar_encrypt_decrypt, [caesar_text, caesar_shift, caesar_mode], caesar_output)
 
-demo = gr.TabbedInterface([caesar, aes, rsa, rsa_key_gen], ["Caesar Cipher", "AES", "RSA", "RSA Key Generation"])
+    with gr.Tab("AES"):
+        with gr.Row():
+            aes_key = gr.Textbox(label="Key")
+            aes_text = gr.Textbox(label="Text")
+            aes_mode = gr.Radio(["Encrypt", "Decrypt"], label="Mode")
+        aes_output = gr.Textbox(label="Output")
+        gr.Button("Submit").click(aes_encrypt_decrypt, [aes_key, aes_text, aes_mode], aes_output)
+
+    with gr.Tab("RSA"):
+        with gr.Row():
+            rsa_key = gr.Textbox(label="Key (Public/Private)")
+            rsa_text = gr.Textbox(label="Text")
+            rsa_mode = gr.Radio(["Encrypt", "Decrypt"], label="Mode")
+            rsa_key_type = gr.Radio(["Public Key", "Private Key"], label="Key Type")
+        rsa_output = gr.Textbox(label="Output")
+        gr.Button("Submit").click(rsa_encrypt_decrypt, [rsa_key, rsa_text, rsa_mode, rsa_key_type], rsa_output)
+
+    with gr.Tab("RSA Key Generation"):
+        rsa_private_output = gr.Textbox(label="Private Key")
+        rsa_public_output = gr.Textbox(label="Public Key")
+        gr.Button("Generate Keys").click(rsa_keys, [], [rsa_private_output, rsa_public_output])
 
 demo.launch()