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 |
|