teetch / app.py
Equityone's picture
Update app.py
5667d33 verified
import gradio as gr
from PIL import Image
import random
# Exemple de base de données de mots
WORDS_DATABASE = {
"cat": {"fr": "chat", "audio": None},
"house": {"fr": "maison", "audio": None},
}
# Simule un modèle de reconnaissance d'image
def recognize_object(image):
return random.choice(list(WORDS_DATABASE.keys()))
# Fonction principale
def handle_drawing(image):
recognized_object = recognize_object(image)
french_translation = WORDS_DATABASE.get(recognized_object, {}).get("fr", "Inconnu")
response = f"Objet reconnu : {recognized_object.capitalize()} (Français : {french_translation})"
return response, None
# Interface utilisateur
def create_interface():
with gr.Blocks() as demo:
gr.Markdown("# Apprends l'Anglais en Dessinant !")
canvas = gr.Sketchpad(label="Dessine ici !", shape=(256, 256))
response_text = gr.Textbox(label="Résultat", interactive=False)
submit_button = gr.Button("Reconnaître")
submit_button.click(handle_drawing, inputs=[canvas], outputs=[response_text, None])
return demo
# Lancement
if __name__ == "__main__":
demo = create_interface()
demo.launch()