import gradio as gr
import pytesseract
from PIL import Image

def extract_text(image):
    # Convertir l'image en noir et blanc pour une meilleure extraction de texte
    image = image.convert('L')
    
    # Utiliser Pytesseract pour extraire le texte de l'image
    text = pytesseract.image_to_string(image)
    print(text)
    return text

# Définir l'interface utilisateur Gradio
inputs = gr.inputs.Image()
outputs = gr.outputs.Textbox()
interface = gr.Interface(fn=extract_text, inputs=inputs, outputs=outputs, 
                         title="Extraction de texte à partir d'une image", 
                         description="Téléchargez une image contenant du texte et cliquez sur 'Predict' pour extraire le texte.")
                         
interface.launch()