Spaces:
Runtime error
Runtime error
| #!/usr/bin/env python3 | |
| """ | |
| Working AutoTrain solution - bypasses user permission issues | |
| Fixed version to handle _TemplateResponse error and static files | |
| """ | |
| import os | |
| import pwd | |
| import getpass | |
| import sys | |
| # Fix environment variables FIRST | |
| os.environ["OMP_NUM_THREADS"] = "1" | |
| os.environ["MKL_NUM_THREADS"] = "1" | |
| os.environ["NUMEXPR_NUM_THREADS"] = "1" | |
| os.environ["OPENBLAS_NUM_THREADS"] = "1" | |
| os.environ["HF_HOME"] = "/tmp/huggingface_cache" | |
| # Disable authentication checks | |
| os.environ["DISABLE_OAUTH"] = "1" | |
| os.environ["IS_RUNNING_IN_SPACE"] = "false" | |
| print("��� Environment variables fixed!") | |
| print(f"✅ OMP_NUM_THREADS = {os.environ.get('OMP_NUM_THREADS')}") | |
| # Create cache and working directories | |
| os.makedirs("/tmp/huggingface_cache", exist_ok=True) | |
| os.makedirs("/tmp/autotrain", exist_ok=True) | |
| # Set working directory for database | |
| os.chdir("/tmp/autotrain") | |
| # Monkey patch the getuser function to fix the permission error | |
| def mock_getuser(): | |
| return "user" | |
| getpass.getuser = mock_getuser | |
| # Also patch pwd.getpwuid for the same issue | |
| original_getpwuid = pwd.getpwuid | |
| def mock_getpwuid(uid): | |
| class MockPwdEntry: | |
| def __init__(self): | |
| self.pw_name = "user" | |
| self.pw_uid = uid | |
| self.pw_gid = uid | |
| self.pw_dir = "/app" | |
| self.pw_shell = "/bin/bash" | |
| def __getitem__(self, index): | |
| if index == 0: | |
| return self.pw_name | |
| elif index == 1: | |
| return "x" # password placeholder | |
| elif index == 2: | |
| return self.pw_uid | |
| elif index == 3: | |
| return self.pw_gid | |
| elif index == 4: | |
| return "User" # gecos | |
| elif index == 5: | |
| return self.pw_dir | |
| elif index == 6: | |
| return self.pw_shell | |
| else: | |
| raise IndexError("list index out of range") | |
| return MockPwdEntry() | |
| pwd.getpwuid = mock_getpwuid | |
| print("��� User permission patches applied!") | |
| if __name__ == "__main__": | |
| print("��� Starting AutoTrain...") | |
| try: | |
| import uvicorn | |
| from fastapi import FastAPI, Request | |
| from fastapi.responses import HTMLResponse | |
| from fastapi.staticfiles import StaticFiles | |
| from starlette.middleware.sessions import SessionMiddleware | |
| from starlette.templating import Jinja2Templates | |
| import autotrain.app.ui_routes as ui_routes | |
| # Create a mock user authentication that returns the right format | |
| def mock_user_authentication(request: Request = None): | |
| # Return a tuple or list that has len() method | |
| return ("user", []) # (username, orgs) | |
| # Patch the authentication function before importing ui_router | |
| ui_routes.user_authentication = mock_user_authentication | |
| # Also patch the load_index function to handle the authentication properly | |
| original_load_index = ui_routes.load_index | |
| def patched_load_index(request: Request): | |
| try: | |
| # Set a mock user in the request session | |
| request.session["user"] = "user" | |
| request.session["orgs"] = [] | |
| # Call the original function | |
| return original_load_index(request) | |
| except Exception as e: | |
| print(f"ERROR in load_index: {e}") | |
| # Return a simple HTML response if there's an error | |
| return HTMLResponse(content=""" | |
| <html> | |
| <head><title>AutoTrain Advanced</title></head> | |
| <body> | |
| <h1>AutoTrain Advanced</h1> | |
| <p>Welcome to AutoTrain! The application is starting up...</p> | |
| <p>If you see this message, the server is running but there may be authentication issues.</p> | |
| </body> | |
| </html> | |
| """) | |
| ui_routes.load_index = patched_load_index | |
| from autotrain.app.ui_routes import ui_router | |
| app = FastAPI(title="AutoTrain Advanced") | |
| # Add session middleware | |
| app.add_middleware(SessionMiddleware, secret_key="autotrain-secret-key") | |
| # Mount static files | |
| app.mount("/static", StaticFiles(directory="static"), name="static") | |
| app.include_router(ui_router) | |
| print("✅ AutoTrain app created successfully!") | |
| uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info") | |
| except Exception as e: | |
| print(f"❌ Error: {e}") | |
| import traceback | |
| traceback.print_exc() | |