app.py

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=gr.themes.Soft()) 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(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"):
        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"):
        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"):
        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"):
        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,summary,pdf_to_json)

    
demo.launch(debug=True)