Update app.py

app.py

@@ -58,11 +58,19 @@ with gr.Blocks(css="style.css",theme='xiaobaiyuan/theme_land') as demo:
                     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.TabItem("pdf"):
+                    with gr.Column(elem_id = "col-container",scale=0.80):  
+                        heading_btn = gr.Button("Clauses Extracter")
+                        heading = gr.Textbox(label = "Clauses", lines=10)
+                  
         
             extractor  = Extractor() 
             upload_button.upload(file_output_fn,upload_button,file_output)
             summary_btn.click( extractor._refine_summary,[upload_button],summary)  
+            ex= HeadingsExtractor()
             
+            heading_btn.click(ex.extract_tags,upload_button,heading)
             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)