Spaces:
Sleeping
Sleeping
File size: 1,314 Bytes
a635b88 8cb26d4 a635b88 8cb26d4 a635b88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import gradio as gr
import yaml
from pathlib import Path
from compliance_analysis import run_compliance_analysis_on_project, run_compliance_analysis_on_data, run_compliance_analysis_on_model
def process_files(files):
results = []
for file in files:
with open(file.name, 'r') as f:
content = f.read()
if Path(file.name).name == "project_cc.yaml":
project_cc_yaml = yaml.safe_load(content)
msg = run_compliance_analysis_on_project(project_cc_yaml)
results.append(msg)
# if Path(file.name).name == "data_cc.yaml":
# data_cc_yaml = yaml.safe_load(content)
# msg = run_compliance_analysis_on_data(data_cc_yaml)
# results.append(msg)
# if Path(file.name).name == "model_cc.yaml":
# model_cc_yaml = yaml.safe_load(content)
# msg = run_compliance_analysis_on_model(model_cc_yaml)
# results.append(msg)
return results
# Gradio interface
with gr.Blocks() as demo:
file_input = gr.File(label="Upload Files", file_count="multiple")
output = gr.Textbox(label="Output", lines=10)
submit_button = gr.Button("Process Files")
submit_button.click(process_files, inputs=file_input, outputs=output)
demo.launch()
|