Spaces:
Sleeping
Sleeping
File size: 2,543 Bytes
5850aac f121a7d 89fa8e9 e0549a6 f121a7d e0549a6 f73df00 f121a7d f73df00 f121a7d f73df00 2c76c6c f121a7d f73df00 5850aac |
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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
from summary_extractor import Extractor
from Tags_Extractor import Tags
from Clauses_Extractor import Clauses
from key_value_extractor import KeyValue
import openai
from pdftojson import PdftoJson
import gradio as gr
def file_output_fn(file_path):
file_path = file_path.name
return file_path
with gr.Blocks(css="style.css",theme='xiaobaiyuan/theme_brief') as demo:
gr.HTML("""<center class="darkblue" style='background-color:#ad6e07; text-align:center;padding:25px;'>
<be>
<h1 style="color:#fff">
Contract
</h1>
</center>""")
with gr.Tab("Contract Summary"):
with gr.Row(elem_id = "col-container",scale=0.80):
with gr.Column(elem_id = "col-container",scale=0.80):
file_output = gr.File(label="File Status",elem_classes="filenameshow")
with gr.Column(elem_id = "col-container",scale=0.20):
upload_button = gr.UploadButton(
"Browse File",file_types=[".txt", ".pdf", ".doc", ".docx",".json",".csv"],
elem_classes="uploadbutton")
summary_btn = gr.Button("Get Summary",elem_classes="uploadbutton")
with gr.Row(elem_id = "col-container",scale=0.60):
summary = gr.Textbox(label = "Summary")
with gr.Row(elem_id = "col-container",scale=0.80):
with gr.Tab("Tags"):
with gr.Column(elem_id = "col-container",scale=0.80):
tags_btn = gr.Button("Tags Extracter")
tags = gr.Textbox(label = "Tags", lines=10)
with gr.Tab("Key Values"):
with gr.Column(elem_id = "col-container",scale=0.80):
key_value_btn = gr.Button("Key Value Extracter")
key_value = gr.Textbox(label = "Key Value",lines=10)
with gr.Tab("Clauses"):
with gr.Column(elem_id = "col-container",scale=0.80):
clauses_btn = gr.Button("Clauses Extracter")
clauses = gr.Textbox(label = "Clauses", lines=10)
extractor = Extractor()
upload_button.upload(file_output_fn,upload_button,file_output)
summary_btn.click( extractor._refine_summary,[upload_button],summary)
tags_btn.click(Tags.extract_tags,summary,tags)
key_value_btn.click( KeyValue.extract_key_value_pair,summary,key_value)
clauses_btn.click(Clauses.get_extracted_clauses,summary,clauses)
with gr.Tab("Tags"):
pass
demo.launch(debug=True) |