Update app.py

app.py

@@ -6,48 +6,109 @@ import openai
 from pdftojson import PdftoJson
 import gradio as gr
 
+def file_output_fn(file_path):
+    file_path = file_path.name
+    return file_path
+
+# with gr.Blocks(css="style.css",theme='xiaobaiyuan/theme_brief') as demo:
+#     gr.HTML("""<center class="darkblue" style='background-color:#ad6e07; text-align:center;padding:25px;'>
+#     <be>
+#     <h1 style="color:#fff">
+#         Contract Management 
+#     </h1>
+#     </center>""")  
+#     with gr.Column(elem_id = "col-container",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"):
+#         with gr.Column(elem_id = "col-container"):
+#             tags_btn = gr.Button("Tags Extractor")
+#             tags = gr.Textbox(lines=4,label = "Tags")
+#             tags_btn.click(Tags.extract_tags,summary,tags)
+
+#     with gr.Tab("Key Values"):
+#         with gr.Column(elem_id = "col-container"):
+#             key_value_btn = gr.Button("Key Value Extractor")
+#             key_value = gr.Textbox(label = "Key Value")
+
+#             key_value_btn.click( KeyValue.extract_key_value_pair,summary,key_value)
+
+#     with gr.Tab("Clauses"):
+#         with gr.Column(elem_id = "col-container"):
+#             clauses_btn = gr.Button("Clauses Extractor")
+#             clauses = gr.Textbox(label = "Clauses")
+    
+#             clauses_btn.click(Clauses.get_extracted_clauses,summary,clauses)
+
+#     # with gr.Tab("pdf to json"):
+#     #     with gr.Column(elem_id = "col-container"):
+#     #         pdf_to_json_btn = gr.Button("pdf to json Extractor")
+#     #         pdf_to_json = gr.Textbox(label = "Pdf to Json")
+    
+#     #         pdf_to_json_btn.click(PdftoJson.extract_text_from_pdf,upload_button,pdf_to_json)
+
 with gr.Blocks(css="style.css",theme='xiaobaiyuan/theme_brief') as demo:
     gr.HTML("""<center class="darkblue" style='background-color:#ad6e07; text-align:center;padding:25px;'>
     <be>
     <h1 style="color:#fff">
-        Contract Management 
+        Contract 
     </h1>
     </center>""")  
-    with gr.Column(elem_id = "col-container",scale=0.10, min_width=160):
+    
+    with gr.Row(elem_id = "col-container",scale=0.80):
+      with gr.Column(elem_id = "col-container",scale=0.80):
+        file_output = gr.File(label="File Status")
+    
+      with gr.Column(elem_id = "col-container",scale=0.20):  
         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"):
-        with gr.Column(elem_id = "col-container"):
-            tags_btn = gr.Button("Tags Extractor")
-            tags = gr.Textbox(lines=4,label = "Tags")
-            tags_btn.click(Tags.extract_tags,summary,tags)
-
-    with gr.Tab("Key Values"):
-        with gr.Column(elem_id = "col-container"):
-            key_value_btn = gr.Button("Key Value Extractor")
-            key_value = gr.Textbox(label = "Key Value")
-
-            key_value_btn.click( KeyValue.extract_key_value_pair,summary,key_value)
-
-    with gr.Tab("Clauses"):
-        with gr.Column(elem_id = "col-container"):
-            clauses_btn = gr.Button("Clauses Extractor")
-            clauses = gr.Textbox(label = "Clauses")
+        summary_btn = gr.Button("Get Summary",elem_classes="filenameshow")
     
-            clauses_btn.click(Clauses.get_extracted_clauses,summary,clauses)
+    with gr.Row(elem_id = "col-container",scale=0.60):    
+        summary = gr.Textbox(label = "Summary")
 
-    # with gr.Tab("pdf to json"):
-    #     with gr.Column(elem_id = "col-container"):
-    #         pdf_to_json_btn = gr.Button("pdf to json Extractor")
-    #         pdf_to_json = gr.Textbox(label = "Pdf to Json")
-    
-    #         pdf_to_json_btn.click(PdftoJson.extract_text_from_pdf,upload_button,pdf_to_json)
+    with gr.Row(elem_id = "col-container",scale=0.80):
+        with gr.Tab("Tags"):
+            with gr.Column(elem_id = "col-container",scale=0.80):
+                tags_btn = gr.Button("Tags Extracter")
+                tags = gr.Textbox(label = "Tags", lines=10)
+
+        with gr.Tab("Key Values"):
+            with gr.Column(elem_id = "col-container",scale=0.80):
+                key_value_btn = gr.Button("Key Value Extracter")
+                key_value = gr.Textbox(label = "Key Value",lines=10)
+
+
+        with gr.Tab("Clauses"):
+            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.Tab("pdf to json"):
+        #     with gr.Column(elem_id = "col-container",scale=0.80):
+        #         pdf_to_json_btn = gr.Button("pdf to json Extracter")
+        #         pdf_to_json = gr.Textbox(label = "Pdf to Json", lines=10)
+        
+        #         # pdf_to_json_btn.click(PdftoJson.extract_text_from_pdf,upload_button,pdf_to_json)
+
+
+
+    summary = gr.Textbox(label = "Summary")
+    extractor  = Extractor()
+          
+    upload_button.upload(file_output_fn,upload_button,file_output)
+    summary_btn.upload( extractor._refine_summary,[upload_button],summary)  
+    
+    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)
     
 demo.launch(debug=True)