Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
#
|
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 |
|