Create app.py

app.py

@@ -0,0 +1,175 @@
+import os
+import time
+import gradio as gr
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.by import By
+import tempfile
+import logging
+
+# Konfigurasi logging dasar
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+# Function to capture a screenshot with configurable options
+def capture_screenshot_gradio(url, window_width, window_height, internal_delay):
+    """
+    Mengambil screenshot halaman web dengan ukuran jendela dan delay yang bisa diatur.
+
+    Args:
+        url (str): URL halaman web yang akan di-screenshot.
+        window_width (int): Lebar jendela browser dalam pixel.
+        window_height (int): Tinggi jendela browser dalam pixel.
+        internal_delay (int): Waktu tunggu tambahan (detik) setelah readyState 'complete'.
+
+    Returns:
+        str: Path ke file screenshot sementara.
+
+    Raises:
+        gr.Error: Jika terjadi kesalahan saat mengambil screenshot.
+    """
+    # Validasi URL sederhana atau tambahkan 'http://' jika tidak ada
+    if not url.startswith(('http://', 'https://')):
+        logging.info(f"Menambahkan 'http://' ke URL: {url}")
+        url = 'http://' + url
+
+    # Pastikan width dan height adalah integer
+    try:
+        window_width = int(window_width)
+        window_height = int(window_height)
+        internal_delay = int(internal_delay) # Delay juga harus integer
+        if window_width <= 0 or window_height <= 0 or internal_delay < 0:
+             raise ValueError("Width, Height harus positif dan Delay tidak boleh negatif")
+    except (ValueError, TypeError) as e:
+        logging.error(f"Input tidak valid untuk ukuran/delay: {e}")
+        raise gr.Error(f"Input tidak valid: Lebar, Tinggi harus angka positif dan Delay tidak boleh negatif.")
+
+    logging.info(f"Opsi yang digunakan: Lebar={window_width}, Tinggi={window_height}, Delay={internal_delay} detik")
+
+    chrome_options = Options()
+    chrome_options.add_argument('--headless')
+    chrome_options.add_argument('--disable-gpu')
+    chrome_options.add_argument('--no-sandbox')
+    # Mengatur ukuran jendela sesuai input pengguna
+    chrome_options.add_argument(f'--window-size={window_width},{window_height}')
+    chrome_options.add_argument('--disable-dev-shm-usage')
+    chrome_options.add_argument('--remote-debugging-port=9222') # Opsi ini mungkin tidak selalu diperlukan
+
+    driver = None
+    screenshot_path = None
+
+    try:
+        logging.info("Menginisialisasi WebDriver Chrome...")
+        # Pertimbangkan menggunakan webdriver-manager jika ChromeDriver tidak ada di PATH
+        # from selenium.webdriver.chrome.service import Service as ChromeService
+        # from webdriver_manager.chrome import ChromeDriverManager
+        # service = ChromeService(executable_path=ChromeDriverManager().install())
+        # driver = webdriver.Chrome(service=service, options=chrome_options)
+        driver = webdriver.Chrome(options=chrome_options)
+
+        logging.info(f"Membuka URL: {url}")
+        driver.get(url)
+
+        # 1. Tunggu hingga status dokumen 'complete'
+        logging.info("Menunggu document.readyState menjadi 'complete'...")
+        WebDriverWait(driver, 30).until( # Timeout lebih lama untuk halaman kompleks
+            lambda d: d.execute_script('return document.readyState') == 'complete'
+        )
+        logging.info("document.readyState sudah 'complete'.")
+
+        # 2. Tambahkan delay internal setelah 'complete'
+        if internal_delay > 0:
+            logging.info(f"Menunggu delay tambahan selama {internal_delay} detik...")
+            time.sleep(internal_delay)
+            logging.info("Delay selesai.")
+        else:
+            logging.info("Tidak ada delay tambahan (0 detik).")
+
+        # Buat file sementara untuk menyimpan screenshot
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_file:
+            screenshot_path = tmp_file.name
+
+        logging.info(f"Mengambil screenshot ke: {screenshot_path}")
+        # Ambil screenshot dari seluruh halaman (mungkin perlu scroll jika konten panjang,
+        # tapi save_screenshot biasanya mengambil viewport saat ini dalam mode headless)
+        driver.save_screenshot(screenshot_path)
+        logging.info("Screenshot berhasil diambil.")
+
+        return screenshot_path
+
+    except Exception as e:
+        logging.error(f"Terjadi kesalahan saat memproses URL {url}: {e}", exc_info=True)
+        if screenshot_path and os.path.exists(screenshot_path):
+            try:
+                os.remove(screenshot_path)
+                logging.info(f"Membersihkan file sementara yang gagal: {screenshot_path}")
+            except OSError as remove_err:
+                logging.error(f"Gagal menghapus file sementara {screenshot_path}: {remove_err}")
+        raise gr.Error(f"Gagal mengambil screenshot: {e}")
+
+    finally:
+        if driver:
+            logging.info("Menutup WebDriver...")
+            driver.quit()
+            logging.info("WebDriver ditutup.")
+
+# --- Membuat Antarmuka Gradio ---
+
+# 1. Definisikan Komponen Input
+url_input = gr.Textbox(
+    label="Masukkan URL Website",
+    placeholder="contoh: https://www.google.com atau example.com"
+)
+
+width_input = gr.Number(
+    label="Lebar Jendela (px)",
+    value=1920, # Nilai default
+    minimum=300,  # Lebar minimum yang masuk akal
+    step=10,      # Langkah penambahan/pengurangan
+    info="Lebar viewport browser saat mengambil screenshot."
+)
+
+height_input = gr.Number(
+    label="Tinggi Jendela (px)",
+    value=1080, # Nilai default
+    minimum=200,  # Tinggi minimum yang masuk akal
+    step=10,      # Langkah penambahan/pengurangan
+    info="Tinggi viewport browser. Untuk halaman panjang, ini mungkin tidak menangkap seluruhnya."
+)
+
+delay_input = gr.Number(
+    label="Delay Tambahan (detik)",
+    value=5,      # Nilai default
+    minimum=0,    # Boleh 0 detik
+    step=1,       # Langkah penambahan/pengurangan
+    info="Waktu tunggu ekstra setelah halaman 'lengkap' untuk membiarkan konten dinamis (mis. JavaScript) selesai dimuat."
+)
+
+# 2. Definisikan Komponen Output
+output_image = gr.Image(
+    type="filepath", # Mengharapkan path file sebagai output
+    label="Hasil Screenshot"
+)
+
+# 3. Membuat instance Interface Gradio
+iface = gr.Interface(
+    fn=capture_screenshot_gradio, # Fungsi yang akan dijalankan
+    inputs=[url_input, width_input, height_input, delay_input], # Daftar komponen input (sesuai urutan argumen fungsi)
+    outputs=output_image,           # Komponen output
+    title="Pengambil Screenshot Website dengan Opsi",
+    description="Masukkan URL, atur ukuran jendela dan delay tambahan, lalu klik 'Submit' untuk mengambil screenshot. Delay membantu menangkap konten yang dimuat secara dinamis.",
+    allow_flagging='never',
+    examples=[
+        ["https://gradio.app", 1280, 720, 3],
+        ["https://github.com", 1920, 1080, 5],
+        ["https://www.google.com/maps", 1024, 768, 8] # Contoh dengan delay lebih lama
+    ]
+)
+
+# Menjalankan aplikasi Gradio
+if __name__ == "__main__":
+    print("Menjalankan aplikasi Gradio...")
+    print("Pastikan Google Chrome dan ChromeDriver yang sesuai sudah terinstall.")
+    print("Anda mungkin perlu menginstal: pip install selenium webdriver-manager gradio")
+    iface.launch()