geethareddy commited on
Commit
86b5f4c
·
verified ·
1 Parent(s): 2b2c54c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -5,17 +5,12 @@ import pandas as pd
5
  from reportlab.lib.pagesizes import letter
6
  from reportlab.pdfgen import canvas
7
  from io import BytesIO
 
8
 
9
- # Load environment variables
10
  load_dotenv()
11
 
12
- # Required env vars check
13
- required_env_vars = ['SF_USERNAME', 'SF_PASSWORD', 'SF_SECURITY_TOKEN']
14
- missing_vars = [var for var in required_env_vars if not os.getenv(var)]
15
- if missing_vars:
16
- raise EnvironmentError(f"Missing required environment variables: {missing_vars}")
17
-
18
- # Salesforce credentials
19
  SF_USERNAME = os.getenv('SF_USERNAME')
20
  SF_PASSWORD = os.getenv('SF_PASSWORD')
21
  SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN')
@@ -51,7 +46,7 @@ def generate_pdf_report(supervisor_name, project_id, checklist, suggestions):
51
 
52
  return file_path
53
 
54
- # Function to generate CSV report
55
  def generate_csv_report(supervisor_name, project_id, checklist, suggestions):
56
  file_path = f"reports/{supervisor_name}_{project_id}_report.csv"
57
 
 
5
  from reportlab.lib.pagesizes import letter
6
  from reportlab.pdfgen import canvas
7
  from io import BytesIO
8
+ from dotenv import load_dotenv
9
 
10
+ # Load environment variables from .env file
11
  load_dotenv()
12
 
13
+ # Salesforce credentials from environment variables
 
 
 
 
 
 
14
  SF_USERNAME = os.getenv('SF_USERNAME')
15
  SF_PASSWORD = os.getenv('SF_PASSWORD')
16
  SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN')
 
46
 
47
  return file_path
48
 
49
+ # Function to generate a CSV report
50
  def generate_csv_report(supervisor_name, project_id, checklist, suggestions):
51
  file_path = f"reports/{supervisor_name}_{project_id}_report.csv"
52