Update aggressive_content_finder.py

aggressive_content_finder.py

@@ -10,11 +10,12 @@ class AggressiveContentFinder:
 
     """
 
-    def __init__(self):
+    def __init__(self,filepath):
         """
         Initialize the AggressiveContentFinder with your OpenAI API key.
         """
         # openai.api_key = openai_api_key
+        self.filepath = filepath
         pass
 
     def _extract_aggressive_content(self, contract_text: str) -> str:
@@ -58,9 +59,9 @@ class AggressiveContentFinder:
         debugging.
         """
         try:
-            print("path:",pdf_file_path.name)
+            print("path:",self.filepath)
             # Open the multi-page PDF using PdfReader
-            pdf = PdfReader(pdf_file_path.name)
+            pdf = PdfReader(self.filepath)
 
             aggressive_terms = ""
 
@@ -83,18 +84,18 @@ class AggressiveContentFinder:
 
     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.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")
-                aggressive_content = gr.Button("Get Aggressive Content",elem_classes="uploadbutton")
+              # 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")
+                # aggressive_content = gr.Button("Get Aggressive Content",elem_classes="uploadbutton")
             
             with gr.Row(elem_id = "col-container",scale=0.60):    
                 headings = gr.Textbox(label = "Aggressive Content")
 
-        upload_button1.upload(self.file_output_fnn,upload_button1,file1)
-        aggressive_content.click(self.get_aggressive_content,upload_button1,headings)  
+        # upload_button1.upload(self.file_output_fnn,upload_button1,file1)
+        aggressive_content.click(self.get_aggressive_content,[],headings)