import gradio as gr
from PyPDF2 import PdfReader
from PIL import Image

def process_file(file, file_type):
    if file_type == "PDF":
        # Membaca konten PDF
        reader = PdfReader(file.name)
        text = ""
        for page in reader.pages:
            text += page.extract_text()
        return None, f"File PDF berisi:\n{text}"
    elif file_type == "Image":
        # Menampilkan gambar
        image = Image.open(file.name)
        return image, None

# Buat antarmuka Gradio
file_type_radio = gr.Radio(choices=["Image", "PDF"], label="Pilih tipe file")
file_input = gr.File(label="Unggah file", file_types=["image", "pdf"])

interface = gr.Interface(
    fn=process_file,
    inputs=[file_input, file_type_radio],
    outputs=[gr.Image(type="pil", label="Hasil Gambar"), gr.Textbox(label="Hasil PDF")],
    title="Proses File Gambar dan PDF",
    description="Pilih tipe file dan unggah file gambar atau PDF untuk diproses."
)

# Jalankan antarmuka tanpa opsi share=True
interface.launch()