Update app.py

app.py

@@ -32,7 +32,7 @@ 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.")
+    logger.error("Salesforce credentials not set in environment variables.")
     raise ValueError("Missing Salesforce credentials in environment variables.")
 
 logger.debug("Using Salesforce credentials - Username and Security Token loaded from environment variables.")
@@ -349,43 +349,12 @@ app.add_middleware(
 # 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_base_path = gr.__path__[0]
-possible_static_paths = [
-    os.path.join(gradio_base_path, "templates", "frontend", "dist"),
-    os.path.join(gradio_base_path, "templates", "frontend", "build"),
-    os.path.join(gradio_base_path, "static"),
-]
-
-gradio_static_path = None
-for path in possible_static_paths:
-    if os.path.exists(path) and os.path.exists(os.path.join(path, "_app")):
-        gradio_static_path = path
-        break
-
-if 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.warning(
-        "Gradio static assets not found in any expected paths: " +
-        ", ".join(possible_static_paths) +
-        ". Falling back to Gradio's built-in server for UI rendering."
-    )
-    # Run Gradio directly if assets cannot be found
-    if __name__ == "__main__":
-        logger.info("Starting Gradio's built-in server...")
-        demo.launch(server_name="0.0.0.0", server_port=7860)
-        exit(0)
-
-# Health check endpoint to verify static asset availability
+# Health check endpoint to verify server status
 @app.get("/health")
 async def health_check():
-    if gradio_static_path and os.path.exists(gradio_static_path):
-        return {"status": "healthy", "gradio_static": "available"}
-    return {"status": "unhealthy", "gradio_static": "missing"}
+    return {"status": "healthy"}
 
-# FastAPI endpoint for Salesforce controller (defined before mounting Gradio)
+# FastAPI endpoint for Salesforce controller
 @app.post("/api/gradio_interface")
 async def api_gradio_interface(
     boq_file: UploadFile = File(...),
@@ -485,8 +454,8 @@ async def api_gradio_interface(
             shutil.rmtree(temp_dir)
             logger.debug(f"Cleaned up temporary directory: {temp_dir}")
 
-# Create Gradio interface with explicit base_path
-demo = gr.Blocks(theme="default", base_path="/")
+# Create Gradio interface
+demo = gr.Blocks(theme="default")
 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.")