from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from botasaurus.browser import browser, Driver
import time
import uvicorn

app = FastAPI()

class RequestData(BaseModel):
    text: str
    service: str = "clever"

@browser(
    headless=True,
    block_images=True,
    reuse_driver=True, # Оставляем браузер открытым для скорости последующих запросов
    cache=False
)
def humanize_task(driver: Driver, text):
    url = "https://cleverhumanizer.ai/"
    
    # 1. Загрузка страницы
    if url not in driver.current_url:
        print(f"Navigating to {url}")
        driver.get(url)
        time.sleep(8)
    
    # 2. Очистка и ввод текста (используем индекс 0 - первый редактор на странице)
    print("Finding editors...")
    editors = driver.select_all('.ql-editor')
    if len(editors) < 2:
        print("Not enough editors found. Refreshing...")
        driver.get(url)
        time.sleep(8)
        editors = driver.select_all('.ql-editor')
    
    if not editors:
        return "ERROR: No editors found"

    print("Typing text into Editor[0]...")
    in_editor = editors[0]
    # Очистка через JS
    driver.driver.execute_script("arguments[0].innerText = '';", in_editor.element)
    in_editor.type(text)
    time.sleep(2)

    # 3. Клик по кнопке (используем индекс 0 - первая кнопка .submitQuill)
    btns = driver.select_all('.submitQuill')
    if btns:
        print("Clicking Submit Button[0]...")
        driver.driver.execute_script("arguments[0].scrollIntoView();", btns[0].element)
        driver.driver.execute_script("arguments[0].click();", btns[0].element)
    else:
        return "ERROR: Submit button not found"

    # 4. Ожидание результата в Editor[1]
    print("Waiting for generation...")
    for i in range(45): # Ждем до 90 секунд
        time.sleep(2)
        # Получаем свежий список редакторов
        current_editors = driver.select_all('.ql-editor')
        if len(current_editors) >= 2:
            out_text = current_editors[1].text.strip()
            # Если текст появился и это не заглушка
            if out_text and len(out_text) > 10 and "New text will appear" not in out_text:
                print("Success! Result captured.")
                return out_text
        
        if i % 5 == 0:
            print(f"Still waiting... ({i*2}s)")
                
    return "TIMEOUT: Generation is too slow or failed"

@app.post("/humanize")
def humanize(data: RequestData):
    print(f"Received request for service: {data.service}")
    if data.service == "clever":
        try:
            result = humanize_task(data.text)
            if "ERROR" in result or "TIMEOUT" in result:
                raise HTTPException(status_code=500, detail=result)
            return {"result": result}
        except Exception as e:
            print(f"Task Exception: {str(e)}")
            raise HTTPException(status_code=500, detail=str(e))
    else:
        return {"error": "Only 'clever' service is supported currently"}

@app.get("/")
def health():
    return {"status": "alive", "service": "AI Humanizer API"}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7860)