diff --git a/main.py b/main.py index 4ca8991..ef3d252 100644 --- a/main.py +++ b/main.py @@ -2,14 +2,22 @@ import datetime import uuid import asyncio from pathlib import Path - -from fastapi import FastAPI, Depends, UploadFile, Form, File, HTTPException, BackgroundTasks -from starlette.requests import Request -from starlette.responses import HTMLResponse, FileResponse -from starlette.staticfiles import StaticFiles - -from sqlalchemy import select, update, func -from sqlalchemy.ext.asyncio.session import AsyncSession +import os +try: + from fastapi import FastAPI, Depends, UploadFile, Form, File, HTTPException, BackgroundTasks + from starlette.requests import Request + from starlette.responses import HTMLResponse, FileResponse + from starlette.staticfiles import StaticFiles + from sqlalchemy import select, update, func + from sqlalchemy.ext.asyncio.session import AsyncSession +except ImportError: + os.system("pip install -r requirements.txt") + from fastapi import FastAPI, Depends, UploadFile, Form, File, HTTPException, BackgroundTasks + from starlette.requests import Request + from starlette.responses import HTMLResponse, FileResponse + from starlette.staticfiles import StaticFiles + from sqlalchemy import select, update, func + from sqlalchemy.ext.asyncio.session import AsyncSession import settings from utils import delete_expire_files, storage, get_code, error_ip_limit, upload_ip_limit