Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,94 @@
+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)