Update app.py

app.py

@@ -15,16 +15,25 @@ 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
+
+# Load environment variables from .env file
+load_dotenv()
 
 # Configure logging to show detailed messages
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
-# Salesforce credentials (use environment variables in production)
-SALESFORCE_USERNAME = os.getenv("SALESFORCE_USERNAME", "[email protected]")
-SALESFORCE_PASSWORD = os.getenv("SALESFORCE_PASSWORD", "#Prasad@1926")
-SALESFORCE_SECURITY_TOKEN = os.getenv("SALESFORCE_SECURITY_TOKEN", "4rmMYJ0augm8HKv9lpWXKYkD6")
-SALESFORCE_DOMAIN = os.getenv("SALESFORCE_DOMAIN", "login")  # or "test" for sandbox
+# Salesforce credentials (loaded from environment variables)
+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
+
+# 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]}...")
 
@@ -337,8 +346,26 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
+
+# Mount directory for temporary files (e.g., Gantt chart PNGs)
 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}")
+
+# Health check endpoint to verify static asset availability
+@app.get("/health")
+async def health_check():
+    if os.path.exists(gradio_static_path):
+        return {"status": "healthy", "gradio_static": "available"}
+    return {"status": "unhealthy", "gradio_static": "missing"}
+
 # FastAPI endpoint for Salesforce controller (defined before mounting Gradio)
 @app.post("/api/gradio_interface")
 async def api_gradio_interface(
@@ -440,8 +467,8 @@ async def api_gradio_interface(
             shutil.rmtree(temp_dir)
             logger.debug(f"Cleaned up temporary directory: {temp_dir}")
 
-# Create Gradio interface
-demo = gr.Blocks(theme="default")
+# Create Gradio interface with explicit base_path
+demo = gr.Blocks(theme="default", base_path="/")
 with demo:
     gr.Markdown("## AI Civil Work Planner")
     gr.Markdown("Generate a project timeline (Gantt chart) and risk tags based on BOQ data and site parameters.")
@@ -465,7 +492,7 @@ with demo:
         outputs=[output_image, risk_tags],
     )
 
-# Mount Gradio app (use a different variable to avoid overwriting 'app')
+# Mount Gradio app
 gradio_app = gr.mount_gradio_app(app, demo, path="/")
 
 if __name__ == "__main__":