Update app.py (#1)

- Update app.py (1d582c9619f293eb2f5871136392b0756404e0bd)


Co-authored-by: K Neeraj Kalyan <[email protected]>

app.py

@@ -88,7 +88,7 @@ def save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions
 
     temp_pdf_path = "/tmp/" + os.path.basename(file_path)
     shutil.copy(file_path, temp_pdf_path)
-    return temp_pdf_path,filename
+    return temp_pdf_path, filename
 
 def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_path, pdf_name, checklist, suggestions):
     try:
@@ -99,51 +99,34 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
             domain=os.getenv('SF_DOMAIN', 'login')
         )
 
-        # Read and encode the file as base64
         with open(pdf_path, "rb") as f:
             encoded = base64.b64encode(f.read()).decode()
 
-        # Create ContentVersion record to upload the PDF to Salesforce
         content = sf.ContentVersion.create({
             'Title': pdf_name,
             'PathOnClient': pdf_name,
             'VersionData': encoded
         })
 
-        # Get the ContentDocumentId for the uploaded PDF
         content_id = content['id']
         download_url = f"https://{sf.sf_instance}/sfc/servlet.shepherd/version/download/{content_id}"
 
-        # Query Salesforce to find the specific Supervisor_AI_Coaching__c record
-        query = sf.query(f"""
-            SELECT Id FROM Supervisor__c 
-            WHERE Name = '{supervisor_name}'
-            LIMIT 1
-        """)
-
-        # Ensure that the Supervisor is found
+        query = sf.query(f"SELECT Id FROM Supervisor__c WHERE Name = '{supervisor_name}' LIMIT 1")
         if query['totalSize'] == 0:
             print("⚠️ Supervisor not found!")
             return ""
 
         supervisor_id = query['records'][0]['Id']
 
-        # Query to get the Project ID in Salesforce
-        project_query = sf.query(f"""
-            SELECT Id FROM Project__c WHERE Name = '{project_id}'
-            LIMIT 1
-        """)
-
-        # Ensure that the Project ID is found
+        project_query = sf.query(f"SELECT Id FROM Project__c WHERE Name = '{project_id}' LIMIT 1")
         if project_query['totalSize'] == 0:
             print(f"⚠️ Project '{project_id}' not found in Salesforce!")
             return ""
 
         project_id_sf = project_query['records'][0]['Id']
 
-        # Create a new Supervisor_AI_Coaching__c record
         sf.Supervisor_AI_Coaching__c.create({
-            'Project_ID__c': project_id_sf,  # Use the Salesforce Project ID
+            'Project_ID__c': project_id_sf,
             'Supervisor_ID__c': supervisor_id,
             'Daily_Checklist__c': checklist,
             'Suggested_Tips__c': suggestions,
@@ -254,7 +237,7 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
 
 def create_interface():
     roles = get_roles_from_salesforce()
-    with gr.Blocks(theme="soft") as demo:
+    with gr.Blocks(theme="soft", css=".svelte-1ipelgc {display: none !important;}") as demo:
         gr.Markdown("## 🧠 AI-Powered Supervisor Assistant")
 
         with gr.Row():
@@ -303,4 +286,3 @@ def create_interface():
 if __name__ == "__main__":
     app = create_interface()
     app.launch()
-