Update app.py

app.py

@@ -1,7 +1,4 @@
 import gradio as gr
-import re
-
-# --- KEEP YOUR ORIGINAL LOGIC IMPORTS ---
 from sql_injection.sql_injection import run_sqlmap
 from qr_detector.qr_detector import qr_code_audit_app
 from data_breach_checker.password_checker import (
@@ -13,215 +10,284 @@ from phishing import scan_phishing
 from vulnerability_scanner.vulnerability_scanner import scan_website
 from encryption_tool.encryption_tool import generate_key, encrypt_message, decrypt_message
 
-# ==========================================
-# 1. HELPER FUNCTIONS (Data Adapters)
-#    These convert your text results into clean UI formats
-# ==========================================
-
-def adapter_sql_output(urls):
-    """Parses SQLMap text output into a clean Dataframe."""
-    try:
-        raw = run_sqlmap(urls)
-        text = str(raw[0]) if isinstance(raw, tuple) else str(raw)
-        
-        data = []
-        if text:
-            lines = text.split('\n')
-            curr_payload = None
-            curr_purpose = None
-            
-            for line in lines:
-                if "Payload:" in line:
-                    curr_payload = line.split("Payload:")[1].strip()
-                elif "Purpose:" in line:
-                    curr_purpose = line.split("Purpose:")[1].strip()
-                    # Add to list when we have a pair
-                    if curr_payload:
-                        data.append([curr_payload, curr_purpose, "Tested"])
-                        curr_payload = None 
-        
-        if not data:
-            return [["No Specific Payload Found", "Scan Complete", "Safe/Unknown"]]
-        return data
-    except Exception as e:
-        return [[f"Error: {str(e)}", "Failed", "Error"]]
-
-def adapter_qr_output(image):
-    """Converts QR tuple to a JSON dictionary."""
-    if image is None: return None
-    res = qr_code_audit_app(image)
-    return {
-        "Status": res[0],
-        "Decoded Data": res[1],
-        "Category": res[2],
-        "Type": res[3]
-    }
-
-def adapter_vuln_scan(url):
-    """Formats vulnerability text to Markdown."""
-    res = scan_website(url)
-    res = res.replace("SSL Status:", "### 🔒 SSL Status\n")
-    res = res.replace("XSS:", "\n### 💉 XSS Analysis\n")
-    res = res.replace("Security Headers:", "\n### 🛡️ Security Headers\n")
-    return res
-
-# ==========================================
-# 2. THEME & STYLING
-# ==========================================
-
-# A professional Dark "Cyber" Theme using native Gradio settings
-theme = gr.themes.Soft(
-    primary_hue="cyan",
-    secondary_hue="slate",
-    neutral_hue="slate",
-).set(
-    body_background_fill="#0f172a",    # Dark Slate Background
-    block_background_fill="#1e293b",   # Slightly lighter blocks
-    block_border_width="1px",
-    block_border_color="#334155",
-    button_primary_background_fill="#0ea5e9", # Bright Blue/Cyan buttons
-    button_primary_text_color="white",
-    body_text_color="#e2e8f0"          # Light text
-)
-
-# Minimal CSS only for header alignment (removing the messy CSS)
+# Theme and CSS
 css = """
-.header-icons { display: flex; gap: 15px; justify-content: flex-end; align-items: center; }
-.header-icons img { height: 28px; filter: invert(1); opacity: 0.7; transition: 0.3s; }
-.header-icons img:hover { opacity: 1; transform: scale(1.1); filter: drop-shadow(0 0 5px #0ea5e9); }
+body { font-family: 'Arial', sans-serif; background-color: #e6f7ff; color: #333; }
+.tool-title { color: #2979ff; font-size: 32px; text-align: center; margin-bottom: 30px; font-weight: 600; }
+.tool-desc { color: #666; text-align: center; margin-bottom: 35px; font-size: 18px; }
+.card { padding: 40px; border-radius: 15px; background-color: rgba(255, 255, 255, 0.8); backdrop-filter: blur(10px); box-shadow: 0 8px 20px rgba(0, 0, 0, 0.1); margin-bottom: 40px; }
+.boxed-output { white-space: pre-wrap; font-family: monospace; background: #f0f8ff; padding: 25px; border-radius: 10px; border: 1px solid #b3e5fc; color: #2f4f4f; font-size: 16px; }
+.submit-btn { background-color: #42a5f5; color: white; padding: 15px 30px; border: none; border-radius: 8px; cursor: pointer; display: block; margin: 40px auto; font-size: 18px; font-weight: 500; transition: background-color 0.3s ease, transform 0.2s ease; }
+.submit-btn:hover { background-color: #1e88e5; transform: translateY(-3px); box-shadow: 0 5px 10px rgba(0, 0, 0, 0.15); }
+.output-section { margin-top: 30px; }
+.output-title { font-weight: 600; color: #2979ff; font-size: 22px; margin-bottom: 15px; }
+.tool-select { margin: 40px auto; width: 400px; display: block; padding: 15px; border: 2px solid #90caf9; border-radius: 10px; font-size: 18px; color: #333; transition: border-color 0.3s ease, box-shadow 0.3s ease; }
+.tool-select:focus { outline: none; border-color: #1e88e5; box-shadow: 0 0 10px rgba(30, 136, 229, 0.5); }
+.tab-container { margin-top: 30px; }
+.output-section table { width: 100%; border-collapse: collapse; margin-top: 20px; }
+.output-section th, .output-section td { border: 1px solid #ddd; padding: 8px; text-align: left; }
+.output-section th { background-color: #f2f2f2; }
+.output-section pre { white-space: pre-wrap; font-family: monospace; background: #f9f9f9; padding: 10px; border-radius: 5px; }
+.social-icons-container { /* New container div */
+    position: relative;
+    width: 150px; /* Adjust width as needed */
+    height: 40px; /* Adjust height as needed */
+    margin-left: auto; /* Push to right */
+}
+.social-icons {
+    position: absolute;
+    top: 5px; /* Adjust top positioning */
+    right: 5px; /* Adjust right positioning */
+    display: flex;
+    gap: 8px; /* Reduce gap */
+}
+.social-icons img {
+    height: 25px; /* Adjust icon height */
+    transition: 0.3s ease;
+    opacity: 0.7;
+}
+.social-icons img:hover {
+    opacity: 1;
+    transform: scale(1.1);
+}
 """
 
-# ==========================================
-# 3. MAIN APP STRUCTURE
-# ==========================================
-
 def create_app():
-    with gr.Blocks(theme=theme, css=css, title="CyberSuite Toolkit") as demo:
-        
-        # --- HEADER ---
-        with gr.Row(variant="panel"):
-            with gr.Column(scale=3):
-                gr.Markdown(
-                    """
-                    # 🛡️ CyberSuite Toolkit
-                    ### **Professional Security Analysis Environment**
-                    """
-                )
+    with gr.Blocks(title="CyberSuite Toolkit", css=css) as demo:
+        # Add Font Awesome CSS link (if needed for other icons)
+        # demo.head += """
+        #     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.3/css/all.min.css">
+        # """
+        # Social Icons in the top-right corner
+        with gr.Row():
             with gr.Column(scale=1):
-                gr.HTML("""
-                    <div class="header-icons">
-                        <a href='https://github.com/deveshpujnabi/CyberSuite-Toolkit' target='_blank'><img src='https://cdn-icons-png.flaticon.com/512/25/25231.png' alt='GitHub'></a>
-                        <a href='https://www.linkedin.com/in/devesh-punjabi-9aa12b251/' target='_blank'><img src='https://cdn-icons-png.flaticon.com/512/174/174857.png' alt='LinkedIn'></a>
-                    </div>
-                """)
-
-        # --- MAIN NAVIGATION (TABS) ---
-        with gr.Tabs():
-            
-            # --- TAB 1: SQL INJECTION ---
-            with gr.TabItem("💉 SQL Injection", id="tab_sql"):
+                gr.Markdown("<h1 class='tool-title'>🛡️ CyberSuite Toolkit</h1>")
                 with gr.Row():
                     with gr.Column(scale=1):
-                        gr.Markdown("### 🎯 Target Setup")
-                        url_in = gr.Textbox(lines=4, placeholder="https://example.com", label="Target URLs", info="One URL per line")
-                        btn_sql = gr.Button("Run SQLMap Scan", variant="primary")
-                    with gr.Column(scale=2):
-                        gr.Markdown("### 📊 Scan Findings")
-                        # Dataframe replaces the ugly HTML table
-                        out_sql = gr.Dataframe(
-                            headers=["Payload", "Purpose", "Status"],
-                            datatype=["str", "str", "str"],
-                            interactive=False,
-                            wrap=True
-                        )
-                btn_sql.click(adapter_sql_output, inputs=[url_in], outputs=[out_sql])
-
-            # --- TAB 2: PHISHING ---
-            with gr.TabItem("🎣 Phishing Detector", id="tab_phish"):
-                with gr.Row():
-                    with gr.Column():
-                        phish_in = gr.Textbox(label="Suspicious URL", placeholder="http://login-verify.com")
-                        btn_phish = gr.Button("Scan URL", variant="primary")
-                    with gr.Column():
-                        # Code component with language=None is safer than HTML
-                        out_phish = gr.Code(label="Analysis Log", language=None)
-                btn_phish.click(scan_phishing, phish_in, out_phish)
-
-            # --- TAB 3: QR DETECTOR ---
-            with gr.TabItem("📷 QR Audit", id="tab_qr"):
-                with gr.Row():
-                    with gr.Column():
-                        qr_img = gr.Image(label="Upload QR Code", type="filepath")
-                        btn_qr = gr.Button("Analyze Image", variant="primary")
-                    with gr.Column():
-                        # JSON component handles structured data beautifully
-                        out_qr = gr.JSON(label="Decoded Metadata")
-                btn_qr.click(adapter_qr_output, qr_img, out_qr)
-
-            # --- TAB 4: PASSWORD MANAGER ---
-            with gr.TabItem("🔑 Passwords", id="tab_pass"):
-                with gr.Tabs():
-                    with gr.TabItem("Check Strength"):
-                        p_in = gr.Textbox(label="Password", type="password")
-                        btn_p = gr.Button("Analyze Entropy")
-                        out_p = gr.Label(label="Strength Score") 
-                        btn_p.click(gradio_password_strength, p_in, out_p)
-                    
-                    with gr.TabItem("Generator"):
+                        gr.HTML("""
+                            <div class='social-icons-container'>
+                                 <div class='social-icons'>
+                                    <a href='https://github.com/deveshpujnabi/CyberSuite-Toolkit' target='_blank'>
+                                        <img src='https://cdn-icons-png.flaticon.com/512/25/25231.png' alt='GitHub'>
+                                    </a>
+                                    <a href='https://www.linkedin.com/in/devesh-punjabi-9aa12b251/' target='_blank'>
+                                        <img src='https://upload.wikimedia.org/wikipedia/commons/thumb/8/81/LinkedIn_icon.svg/2048px-LinkedIn_icon.svg.png' alt='LinkedIn'>
+                                    </a>
+                                    <a href='https://huggingface.co/spaces/deveshpunjabi/CyberSuite-Toolkit/tree/main' target='_blank'>
+                                        <img src='https://huggingface.co/front/assets/huggingface_logo-noborder.svg' alt='HF'>
+                                    </a>
+                                </div>
+                            </div>
+                        """)
+                tool_selector = gr.Dropdown(
+                    choices=[
+                        "SQL Injection Test",
+                        "Phishing Detection Tool",
+                        "QR Detector",
+                        "Password & Data Breach Management",
+                        "Vulnerability Scanner",
+                        "Encryption Tool"
+                    ],
+                    label="Select Tool",
+                    elem_classes="tool-select"
+                )
+                with gr.Column(elem_classes="tab-container"):
+                    tab_sql = gr.Column(visible=True)
+                    with tab_sql:
+                        gr.Markdown("<h2 class='tool-title'>🛡️ SQL Injection Test</h2>")
+                        gr.Markdown("<p class='tool-desc'>Enter URLs (one per line) to test for SQL Injection vulnerabilities.</p>")
+                        url_input = gr.Textbox(lines=5, placeholder="Enter URLs (e.g., https://example.com\nhttps://testsite.com)", label="Target URLs")
                         with gr.Row():
-                            sl_len = gr.Slider(8, 64, value=12, label="Length", step=1)
-                            chk_grp = gr.CheckboxGroup(["Uppercase", "Lowercase", "Digits", "Special"], 
-                                                     value=["Uppercase", "Lowercase", "Digits", "Special"], 
-                                                     label="Complexity")
-                        btn_gen = gr.Button("Generate", variant="primary")
-                        out_gen = gr.Textbox(label="Result", interactive=False, show_copy_button=True)
-                        
-                        def gen_wrap(l, c):
-                            return gradio_generate_password(l, "Uppercase" in c, "Lowercase" in c, "Digits" in c, "Special" in c)
-                        btn_gen.click(gen_wrap, [sl_len, chk_grp], out_gen)
-
-                    with gr.TabItem("Breach Check"):
-                        b_in = gr.Textbox(label="Password", type="password")
-                        btn_b = gr.Button("Search Database")
-                        out_b = gr.Textbox(label="Breach Status")
-                        btn_b.click(gradio_breach_checker, b_in, out_b)
-
-            # --- TAB 5: VULNERABILITY SCANNER ---
-            with gr.TabItem("🏥 Site Scanner", id="tab_vuln"):
-                with gr.Row():
-                    with gr.Column():
-                        v_in = gr.Textbox(label="Website URL", placeholder="http://example.com")
-                        btn_v = gr.Button("Start Full Scan", variant="primary")
-                    with gr.Column():
-                        out_v = gr.Markdown("Results will appear here...")
-                btn_v.click(adapter_vuln_scan, v_in, out_v)
-
-            # --- TAB 6: ENCRYPTION ---
-            with gr.TabItem("🔒 Crypto Tool", id="tab_enc"):
-                with gr.Accordion("Key Management", open=True):
-                    btn_k = gr.Button("Generate New Key")
-                    out_k = gr.Code(label="Secret Key", language=None, interactive=False)
-                    btn_k.click(generate_key, None, out_k)
-                
-                with gr.Row():
-                    with gr.Column():
-                        gr.Markdown("### Encrypt")
-                        e_msg = gr.Textbox(label="Message")
-                        e_key = gr.Textbox(label="Key")
-                        btn_e = gr.Button("Encrypt Data", variant="secondary")
-                        out_e = gr.Code(label="Encrypted Output", language=None)
-                        btn_e.click(encrypt_message, [e_msg, e_key], out_e)
-                    with gr.Column():
-                        gr.Markdown("### Decrypt")
-                        d_msg = gr.Textbox(label="Encrypted String")
-                        d_key = gr.Textbox(label="Key")
-                        btn_d = gr.Button("Decrypt Data", variant="secondary")
-                        out_d = gr.Code(label="Decrypted Output", language=None)
-                        btn_d.click(decrypt_message, [d_msg, d_key], out_d)
-
+                            submit_btn = gr.Button("Run SQL Injection Test", elem_classes="submit-btn")
+                        output = gr.HTML(label="Results")
+                        def handle_sql_submit(urls):
+                            try:
+                                result = run_sqlmap(urls)
+                                if isinstance(result, tuple):
+                                    result_str = str(result[0])
+                                else:
+                                    result_str = str(result)
+                                if result_str:
+                                    lines = result_str.split('\n')
+                                    results = []
+                                    unique_payloads = set()
+                                    payload_line = None
+                                    purpose_line = None
+                                    response_preview = None
+                                    for line in lines:
+                                        if "Payload:" in line:
+                                            payload_line = line.split("Payload:")[1].strip()
+                                        elif "Purpose:" in line:
+                                            purpose_line = line.split("Purpose:")[1].strip()
+                                        elif "Response Preview:" in line:
+                                            response_preview = "\n".join(lines[lines.index(line) + 1:])
+                                        if payload_line and payload_line not in unique_payloads:
+                                            unique_payloads.add(payload_line)
+                                            result_dict = {
+                                                "payload": payload_line,
+                                                "purpose": purpose_line,
+                                                "vulnerable": "vulnerability detected" not in result_str.lower(),
+                                                "response": response_preview
+                                            }
+                                            results.append(result_dict)
+                                            payload_line = None
+                                            purpose_line = None
+                                            response_preview = None
+                                    formatted_output = "<div class='output-section'><h3 class='output-title'>SQL Injection Test Results:</h3>"
+                                    formatted_output += "<table style='width:100%; border-collapse: collapse;'>"
+                                    formatted_output += "<tr><th style='border: 1px solid #ddd; padding: 8px; text-align: left;'>Payload Tested</th><th style='border: 1px solid #ddd; padding: 8px; text-align: left;'>Purpose</th><th style='border: 1px solid #ddd; padding: 8px; text-align: left;'>Result</th><th style='border: 1px solid #ddd; padding: 8px; text-align: left;'>Response Preview</th></tr>"
+                                    for result in results:
+                                        formatted_output += f"<tr><td style='border: 1px solid #ddd; padding: 8px;'>{result['payload']}</td>"
+                                        formatted_output += f"<td style='border: 1px solid #ddd; padding: 8px;'>{result['purpose']}</td>"
+                                        if result['vulnerable']:
+                                            formatted_output += "<td style='border: 1px solid #ddd; padding: 8px; color: red;'>Vulnerable</td>"
+                                        else:
+                                            formatted_output += "<td style='border: 1px solid #ddd; padding: 8px; color: green;'>Not Vulnerable</td>"
+                                        if result['response']:
+                                            truncated_preview = result['response'][:500] + ("..." if len(result['response']) > 500 else "")
+                                            formatted_output += f"<td style='border: 1px solid #ddd; padding: 8px;'><pre>{truncated_preview}</pre></td></tr>"
+                                        else:
+                                            formatted_output += "<td style='border: 1px solid #ddd; padding: 8px;'></td></tr>"
+                                    formatted_output += "</table></div>"
+                                    return formatted_output
+                                else:
+                                    return "<div class='output-section'><h3 class='output-title'>SQL Injection Test Results:</h3><p>No vulnerabilities found or an issue occurred during the scan.</p></div>"
+                            except Exception as e:
+                                return f"<div class='output-section'><h3 class='output-title'>SQL Injection Test Error:</h3><p>An error occurred during the SQL injection test: {str(e)}</p></div>"
+                        submit_btn.click(fn=handle_sql_submit, inputs=[url_input], outputs=[output])
+                    tab_phishing = gr.Column(visible=False)
+                    with tab_phishing:
+                        gr.Markdown("<h2 class='tool-title'>🕵️ Phishing Detection Tool</h2>")
+                        gr.Markdown("<p class='tool-desc'>Enter a URL to check for phishing risks.</p>")
+                        phishing_input = gr.Textbox(label="Enter a URL", placeholder="https://example.com")
+                        phishing_submit = gr.Button("Scan URL", elem_classes="submit-btn")
+                        phishing_output = gr.HTML(label="Scan Result")
+                        def handle_phishing_scan(url):
+                            result = scan_phishing(url)
+                            formatted_result = "<div class='output-section'><h3 class='output-title'>Phishing Scan Result:</h3><pre>" + result.replace("\n", "<br>") + "</pre></div>"
+                            return formatted_result
+                        phishing_submit.click(fn=handle_phishing_scan, inputs=[phishing_input], outputs=[phishing_output])
+                    tab_qr = gr.Column(visible=False)
+                    with tab_qr:
+                        gr.Markdown("<h2 class='tool-title'>📷 QR Detector</h2>")
+                        gr.Markdown("<p class='tool-desc'>Scan and audit QR codes for potential threats.</p>")
+                        qr_image = gr.Image(label="Upload or Capture QR Code")
+                        qr_submit = gr.Button("Analyze QR Code", elem_classes="submit-btn")
+                        qr_output = gr.HTML(label="QR Code Analysis")
+                        def handle_qr_scan(image):
+                            results = qr_code_audit_app(image)
+                            formatted_result = f"""
+                            <div class='output-section'>
+                                <h3 class='output-title'>QR Code Analysis:</h3>
+                                <p><strong>Analysis Result:</strong> {results[0]}</p>
+                                <p><strong>Decoded QR Data:</strong> {results[1]}</p>
+                                <p><strong>Link Category:</strong> {results[2]}</p>
+                                <p><strong>URL Type:</strong> {results[3]}</p>
+                            </div>
+                            """
+                            return formatted_result
+                        qr_submit.click(fn=handle_qr_scan, inputs=[qr_image], outputs=[qr_output])
+                    tab_password = gr.Column(visible=False)
+                    with tab_password:
+                        gr.Markdown("<h2 class='tool-title'>🔑 Password & Data Breach Management</h2>")
+                        with gr.Tab("Check Password Strength"):
+                            password_strength_input = gr.Textbox(label="Enter Password")
+                            password_strength_output = gr.Textbox(label="Password Strength")
+                            password_strength_button = gr.Button("Check Strength")
+                            def check_password_strength(password):
+                                return gradio_password_strength(password)
+                            password_strength_button.click(check_password_strength, inputs=[password_strength_input], outputs=[password_strength_output])
+                        with gr.Tab("Generate Password"):
+                            password_length = gr.Slider(label="Password Length", minimum=8, maximum=64, step=1, value=12)
+                            uppercase = gr.Checkbox(label="Include Uppercase Letters", value=True)
+                            lowercase = gr.Checkbox(label="Include Lowercase Letters", value=True)
+                            digits = gr.Checkbox(label="Include Digits", value=True)
+                            special_chars = gr.Checkbox(label="Include Special Characters", value=True)
+                            generate_button = gr.Button("Generate Password")
+                            generated_password = gr.Textbox(label="Generated Password")
+                            def generate_password_handler(length, upper, lower, digit, special):
+                                return gradio_generate_password(length, upper, lower, digit, special)
+                            generate_button.click(generate_password_handler, inputs=[password_length, uppercase, lowercase, digits, special_chars], outputs=[generated_password])
+                        with gr.Tab("Check Password Breach"):
+                            breach_input = gr.Textbox(label="Enter Password")
+                            breach_output = gr.Textbox(label="Breach Check Result")
+                            breach_button = gr.Button("Check Breach")
+                            def check_breach(password):
+                                return gradio_breach_checker(password)
+                            breach_button.click(check_breach, inputs=[breach_input], outputs=[breach_output])
+                    tab_vuln = gr.Column(visible=False)
+                    with tab_vuln:
+                        gr.Markdown("<h2 class='tool-title'>🔍 Vulnerability Scanner</h2>")
+                        gr.Markdown("<p class='tool-desc'>Scan a website for common vulnerabilities.</p>")
+                        vuln_input = gr.Textbox(label="Enter Website URL", placeholder="http://example.com")
+                        vuln_submit = gr.Button("Scan Website", elem_classes="submit-btn")
+                        vuln_output = gr.HTML(label="Scan Report")
+                        def handle_vuln_scan(url):
+                            result = scan_website(url)
+                            lines = result.split('\n')
+                            formatted_result = "<div class='output-section'><h3 class='output-title'>Vulnerability Scan Report:</h3>"
+                            formatted_result += "<table style='width:100%; border-collapse: collapse;'>"
+                            for line in lines:
+                                if "SSL Status:" in line:
+                                    ssl_status = line.split("SSL Status:")[1].strip()
+                                    formatted_result += f"<tr><td><strong>SSL Status:</strong></td><td>{ssl_status}</td></tr>"
+                                elif "XSS:" in line:
+                                    xss_status = line.split("XSS:")[1].strip()
+                                    formatted_result += f"<tr><td><strong>XSS:</strong></td><td>{xss_status}</td></tr>"
+                                elif "Security Headers:" in line:
+                                    headers_status = line.split("Security Headers:")[1].strip()
+                                    formatted_result += f"<tr><td><strong>Security Headers:</strong></td><td>{headers_status}</td></tr>"
+                                elif "Vulnerability Report Summary:" in line:
+                                    formatted_result += f"<tr><td colspan='2'><strong>{line.strip()}</strong></td></tr>"
+                                elif "Detailed Vulnerability Report" in line:
+                                    formatted_result += f"<tr><td colspan='2'><strong>{line.strip()}</strong></td></tr>"
+                                else:
+                                    if line.strip():
+                                        formatted_result += f"<tr><td colspan='2'>{line.strip()}</td></tr>"
+                            formatted_result += "</table></div>"
+                            return formatted_result
+                        vuln_submit.click(fn=handle_vuln_scan, inputs=[vuln_input], outputs=[vuln_output])
+                    tab_encrypt = gr.Column(visible=False)
+                    with tab_encrypt:
+                        gr.Markdown("<h2 class='tool-title'>🔒 Encryption Tool</h2>")
+                        with gr.Tab("Generate Key"):
+                            key_gen_btn = gr.Button("Generate New Key", elem_classes="submit-btn")
+                            key_output = gr.Textbox(label="Generated Key", elem_classes="boxed-output")
+                            def handle_key_generation():
+                                result = generate_key()
+                                return result
+                            key_gen_btn.click(fn=handle_key_generation, inputs=None, outputs=[key_output])
+                        with gr.Tab("Encrypt Message"):
+                            enc_msg = gr.Textbox(label="Enter message to encrypt")
+                            enc_key = gr.Textbox(label="Enter encryption key")
+                            enc_btn = gr.Button("Encrypt", elem_classes="submit-btn")
+                            enc_output = gr.Textbox(label="Encrypted Message", elem_classes="boxed-output")
+                            def handle_encryption(message, key):
+                                result = encrypt_message(message, key)
+                                return result
+                            enc_btn.click(fn=handle_encryption, inputs=[enc_msg, enc_key], outputs=[enc_output])
+                        with gr.Tab("Decrypt Message"):
+                            dec_msg = gr.Textbox(label="Enter encrypted message")
+                            dec_key = gr.Textbox(label="Enter decryption key")
+                            dec_btn = gr.Button("Decrypt", elem_classes="submit-btn")
+                            dec_output = gr.Textbox(label="Decrypted Message", elem_classes="boxed-output")
+                            def handle_decryption(message, key):
+                                result = decrypt_message(message, key)
+                                return result
+                            dec_btn.click(fn=handle_decryption, inputs=[dec_msg, dec_key], outputs=[dec_output])
+                def switch_tabs(tool):
+                    return [
+                        gr.update(visible=tool == "SQL Injection Test"),
+                        gr.update(visible=tool == "Phishing Detection Tool"),
+                        gr.update(visible=tool == "QR Detector"),
+                        gr.update(visible=tool == "Password & Data Breach Management"),
+                        gr.update(visible=tool == "Vulnerability Scanner"),
+                        gr.update(visible=tool == "Encryption Tool")
+                    ]
+                tool_selector.change(switch_tabs, inputs=tool_selector, outputs=[tab_sql, tab_phishing, tab_qr, tab_password, tab_vuln, tab_encrypt])
     return demo
 
 if __name__ == "__main__":
     demo = create_app()
     demo.queue()
-    demo.launch(share=False, quiet=True)
+    demo.launch(share=False, debug=False, show_error=True, max_threads=16, favicon_path=None, quiet=True)