Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import json
 os.system('pip install transformers')
 os.system('pip install gradio')
 os.system('pip install requests')
@@ -51,6 +52,12 @@ def evaluate_pci_compliance(audit_data):
     compliance_analysis = respond(audit_data, system_message, max_tokens=1024, temperature=0.7, top_p=0.95)
     return compliance_analysis
 
+def analyze_json_file(file):
+    # Load and preprocess JSON data for the AI model
+    json_data = json.load(file)
+    audit_data = json.dumps(json_data, indent=2)  # Convert JSON to a formatted string
+    return audit_data
+
 # Custom CSS for the specified theme
 custom_css = """
 body {
@@ -124,21 +131,22 @@ footer {
 # Gradio Interface
 with gr.Blocks(css=custom_css) as demo:
     with gr.Column():
-        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")
+        gr.Markdown("# GDPR and PCI Compliance Evaluation\n### Upload Audit Data in JSON Format")
+        json_file = gr.File(label="Upload JSON file")
         
         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_compliance_checks(audit_data):
+        def run_compliance_checks(json_file):
+            audit_data = analyze_json_file(json_file)
             gdpr_analysis = evaluate_gdpr_compliance(audit_data)
             pci_analysis = evaluate_pci_compliance(audit_data)
             return gdpr_analysis, pci_analysis
         
         check_compliance_btn = gr.Button("Run Compliance Checks")
-        check_compliance_btn.click(run_compliance_checks, inputs=[audit_data], outputs=[gdpr_compliance, pci_compliance])
+        check_compliance_btn.click(run_compliance_checks, inputs=[json_file], outputs=[gdpr_compliance, pci_compliance])
         
         clear_btn = gr.Button("Clear")
-        clear_btn.click(lambda: ("", "", ""), None, [audit_data, gdpr_compliance, pci_compliance])
+        clear_btn.click(lambda: ("", "", ""), None, [json_file, gdpr_compliance, pci_compliance])
 
 demo.launch()