app.py

import gradio as gr
from pdf2image import convert_from_path
from PIL import Image

def process_file(file, file_type):
 if file is None:
 return None, "File tidak diunggah dengan benar. Silakan coba lagi."
 
 if file_type == "PDF":
 try:
 # Mengonversi PDF ke gambar
 images = convert_from_path(file.name)
 # Menyimpan gambar pertama sebagai output
 image = images[0] # Mengambil halaman pertama
 return image, None
 except Exception as e:
 return None, f"Terjadi kesalahan saat memproses file PDF: {str(e)}"
 elif file_type == "Image":
 try:
 # Menampilkan gambar
 image = Image.open(file.name)
 return image, None
 except Exception as e:
 return None, f"Terjadi kesalahan saat memproses file gambar: {str(e)}"

# 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 dengan opsi share=True
interface.launch(share=True)