File size: 2,050 Bytes
5850aac
 
 
 
 
 
 
 
89fa8e9
e0549a6
 
 
 
 
 
f701e68
5850aac
 
 
 
 
 
 
 
 
f701e68
3f71ccb
2c76c6c
 
5850aac
 
f701e68
3f71ccb
2c76c6c
5850aac
2c76c6c
5850aac
 
f701e68
3f71ccb
2c76c6c
 
 
5850aac
89fa8e9
 
3f71ccb
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
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("""<center class="darkblue" style='background-color:#ad6e07; text-align:center;padding:25px;'>
    <be>
    <h1 style="color:#fff">
        Contract Management 
    </h1>
    </center>""")  
    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 Extractor")
            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 Extractor")
            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 Extractor")
            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 Extractor")
    #         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)