Update main.py

main.py

@@ -9,13 +9,15 @@ app = FastAPI(
     version="1.0.0",
 )
 
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+
 def ensure_dir(path: str):
     os.makedirs(path, exist_ok=True)
     os.chmod(path, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
     return path
 
-OUTPUT_DIR = ensure_dir(os.path.join(os.getcwd(), "output"))
-TMP_DIR = ensure_dir(os.path.join(os.getcwd(), "tmp"))
+OUTPUT_DIR = ensure_dir(os.path.join(BASE_DIR, "output"))
+TMP_DIR = ensure_dir(os.path.join(BASE_DIR, "tmp"))
 
 async def delete_later(path: str, delay: int = 300):
     await asyncio.sleep(delay)
@@ -23,7 +25,17 @@ async def delete_later(path: str, delay: int = 300):
         os.remove(path)
 
 # =========================
-# BRAT IMAGE
+# ROOT → index.html
+# =========================
+@app.get("/")
+async def root():
+    html = os.path.join(BASE_DIR, "index.html")
+    if not os.path.exists(html):
+        return JSONResponse(status_code=404, content={"error": "index.html not found"})
+    return FileResponse(html, media_type="text/html")
+
+# =========================
+# BRAT IMAGE (INLINE VIEW)
 # =========================
 @app.get("/api/brat")
 async def brat_image(text: str = Query(...)):
@@ -31,13 +43,13 @@ async def brat_image(text: str = Query(...)):
     if not text:
         return JSONResponse(status_code=400, content={"error": "text is required"})
 
-    filename = f"brat-{uuid.uuid4().hex}.png"
-    filepath = os.path.join(OUTPUT_DIR, filename)
+    filepath = os.path.join(OUTPUT_DIR, f"brat-{uuid.uuid4().hex}.png")
 
     async with async_playwright() as p:
         browser = await p.chromium.launch(args=["--no-sandbox"])
-        context = await browser.new_context(viewport={"width": 1536, "height": 695})
-        page = await context.new_page()
+        page = await (await browser.new_context(
+            viewport={"width": 1536, "height": 695}
+        )).new_page()
 
         await page.goto("https://www.bratgenerator.com/", wait_until="domcontentloaded")
         try:
@@ -50,8 +62,8 @@ async def brat_image(text: str = Query(...)):
         await page.fill("#textInput", text)
         await asyncio.sleep(0.4)
 
-        element = await page.query_selector("#textOverlay")
-        box = await element.bounding_box()
+        el = await page.query_selector("#textOverlay")
+        box = await el.bounding_box()
 
         img = await page.screenshot(
             clip={"x": box["x"], "y": box["y"], "width": 500, "height": 440}
@@ -64,10 +76,11 @@ async def brat_image(text: str = Query(...)):
 
     asyncio.create_task(delete_later(filepath))
 
-    return FileResponse(filepath, media_type="image/png", filename=filename)
+    # ⬅️ INLINE (preview di browser)
+    return FileResponse(filepath, media_type="image/png")
 
 # =========================
-# BRAT VIDEO
+# BRAT VIDEO (INLINE VIEW)
 # =========================
 @app.get("/api/bratvid")
 async def brat_video(text: str = Query(...)):
@@ -80,8 +93,9 @@ async def brat_video(text: str = Query(...)):
 
     async with async_playwright() as p:
         browser = await p.chromium.launch(args=["--no-sandbox"])
-        context = await browser.new_context(viewport={"width": 1536, "height": 695})
-        page = await context.new_page()
+        page = await (await browser.new_context(
+            viewport={"width": 1536, "height": 695}
+        )).new_page()
 
         await page.goto("https://www.bratgenerator.com/", wait_until="domcontentloaded")
         try:
@@ -93,11 +107,11 @@ async def brat_video(text: str = Query(...)):
         await page.click("#textOverlay")
 
         for i in range(len(words)):
-            await page.fill("#textInput", " ".join(words[: i + 1]))
+            await page.fill("#textInput", " ".join(words[:i+1]))
             await asyncio.sleep(0.2)
 
-            element = await page.query_selector("#textOverlay")
-            box = await element.bounding_box()
+            el = await page.query_selector("#textOverlay")
+            box = await el.bounding_box()
 
             img = await page.screenshot(
                 clip={"x": box["x"], "y": box["y"], "width": 500, "height": 440}
@@ -108,25 +122,21 @@ async def brat_video(text: str = Query(...)):
 
         await browser.close()
 
-    output_video = os.path.join(OUTPUT_DIR, f"bratvid-{uuid.uuid4().hex}.mp4")
+    output = os.path.join(OUTPUT_DIR, f"bratvid-{uuid.uuid4().hex}.mp4")
 
-    ffmpeg_cmd = [
+    await asyncio.create_subprocess_exec(
         "ffmpeg", "-y",
         "-framerate", "1.4",
         "-i", os.path.join(temp_dir, "frame%03d.png"),
         "-vf", "scale=trunc(iw/2)*2:trunc(ih/2)*2,fps=30",
         "-pix_fmt", "yuv420p",
-        output_video
-    ]
-
-    process = await asyncio.create_subprocess_exec(
-        *ffmpeg_cmd,
+        output,
         stdout=asyncio.subprocess.DEVNULL,
         stderr=asyncio.subprocess.DEVNULL
     )
-    await process.wait()
 
     shutil.rmtree(temp_dir, ignore_errors=True)
-    asyncio.create_task(delete_later(output_video))
+    asyncio.create_task(delete_later(output))
 
-    return FileResponse(output_video, media_type="video/mp4", filename=os.path.basename(output_video))
+    # ⬅️ INLINE VIDEO (play langsung)
+    return FileResponse(output, media_type="video/mp4")