Update app.py

app.py

@@ -8,6 +8,11 @@ from pdftojson import PdftoJson
 from headings_extractor import HeadingsExtractor
 from incompletesentencefinder import IncompleteSentenceFinder
 from aggressive_content_finder import AggressiveContentFinder
+from extract_date import ExtractDateAndDuration
+from invoice_extractor import InvoiceKeyValuePair
+from pdftojson import PdftoJson
+from ContractGenerator import ContractGenerator
+from contract_missing_clausses import ContractMissingClauses
 import gradio as gr
 
 def file_output_fn(file_path):
@@ -72,6 +77,32 @@ with gr.Blocks(css="style.css",theme='freddyaboulton/test-blue') as demo:
                 aggres_sen_btn = gr.Button("Aggressive Content Extractor")
                 aggressive_sentence = gr.Textbox(label = "Aggressive Content", lines=10)   
                 
+        with gr.TabItem("Extract Date"):
+            with gr.Column(elem_id = "col-container",scale=0.80):  
+                extract_date_btn = gr.Button("Extract date")
+                extracted_date = gr.Textbox(label = "Extract date", lines=10)  
+
+        with gr.TabItem("Invoice"):
+            with gr.Column(elem_id = "col-container",scale=0.80):  
+                extract_invoice_key_value = gr.Button("Extract Invoice Key Value")
+                extracted_invoice_key_value = gr.Textbox(label = "Extracted  Key Value", lines=10)   
+
+        with gr.TabItem("Pdf to Json"):
+            with gr.Column(elem_id = "col-container",scale=0.80):  
+                extract_json_btn = gr.Button("Extract Json from PDF")
+                extracted_json = gr.Textbox(label = "Json", lines=10)   
+
+        with gr.TabItem("Contract Generator"):
+            with gr.Column(elem_id = "col-container",scale=0.80):  
+                prompt = gr.Textbox(label="Prompt")
+                generate_btn = gr.Button("Generate")
+                contract_box = gr.Textbox(label = "Contract", lines=10)   
+
+        with gr.TabItem("Missing Features"):
+            with gr.Column(elem_id = "col-container",scale=0.80):  
+                find_missing_features_btn = gr.Button("Find Missing Features")
+                Output_box = gr.Textbox(label = "Output", lines=10)   
+                
     extractor  = Extractor() 
     upload_button.upload(file_output_fn,upload_button,file_output)
     summary_btn.click( extractor._refine_summary,[upload_button],summary)  
@@ -91,6 +122,22 @@ with gr.Blocks(css="style.css",theme='freddyaboulton/test-blue') as demo:
     incom_sen_btn.click(inc.get_incomplete_sentence,upload_button,incomplete_sentence)
 
     agg= AggressiveContentFinder()
-    aggres_sen_btn.click(agg.get_aggressive_content,upload_button,aggressive_sentence)      
-        
+    aggres_sen_btn.click(agg.get_aggressive_content,upload_button,aggressive_sentence)    
+
+    date =  ExtractDateAndDuration()
+    extract_date_btn.click(date.itrate_each_page,upload_button,extracted_date)
+
+    invoice = InvoiceKeyValuePair()
+    extract_invoice_key_value.click(invoice.extract_key_value_pair,upload_button,extracted_invoice_key_value)
+
+    pdftojson = PdftoJson()
+    extract_json_btn.click(pdftojson.extract_text_from_pdf,upload_button,extracted_json)
+
+    contract_gen = ContractGenerator()
+    generate_btn.click(contract_gen.generate_contract,prompt,contract_box)
+
+    cont_mis_clauses = ContractMissingClauses()
+    find_missing_features_btn(cont_mis_clauses.iterate_each_page,upload_button,Output_box)
+
+
 demo.launch(debug=True)