Update app.py

app.py

@@ -19,7 +19,7 @@ airtable_api_key = os.environ.get('AIRTABLE_API_KEY')
 prompts_table_name = 'tblYIZEB8m6JkGDEP'
 users_table_name = 'tblLNe5ZL47SvrAEk'
 user_log_table_name = 'tblrlTsRrkl6BqMAJ'
-scoutlog_table_name = 'tblF30p0krA1FwOiD'
+compliancelog_table_name = 'tblQMXWKGlOonkIw2'
 policies_table_name = 'tbla6PC65qZfqdJhE'
 
 
@@ -33,7 +33,7 @@ response_label = f'<span style="display: inline-block; position: relative; z-ind
 base_id = 'appcUK3hUWC7GM2Kb'
 
 # App name for user login logging
-app = "Scout"
+app = "Compliance"
 
 headers = {
     "Authorization": f"Bearer {airtable_api_key}",
@@ -147,19 +147,13 @@ def get_prompt(header, template_content):
     return header, template_content
 
 
-def append_to_at_scountlog(school_selection, output_format, gender, interests, sports, gpt_response, number_of_athletes,
-                           social_media_channel, response_time, question_cost, prompt_tokens, completion_tokens):
-    airtable_endpoint = f'https://api.airtable.com/v0/{base_id}/{scoutlog_table_name}'
+def append_to_at_compliancelog(policy_name_dd,contract_text,gpt_response, response_time, question_cost, prompt_tokens, completion_tokens):
+    airtable_endpoint = f'https://api.airtable.com/v0/{base_id}/{compliancelog_table_name}'
 
     # Organize data for Airtable
     new_fields = {
-        'school_selection': str(school_selection),
-        'gender': str(gender),
-        'social_media_channel': str(social_media_channel),
-        'sports': str(sports),
-        'interests': str(interests),
-        'number_of_athletes': int(number_of_athletes),
-        'output_format': str(output_format),
+        'policy_name': str(policy_name_dd),
+        'contract_text': str(contract_text),
         'gpt_response': str(gpt_response),
         'response_time': str(response_time),
         'question_cost': question_cost,
@@ -191,7 +185,7 @@ def append_to_at_scountlog(school_selection, output_format, gender, interests, s
 
 
 # Chatbot Function
-def chatbot(policy_name,contract_text,progress=gr.Progress()):
+def chatbot(policy_name_dd,contract_text,progress=gr.Progress()):
 
     start_time = datetime.datetime.now()
 
@@ -206,7 +200,7 @@ def chatbot(policy_name,contract_text,progress=gr.Progress()):
     #print(policy_name)
 
     #students = get_students(school_selection)
-    get_policy_text(policy_name)
+    get_policy_text(policy_name_dd)
 
     #print(policy_text)
     #print(contract_text)
@@ -227,8 +221,10 @@ def chatbot(policy_name,contract_text,progress=gr.Progress()):
 
     trimmed_input = prompt_trim(analysis_input)
 
+    """   
     with open('analysis_input.txt', 'w', encoding='utf-8') as out_file:
         out_file.write(trimmed_input)
+    """
 
     response = openai.ChatCompletion.create(
         model="gpt-4",
@@ -254,9 +250,10 @@ def chatbot(policy_name,contract_text,progress=gr.Progress()):
     prompt_tokens = tokens_used.get('prompt_tokens', )
     completion_tokens = tokens_used.get('completion_tokens', 0)
 
+    """
     with open('response.txt', 'w', encoding='utf-8') as out_file:
         out_file.write(gpt_response)
-
+    """
     end_time = datetime.datetime.now()
 
     response_time = end_time - start_time
@@ -265,7 +262,7 @@ def chatbot(policy_name,contract_text,progress=gr.Progress()):
 
     #print(contract_redline)
 
-    #append_to_at_scountlog(school_selection, output_format, gender, interests, sports, gpt_response, number_of_athletes,social_media_channel, response_time, question_cost, prompt_tokens, completion_tokens)
+    append_to_at_compliancelog(policy_name_dd,contract_text,contract_redline, response_time, question_cost, prompt_tokens, completion_tokens)
 
     return {contract_redline_html: contract_redline, download_row: gr.Row(visible=True)}
 
@@ -321,27 +318,22 @@ def login_auth(username, password):
 
     return False
 
-def pdf_to_text(contract_file_cmpt):
 
-    file_text = extract_text(contract_file_cmpt.name)
+def pdf_to_text(contract_file_cmpt, contract_text_tbox, file_name_tbox):
 
+    file_text = extract_text(contract_file_cmpt.name)
     original_file_name = contract_file_cmpt.name.split("/")[-1]
     redline_file_name = original_file_name.split(".")[0]+" Redline.pdf"
 
-    return file_text,redline_file_name
+    return file_text, redline_file_name
 
-def download_pdf(file_name,compliance_comments,contract_redline_html):
+def download_pdf(compliance_comments,contract_redline_html,file_name_tbox):
 
-    contract_redline_comments = "<h2><u>Contract Redline:</u></h2><br>"+contract_redline + "<br><h2><u>Compliance Comments:</u></h2><br>"+compliance_comments
-
-    """
-    pdf_download = pdfkit.from_string(contract_redline_comments,False)
-
-    with open(file_name, 'wb') as f:
-        f.write(pdf_download)
-    """
+    contract_redline_comments = "<h2><u>Contract Redline:</u></h2><br>"+contract_redline_html + "<br><h2><u>Compliance Comments:</u></h2><br>"+compliance_comments
+    global pdf_download
+    pdf_download = pdfkit.from_string(contract_redline_comments,None)
     
-    return file_name
+    return {pdf_download_file: file_name_tbox}
 
 # Gradio UI
 CIMStheme = gr.themes.Soft().set(
@@ -355,7 +347,7 @@ schools = get_schools()
 # scout_response = "<!DOCTYPEhtml><html><head><title>HTMLTableGenerator</title><style>table{width:100%;border:2pxsolid#000000;padding:5px;}tableth{border:2pxsolid#000000;padding:5px;background:#f0f0f0;color:#000000;}tabletd{border:2pxsolid#000000;text-align:left;padding:5px;background:#ffffff;color:#000000;}</style></head><body><table><thead><tr><th>Rank</th><th>CIMS.AI <br>Scout Ranking</th><th>Athlete Name</th><th>Gender</th><th>Primary Sport</th><th>Social Media <br>Fan Total</th><th>Interests</th><th>Commentary on Ranking</th></tr></thead><tbody><tr><td></td><td></td><td></td><td></td><td></td><td></td><td></td><td></td></tr></tbody></table></body></html>"
 scout_response = "<!DOCTYPEhtml><html><head><title>HTMLTableGenerator</title><style>table{width:100%;border:2pxsolid#000000;padding:5px;background:#ffffff}tableth{border:2pxsolid#000000;padding:5px;background:#f0f0f0;color:#000000;}tabletd{border:2pxsolid#000000;text-align:left;padding:5px;background:#ffffff;color:#000000;}</style></head><body><table><thead><tr><th>Rank</th><th>CIMS.AI<br>Scout<br>Ranking</th><th>Athlete Name</th><th>Gender</th><th>Primary Sport</th><th>Social Media<br>Fan Total</th><th>Interests</th><th>Commentary on Ranking</th></tr></thead><tbody><tr><td></td><td></td><td></td><td></td><td></td><td></td><td></td><td></td></tr></tbody></table></body></html>"
 
-contract_redline = "<p><strong>Campaign:</strong></p><p><strong>Engagement Name:</strong> Applebee's Fall Burger Promo</p><p><strong>Engagement Id:</strong> 7015j000001DvCAAA0</p><p><strong>Sponsor:</strong> Applebee's</p><p><strong>Start Date:</strong> 2023-10-28</p><p><strong>End Date:</strong> 2023-11-11</p><p><strong>Engagement Description</strong></p><p>The goal of the engagement is to Increase Sales by having the Student-Athlete Social Media Post. For the Social Media Post, the sponsor is requesting the student athlete take a photo in front of the <span style='text-decoration: line-through;'>football stadium</span> in eating a Applebee's burger <span style='text-decoration: line-through;'>in your team jersey</span>. The Media rights for the content will be 90 Days.</p><p><strong>Engagement Compensation.</strong></p><p>For successful completion of the engagement the student-athlete will receive payment in the form of Cash.</p><p>Part or all of the payment will be in cash, paid via PayPal.</p><p>The total value of compensation will be 250.</p>"
+#contract_redline = "<p><strong>Campaign:</strong></p><p><strong>Engagement Name:</strong> Applebee's Fall Burger Promo</p><p><strong>Engagement Id:</strong> 7015j000001DvCAAA0</p><p><strong>Sponsor:</strong> Applebee's</p><p><strong>Start Date:</strong> 2023-10-28</p><p><strong>End Date:</strong> 2023-11-11</p><p><strong>Engagement Description</strong></p><p>The goal of the engagement is to Increase Sales by having the Student-Athlete Social Media Post. For the Social Media Post, the sponsor is requesting the student athlete take a photo in front of the <span style='text-decoration: line-through;'>football stadium</span> in eating a Applebee's burger <span style='text-decoration: line-through;'>in your team jersey</span>. The Media rights for the content will be 90 Days.</p><p><strong>Engagement Compensation.</strong></p><p>For successful completion of the engagement the student-athlete will receive payment in the form of Cash.</p><p>Part or all of the payment will be in cash, paid via PayPal.</p><p>The total value of compensation will be 250.</p>"
 #pdf_download = pdfkit.from_string(contract_redline, False)
 
 #print(pdf_download)
@@ -366,6 +358,7 @@ contract_text = ''
 policy_text = ''
 compliance_comments = ''
 file_name = 'redline.pdf'
+pdf_download = ''
 
 with gr.Blocks(CIMStheme) as iface:
     with gr.Row():
@@ -379,17 +372,17 @@ with gr.Blocks(CIMStheme) as iface:
 
     with gr.Row():
         with gr.Column(variant='panel',scale=1):
-            policy_name = gr.components.Dropdown(schools, multiselect=False,label="NIL Policy Selection")
+            policy_name_dd = gr.components.Dropdown(schools, multiselect=False,label="NIL Policy Selection")
             contract_file_cmpt = gr.File(label="Select Contract File",file_count="single",file_types=[".pdf"],height=150)
         with gr.Column(variant='panel',scale=3):
             contract_text_tbox = gr.Textbox(label="Contract Text",interactive=True,info="Upload .pdf or paste in text")
     with gr.Row():
         with gr.Column():
-            btn_upload = gr.components.Button(value="Upload Contract", size='sm', variant="primary")
+            upload_btn = gr.components.Button(value="Upload Contract", size='sm', variant="primary")
         with gr.Column():
             gr.Markdown("")
         with gr.Column():
-            btn_redline = gr.components.Button(value="Redline Contract", size='sm', variant="primary")
+            redline_btn = gr.components.Button(value="Redline Contract", size='sm', variant="primary")
         with gr.Column():
             gr.Markdown("")
     with gr.Row():
@@ -399,14 +392,18 @@ with gr.Blocks(CIMStheme) as iface:
             compliance_comments_tbox = gr.Textbox(interactive=True,label='Compliance Comments')
     with gr.Row(visible=False) as download_row:
         with gr.Column(variant='panel'):
-            file_name_tbox = gr.Textbox(interactive=True,label='File Name')
-            btn_download = gr.components.Button(value="Download Redline", size='sm', variant="primary")
-            #pdf_download = gr.File(label="Download Redline File",height=150)
-            btn_download.click(download_pdf,inputs=[file_name_tbox,compliance_comments_tbox,contract_redline_html])
-            btn_redline.click(chatbot,inputs=[policy_name,contract_text_tbox],outputs=[contract_redline_html,download_row])
-            btn_upload.click(pdf_to_text,inputs=contract_file_cmpt,outputs=[contract_text_tbox,file_name_tbox])
+            file_name_tbox = gr.Textbox(interactive=False,label='File Name',visible=False)
+            pdf_download_file = gr.File()
+            download_btn = gr.components.Button(value="Create Redline PDF", size='sm', variant="primary")
+            upload_btn.click(pdf_to_text,inputs=[contract_file_cmpt,contract_text_tbox,file_name_tbox],outputs=[contract_text_tbox,file_name_tbox])
+            redline_btn.click(chatbot,inputs=[policy_name_dd,contract_text_tbox],outputs=[contract_redline_html,download_row])
+            download_btn.click(download_pdf,inputs=[compliance_comments_tbox,contract_redline_html,file_name_tbox],outputs=pdf_download_file)
         with gr.Column():
             gr.Markdown("")
+            #pdf_download_file = gr.File()
+            #email_btn = gr.components.Button(value="Email Redline", size='sm', variant="primary")
+            #email_btn.click(send_contract_email)
+            #download_btn.click(download_pdf,inputs=[file_name_tbox,compliance_comments_tbox,contract_redline_html])
         with gr.Column():
             gr.Markdown("")
         with gr.Column():
@@ -415,6 +412,5 @@ with gr.Blocks(CIMStheme) as iface:
         with gr.Column():
             gr.HTML('<center><i>CIMS.AI Confidential 2023</i></center>')
 
-
 #iface.queue(concurrency_count=20).launch(auth=login_auth, auth_message= "Enter your username and password that you received from CIMS.AI. To request a login, please email 'info@cims.ai'")
 iface.queue(concurrency_count=20).launch()