Update app.py

app.py

@@ -1,17 +1,32 @@
 import gradio as gr
-import crypt
+import pixel_encrypt as pe
+from diff import SendIt as si 
+go=si()
+models = go.models
 
 with gr.Blocks() as app:
     with gr.Row():
         gr.Column()
         with gr.Column():
-            rsa_to_enc = gr.Textbox(label="txt to encrypt")
-            pub_key_in = gr.Image(label="Public Key", type="filepath")
-            priv_key_in1 = gr.Image(label="Private Key(sig)", type="filepath")
-            rsa_enc_btn = gr.Button("RSA Encrypt")
-            rsa_enc_mes = gr.Textbox(label="encoded", max_lines=4)
-            qr_enc_mes = gr.Image(type="filepath")
-
+            with gr.Row():
+                with gr.Tab("Encrypt"):
+                    rsa_to_enc = gr.Textbox(label="Message to Encrypt",lines=3)
+                    pub_key_in = gr.Image(label="Public Key", type="filepath")
+                    priv_key_in1 = gr.Image(label="Private Key(sig)", type="filepath")
+                    rsa_enc_btn = gr.Button("RSA Encrypt")
+                    rsa_enc_mes = gr.Textbox(label="encoded", max_lines=4)
+                    qr_enc_mes = gr.Image(type="filepath")
+                with gr.Tab("Custom"):
+                    with gr.Group():
+                        with gr.Column():
+                            inp=gr.Textbox(label="Prompt")
+                            model=gr.Dropdown(type='index',choices=[m for m in models], value=models[5])
+                            cnt = gr.Slider(minimum=1,maximum=4,step=1,value=1)                    
+                            diff_btn = gr.Button()
+                    diff_im = gr.Gallery()
+                    im_choice=gr.Image(label='Place image here',type='filepath')
         gr.Column()
-    rsa_enc_btn.click(crypt.encrypt_text,[rsa_to_enc,pub_key_in,priv_key_in1,out3],[rsa_enc_mes,qr_enc_mes]).then(crypt.test_fn,qr_enc_mes,mes_in)
+    rsa_enc_btn.click(pe.encrypt_text,[rsa_to_enc,pub_key_in,priv_key_in1,out3],[rsa_enc_mes,qr_enc_mes])
+    diff_btn.click(go.send_it,[inp,model,cnt],diff_im)
+
 app.launch()