Spaces:
Running
Running
File size: 2,132 Bytes
5850aac 89fa8e9 5850aac f701e68 5850aac f701e68 2c76c6c 5850aac f701e68 2c76c6c 5850aac 2c76c6c 5850aac f701e68 2c76c6c 5850aac 89fa8e9 2c76c6c 89fa8e9 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 |
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
with gr.Blocks(css="style.css",theme='xiaobaiyuan/theme_brief') as demo:
gr.HTML(
"""<hr style="border-top: 5px solid white;">"""
)
gr.HTML(
"""<br>
<h1 style="text-align:center;">
Contract Management
</h1> """
)
gr.HTML(
"""<hr style="border-top: 5px solid white;">"""
)
with gr.Column(elem_id = "col-container",scale=0.10, min_width=160):
upload_button = gr.UploadButton(
"Browse File",file_types=[".txt", ".pdf", ".doc", ".docx",".json",".csv"],
elem_classes="filenameshow")
summary = gr.Textbox(label = "Summary")
extractor = Extractor()
upload_button.upload( extractor._refine_summary,[upload_button],summary)
with gr.Tab("Tags"):
with gr.Column(elem_id = "col-container"):
tags_btn = gr.Button("Tags Extracter")
tags = gr.Textbox(lines=4,label = "Tags")
tags_btn.click(Tags.extract_tags,summary,tags)
with gr.Tab("Key Values"):
with gr.Column(elem_id = "col-container"):
key_value_btn = gr.Button("Key Value Extracter")
key_value = gr.Textbox(label = "Key Value")
key_value_btn.click( KeyValue.extract_key_value_pair,summary,key_value)
with gr.Tab("Clauses"):
with gr.Column(elem_id = "col-container"):
clauses_btn = gr.Button("Clauses Extracter")
clauses = gr.Textbox(label = "Clauses")
clauses_btn.click(Clauses.get_extracted_clauses,summary,clauses)
# with gr.Tab("pdf to json"):
# with gr.Column(elem_id = "col-container"):
# pdf_to_json_btn = gr.Button("pdf to json Extracter")
# pdf_to_json = gr.Textbox(label = "Pdf to Json")
# pdf_to_json_btn.click(PdftoJson.extract_text_from_pdf,upload_button,pdf_to_json)
demo.launch(debug=True) |