Update app.py

app.py

@@ -31,23 +31,25 @@ def respond(message, system_message, max_tokens, temperature, top_p):
             response += token
     return response
 
-def generate_prosecution_argument(case_details):
+# GDPR Compliance Expert
+def evaluate_gdpr_compliance(audit_data):
     system_message = (
-        "You are an expert Prosecution Attorney. Provide the best and most detailed arguments "
-        "to prosecute the case based on the given case details. Include thorough analysis, "
-        "evidence presentation, and any relevant legal precedents."
+        "You are an expert GDPR compliance officer. Assess the audit data for compliance with GDPR regulations. "
+        "Provide an analysis that identifies any compliance issues and suggestions for remediation. "
+        "Ensure a thorough evaluation of data processing, storage, and protection practices in line with GDPR requirements."
     )
-    arguments = respond(case_details, system_message, max_tokens=1024, temperature=0.7, top_p=0.95)
-    return arguments
+    compliance_analysis = respond(audit_data, system_message, max_tokens=1024, temperature=0.7, top_p=0.95)
+    return compliance_analysis
 
-def generate_defense_argument(prosecution_argument):
+# PCI Compliance Expert
+def evaluate_pci_compliance(audit_data):
     system_message = (
-        "You are an expert Defense Attorney. Provide the best and most detailed arguments "
-        "to defend the case based on the given case details. Include thorough analysis, "
-        "evidence presentation, and any relevant legal precedents."
+        "You are an expert PCI compliance officer. Assess the audit data for compliance with PCI DSS regulations. "
+        "Provide an analysis that identifies any compliance issues and suggestions for remediation. "
+        "Ensure a thorough evaluation of payment card data security, storage, and processing practices in line with PCI requirements."
     )
-    arguments = respond(prosecution_argument, system_message, max_tokens=1024, temperature=0.7, top_p=0.95)
-    return arguments
+    compliance_analysis = respond(audit_data, system_message, max_tokens=1024, temperature=0.7, top_p=0.95)
+    return compliance_analysis
 
 # Custom CSS for the specified theme
 custom_css = """
@@ -122,25 +124,21 @@ footer {
 # Gradio Interface
 with gr.Blocks(css=custom_css) as demo:
     with gr.Column():
-        gr.Markdown("# Court Argument Simulation\n### Provide Initial Case Details")
-        case_details = gr.Textbox(lines=5, placeholder="Enter initial case details here...", label="Initial Case Details", elem_classes="label-hidden")
-        evidence = gr.Textbox(lines=3, placeholder="Enter evidence details here...", label="Evidence", elem_classes="label-hidden")
-        witness_statements = gr.Textbox(lines=3, placeholder="Enter witness statements here...", label="Witness Statements", elem_classes="label-hidden")
-        legal_references = gr.Textbox(lines=3, placeholder="Enter legal references here...", label="Legal References", elem_classes="label-hidden")
+        gr.Markdown("# GDPR and PCI Compliance Evaluation\n### Provide Audit Data for Compliance Check")
+        audit_data = gr.Textbox(lines=5, placeholder="Enter audit data here...", label="Audit Data", elem_classes="label-hidden")
         
-        prosecution_argument = gr.Textbox(lines=10, placeholder="Prosecution's Argument...", label="Prosecution's Argument", elem_classes="label-hidden")
-        defense_argument = gr.Textbox(lines=10, placeholder="Defense's Argument...", label="Defense's Argument", elem_classes="label-hidden")
+        gdpr_compliance = gr.Textbox(lines=10, placeholder="GDPR Compliance Analysis...", label="GDPR Compliance Analysis", elem_classes="label-hidden")
+        pci_compliance = gr.Textbox(lines=10, placeholder="PCI Compliance Analysis...", label="PCI Compliance Analysis", elem_classes="label-hidden")
         
-        def run_simulation(case_details, evidence, witness_statements, legal_references):
-            full_case_details = f"Case Details: {case_details}\n\nEvidence: {evidence}\n\nWitness Statements: {witness_statements}\n\nLegal References: {legal_references}"
-            prosecution_arg = generate_prosecution_argument(full_case_details)
-            defense_arg = generate_defense_argument(prosecution_arg)
-            return prosecution_arg, defense_arg
+        def run_compliance_checks(audit_data):
+            gdpr_analysis = evaluate_gdpr_compliance(audit_data)
+            pci_analysis = evaluate_pci_compliance(audit_data)
+            return gdpr_analysis, pci_analysis
         
-        simulate_btn = gr.Button("Start Argument Simulation")
-        simulate_btn.click(run_simulation, inputs=[case_details, evidence, witness_statements, legal_references], outputs=[prosecution_argument, defense_argument])
+        check_compliance_btn = gr.Button("Run Compliance Checks")
+        check_compliance_btn.click(run_compliance_checks, inputs=[audit_data], outputs=[gdpr_compliance, pci_compliance])
         
         clear_btn = gr.Button("Clear")
-        clear_btn.click(lambda: ("", "", "", "", "", ""), None, [case_details, evidence, witness_statements, legal_references, prosecution_argument, defense_argument])
+        clear_btn.click(lambda: ("", "", ""), None, [audit_data, gdpr_compliance, pci_compliance])
 
 demo.launch()