Update app.py

app.py

@@ -351,23 +351,37 @@ 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]
-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}")
+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
 @app.get("/health")
 async def health_check():
-    if os.path.exists(gradio_static_path):
+    if gradio_static_path and os.path.exists(gradio_static_path):
         return {"status": "healthy", "gradio_static": "available"}
     return {"status": "unhealthy", "gradio_static": "missing"}