import gradio as gr
import pytesseract
import cv2
import numpy as np
from PIL import Image
from io import BytesIO

def extract_text(image):
    image = Image.open(image.name)
    image = image.convert('L')
    image = np.array(image)
    text = pytesseract.image_to_string(image)
    print(text)
    return text

# Définir l'interface utilisateur 
inputs = gr.Image(label='Upload an image',image_mode='L',tool='select',type='pil')
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()