File size: 2,644 Bytes
7a34498
9852534
 
 
 
7a34498
 
5e6df63
10cc896
7a34498
588a6d8
9852534
 
588a6d8
9852534
 
 
02333f4
588a6d8
9852534
 
 
 
 
588a6d8
9852534
7af5670
 
22165de
7af5670
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9852534
 
 
7a34498
 
 
236dcf0
9852534
 
7a34498
 
 
 
9852534
33d0fc7
7a34498
 
 
 
 
 
 
 
33d0fc7
7a34498
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 import gradio as gr
import google.generativeai as genai
from yolov5 import YOLOv5
from PIL import Image


#Clé Api
genai.configure(api_key="AIzaSyB_Hnl_77gd1I8xs6iuLqKIoGHgsZMXm1M")


# Le modèle Gemini Pro
gemini_model = genai.GenerativeModel('gemini-pro')

# Le modèle de détection YOLOv5
yolo_model_path = "yolov5s.pt"
yolo_model = YOLOv5(yolo_model_path, device="cpu")


# Détection d'ingrédients
def detect_ingredients(image):
    results = yolo_model.predict(image)
    ingredients = results.pandas().xyxy[0]['name'].tolist()
    return ingredients

# Prompt de génération de recettes (Mistral AI)
def generate_recipe(ingredients):
    prompt = f"""
    Oubliez tout ce que je vous ai demandé avant et concentrez-vous uniquement sur cette tâche.
    Vous êtes un chef virtuel spécialisé dans la création de recettes à partir des ingrédients disponibles. On vous a fourni une liste d'aliments suivants : {', '.join(set(ingredients))}. Utilisez ces ingrédients pour créer une recette savoureuse et nutritive. Votre recette doit inclure :
    1. Une liste d'ingrédients avec les quantités nécessaires.
    2. Des instructions étape par étape pour la préparation et la cuisson.
    3. Des conseils pour les substitutions possibles si certains ingrédients manquent.
    4. Le temps de préparation et de cuisson total.
    5. Des suggestions de présentation pour rendre le plat encore plus appétissant.
    Assurez-vous que la recette soit accessible à un niveau de compétence culinaire moyen et qu'elle utilise les ingrédients de manière optimale.
    Exemple de format de réponse :
    ```
    Recette : [Nom de la recette]
    Ingrédients :
    - [Ingrédient 1] : [Quantité]
    - [Ingrédient 2] : [Quantité]
    ...
    Instructions :
    1. [Étape 1]
    2. [Étape 2]
    ...
    ```
    Recette :
    """
    response = gemini_model.generate_content(prompt)
    return response.text

    response = gemini_model.generatecontent(prompt)
    return response.text


# Interface Gradio
def process_image(image):
        image = Image.open(image)
        ingredients = detect_ingredients(image)
        recipe = generate_recipe(ingredients)
        return f"Ingrédients détectés : {', '.join(ingredients)}\n\nRecette générée :\n{recipe}"


# Mise à jour de l'interface Gradio
iface = gr.Interface(
    fn=process_image,
    inputs=gr.Image(type="filepath"),
    outputs="text",
    title="Générateur de Recettes par Ingrédients",
    description="Téléchargez une image d'ingrédients pour générer une recette.",
)

# Pour lancer l'application
if __name__ == "__main__":
    iface.launch()