Update app.py

app.py

@@ -15,7 +15,7 @@ from fastapi import FastAPI, Form, File, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from fastapi.staticfiles import StaticFiles
-from dotenv import load_dotenv  # Added for environment variable loading
+from dotenv import load_dotenv
 
 # Load environment variables from .env file
 load_dotenv()
@@ -28,14 +28,14 @@ logger = logging.getLogger(__name__)
 SALESFORCE_USERNAME = os.getenv("SALESFORCE_USERNAME")
 SALESFORCE_PASSWORD = os.getenv("SALESFORCE_PASSWORD")
 SALESFORCE_SECURITY_TOKEN = os.getenv("SALESFORCE_SECURITY_TOKEN")
-SALESFORCE_DOMAIN = os.getenv("SALESFORCE_DOMAIN", "login")  # Default to 'login' if not set
+SALESFORCE_DOMAIN = os.getenv("SALESFORCE_DOMAIN", "login")
 
 # Validate that credentials are set
 if not all([SALESFORCE_USERNAME, SALESFORCE_PASSWORD, SALESFORCE_SECURITY_TOKEN]):
     logger.error("Salesforce credentials not set in environment variables. Please set SALESFORCE_USERNAME, SALESFORCE_PASSWORD, and SALESFORCE_SECURITY_TOKEN.")
     raise ValueError("Missing Salesforce credentials in environment variables.")
 
-logger.debug(f"Using Salesforce credentials - Username: {SALESFORCE_USERNAME}, Security Token: {SALESFORCE_SECURITY_TOKEN[:5]}...")
+logger.debug("Using Salesforce credentials - Username and Security Token loaded from environment variables.")
 
 # Function to authenticate with Salesforce
 def get_salesforce_connection():
@@ -318,7 +318,6 @@ def gradio_interface(boq_file, weather, workforce, location, project_title):
         if not update_result:
             logger.warning("Failed to update record with PDF URL, but record was created")
 
-        # Upload Gantt chart image to Salesforce for a public URL
         image_content_id = upload_file_to_salesforce(file_path, f"{project_title}_Gantt_Chart.png", record_id)
         image_url = None
         if image_content_id:
@@ -351,13 +350,19 @@ app.add_middleware(
 app.mount("/static", StaticFiles(directory=tempfile.gettempdir()), name="static")
 
 # Mount Gradio's static assets to fix UI rendering issue
-gradio_static_path = os.path.join(gr.__path__[0], "templates", "frontend", "dist")
-if os.path.exists(gradio_static_path):
-    app.mount("/_app", StaticFiles(directory=gradio_static_path), name="gradio_static")
-    logger.info(f"Mounted Gradio static assets at {gradio_static_path}")
-else:
-    logger.error(f"Gradio static assets directory not found at {gradio_static_path}")
-    raise FileNotFoundError(f"Gradio static assets not found at {gradio_static_path}")
+gradio_base_path = gr.__path__[0]
+gradio_static_path = os.path.join(gradio_base_path, "templates", "frontend", "dist")
+
+# Fallback to alternative path if dist is not found
+if not os.path.exists(gradio_static_path):
+    gradio_static_path = os.path.join(gradio_base_path, "templates", "frontend", "build")
+    if not os.path.exists(gradio_static_path):
+        logger.error(f"Gradio static assets directory not found at {gradio_static_path}. Attempted paths: templates/frontend/dist, templates/frontend/build.")
+        logger.error("Please ensure Gradio's frontend assets are built by reinstalling Gradio or manually building the frontend with 'npm run build' in the Gradio frontend directory.")
+        raise FileNotFoundError(f"Gradio static assets not found. Please rebuild Gradio's frontend assets.")
+
+app.mount("/_app", StaticFiles(directory=gradio_static_path), name="gradio_static")
+logger.info(f"Mounted Gradio static assets at {gradio_static_path}")
 
 # Health check endpoint to verify static asset availability
 @app.get("/health")
@@ -379,7 +384,6 @@ async def api_gradio_interface(
     try:
         logger.info("Starting api_gradio_interface...")
 
-        # Save uploaded file to temporary path
         temp_dir = tempfile.mkdtemp()
         boq_file_path = os.path.join(temp_dir, boq_file.filename)
         with open(boq_file_path, "wb") as f: