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 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)

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.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")
    
      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="filenameshow")
        summary_btn = gr.Button("Get Summary",elem_classes="filenameshow")
    
    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)


        # with gr.Tab("pdf to json"):
        #     with gr.Column(elem_id = "col-container",scale=0.80):
        #         pdf_to_json_btn = gr.Button("pdf to json Extracter")
        #         pdf_to_json = gr.Textbox(label = "Pdf to Json", lines=10)
        
        #         # pdf_to_json_btn.click(PdftoJson.extract_text_from_pdf,upload_button,pdf_to_json)



    summary = gr.Textbox(label = "Summary")
    extractor  = Extractor()
          
    upload_button.upload(file_output_fn,upload_button,file_output)
    summary_btn.upload( 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)
    
demo.launch(debug=True)