michaelmc1618 commited on
Commit
decc4c9
·
verified ·
1 Parent(s): 04e48f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -1,14 +1,12 @@
1
  import os
2
  import json
3
- os.system('pip install transformers')
4
- os.system('pip install gradio')
5
- os.system('pip install requests')
6
-
7
- import requests
8
  import gradio as gr
9
  from huggingface_hub import InferenceClient
10
  from transformers import pipeline
11
 
 
 
 
12
  # Inference client for chat completion
13
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
14
 
@@ -52,9 +50,9 @@ def evaluate_pci_compliance(audit_data):
52
  compliance_analysis = respond(audit_data, system_message, max_tokens=1024, temperature=0.7, top_p=0.95)
53
  return compliance_analysis
54
 
55
- def analyze_json_file(file):
56
- # Convert the file's bytes to a JSON-serializable string and parse it
57
- file_str = file.read().decode("utf-8")
58
  json_data = json.loads(file_str)
59
  audit_data = json.dumps(json_data, indent=2) # Convert JSON to a formatted string
60
  return audit_data
@@ -139,6 +137,8 @@ with gr.Blocks(css=custom_css) as demo:
139
  pci_compliance = gr.Textbox(lines=10, placeholder="PCI Compliance Analysis...", label="PCI Compliance Analysis", elem_classes="label-hidden")
140
 
141
  def run_compliance_checks(json_file):
 
 
142
  audit_data = analyze_json_file(json_file)
143
  gdpr_analysis = evaluate_gdpr_compliance(audit_data)
144
  pci_analysis = evaluate_pci_compliance(audit_data)
@@ -148,6 +148,6 @@ with gr.Blocks(css=custom_css) as demo:
148
  check_compliance_btn.click(run_compliance_checks, inputs=[json_file], outputs=[gdpr_compliance, pci_compliance])
149
 
150
  clear_btn = gr.Button("Clear")
151
- clear_btn.click(lambda: ("", "", ""), None, [json_file, gdpr_compliance, pci_compliance])
152
 
153
  demo.launch(share=True)
 
1
  import os
2
  import json
 
 
 
 
 
3
  import gradio as gr
4
  from huggingface_hub import InferenceClient
5
  from transformers import pipeline
6
 
7
+ # Install necessary dependencies
8
+ os.system('pip install transformers gradio requests')
9
+
10
  # Inference client for chat completion
11
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
12
 
 
50
  compliance_analysis = respond(audit_data, system_message, max_tokens=1024, temperature=0.7, top_p=0.95)
51
  return compliance_analysis
52
 
53
+ def analyze_json_file(file_obj):
54
+ # Assuming file_obj is a file-like object (e.g., from Gradio)
55
+ file_str = file_obj.read().decode("utf-8")
56
  json_data = json.loads(file_str)
57
  audit_data = json.dumps(json_data, indent=2) # Convert JSON to a formatted string
58
  return audit_data
 
137
  pci_compliance = gr.Textbox(lines=10, placeholder="PCI Compliance Analysis...", label="PCI Compliance Analysis", elem_classes="label-hidden")
138
 
139
  def run_compliance_checks(json_file):
140
+ if json_file is None:
141
+ return "No file uploaded", "No file uploaded"
142
  audit_data = analyze_json_file(json_file)
143
  gdpr_analysis = evaluate_gdpr_compliance(audit_data)
144
  pci_analysis = evaluate_pci_compliance(audit_data)
 
148
  check_compliance_btn.click(run_compliance_checks, inputs=[json_file], outputs=[gdpr_compliance, pci_compliance])
149
 
150
  clear_btn = gr.Button("Clear")
151
+ clear_btn.click(lambda: ("", ""), None, [gdpr_compliance, pci_compliance])
152
 
153
  demo.launch(share=True)