Update headings_extractor.py

headings_extractor.py

@@ -16,6 +16,10 @@ class HeadingsExtractor:
             str: Extracted headings.
         """
         pass
+        
+    def file_output_fnn(self,file_path):
+        file_path = file_path.name
+        return file_path        
 
     def extarct_headings(self,contract_page: str) -> str:
 
@@ -75,3 +79,21 @@ class HeadingsExtractor:
         except Exception as e:
             # If an error occurs during the key-value extraction process, log the error
             print(f"Error while extracting text from PDF: {str(e)}")
+
+    def gradio_interface(self):
+        with gr.Blocks(css="style.css",theme='xiaobaiyuan/theme_brief') as demo:    
+            with gr.Row(elem_id = "col-container",scale=0.80):
+              with gr.Column(elem_id = "col-container",scale=0.80):
+                file1 = gr.File(label="File",elem_classes="filenameshow")
+            
+              with gr.Column(elem_id = "col-container",scale=0.20):  
+                upload_button1 = gr.UploadButton(
+                    "Browse File",file_types=[".txt", ".pdf", ".doc", ".docx",".json",".csv"],
+                    elem_classes="uploadbutton")
+                headings_btn = gr.Button("Get Headings",elem_classes="uploadbutton")
+            
+            with gr.Row(elem_id = "col-container",scale=0.60):    
+                headings = gr.Textbox(label = "Headings")
+
+        upload_button1.upload(file_output_fnn,upload_button1,file1)
+        headings_btn.click(extract_text,upload_button1,headings)